From 734c4dacd95362f8d54ebcf9878aed3f4fb9780b Mon Sep 17 00:00:00 2001
From: Raphael Pierzina <raphael@hackebrot.de>
Date: Thu, 28 Jun 2018 13:11:22 +0200
Subject: [PATCH] Add type hints to test_login.py

---
 tests/test_login.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/tests/test_login.py b/tests/test_login.py
index 4457be4..6d4d1d9 100644
--- a/tests/test_login.py
+++ b/tests/test_login.py
@@ -3,12 +3,16 @@
 """UI tests for the login page."""
 
 import pytest
-
 from selenium.common.exceptions import TimeoutException
 
+from models.user import User
+from pages.login import LoginPage
+
 
 @pytest.mark.nondestructive
-def test_login_wrong_user_credentials(login_page, unknown_user):
+def test_login_wrong_user_credentials(
+    login_page: LoginPage, unknown_user: User
+) -> None:
     """Test for a failed login attempt."""
     assert login_page.title == "Login to Redash"
 
@@ -22,7 +26,7 @@ def test_login_wrong_user_credentials(login_page, unknown_user):
 
 
 @pytest.mark.nondestructive
-def test_login(login_page, user):
+def test_login(login_page: LoginPage, user: User) -> None:
     """Test for a successful login attempt."""
     assert login_page.title == "Login to Redash"