def test_registered_user_login_by_username(self):
""" Test for login of registered user by username """
response = register_user(self, "user", "*****@*****.**", "pass123")
self.assertEqual(response.json['message'], 'Successfully registered')
self.assertEqual(response.status_code, 201)
response = login_user(self, 'user', 'pass123')
self.assertEqual(response.json['message'], 'Successfully logged in')
self.assertTrue(response.json['token'])
self.assertEqual(response.status_code, 200)
def test_registered_with_already_registered_user(self) -> None:
"""Test registration with already registered username or email"""
user = User(username="******",
email="*****@*****.**",
password="******")
db.session.add(user)
db.session.commit()
response = register_user(self, "user", "*****@*****.**", "pass321")
self.assertEqual(response.status_code, 409)
self.assertEqual(response.json["message"], "User with this username "
"or email already exists")
def test_logout_logged_in_user(self):
"""Test user logout if user logged-in"""
response = register_user(self, "user", "*****@*****.**", "pass123")
self.assertEqual(response.status_code, 201)
self.assertEqual(response.json['message'], 'Successfully registered')
response = login_user(self, '*****@*****.**', 'pass123')
self.assertEqual(response.json['message'], 'Successfully logged in')
self.assertEqual(response.status_code, 200)
token = response.json["token"]
response = self.client.get(
"/auth/logout", headers={"Authorization": f"Bearer {token}"})
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json["message"], "Successfully logged out")
def test_login_without_auth(self):
""" Test for login of user without 'Authorization' header"""
response = register_user(self, "user", "*****@*****.**", "pass123")
self.assertEqual(response.json['message'], 'Successfully registered')
self.assertEqual(response.status_code, 201)
response = self.client.get(
"/auth/login",
data=json.dumps({
"username": '******',
"password": '******'
}),
headers={"Content-Type": "application/json"})
self.assertEqual(response.headers['WWW-Authenticate'],
"Basic realm='Authentication required'")
self.assertEqual(response.json['message'], "Authentication required")
self.assertEqual(response.status_code, 401)
def test_registration_wrong_data(self) -> None:
"""Test for user registration if wrong data"""
response = register_user(self, "user", "*****@*****.**", "pass123")
self.assertEqual(response.json["message"], "Successfully registered")
self.assertEqual(response.status_code, 201)