def test_fetch_user_valid_token_login(self):
username = "******"
headers = generate_auth_header(username, "test1")
response = self.app.get(
"/api/v1.0/generate-token",
follow_redirects=True,
headers=headers
)
self.assertEqual(response.status_code, 200)
data = json.loads(response.get_data(as_text=True))
headers = generate_auth_header(data['token'], "")
response = self.app.get(
"/api/v1.0/users/1", follow_redirects=True, headers=headers
)
data = json.loads(response.get_data(as_text=True))
self.assertEqual(username, data['user']['username'])
def test_update_user_non_string_username(self):
json = {"username": 1}
headers = generate_auth_header("user-1", "test1")
response = self.app.put(
"/api/v1.0/users/1", follow_redirects=True, headers=headers, json=json
)
self.assertEqual(response.status_code, 400)
def test_update_user_invalid_login(self):
json = {"username": "******", "password": "******"}
headers = generate_auth_header("user-1", "invalid")
response = self.app.put(
"/api/v1.0/users/1", follow_redirects=True, headers=headers, json=json
)
self.assertEqual(response.status_code, 401)
def test_fetch_user_token_login_invalid_token(self):
username = "******"
headers = generate_auth_header("INVALID_TOKEN", "")
response = self.app.get(
"/api/v1.0/users/1", follow_redirects=True, headers=headers
)
data = json.loads(response.get_data(as_text=True))
self.assertEqual(response.status_code, 401)
def test_fetch_user_invalid_token_login(self):
username = "******"
headers = generate_auth_header(username, "invalid")
response = self.app.get(
"/api/v1.0/generate-token",
follow_redirects=True,
headers=headers
)
self.assertEqual(response.status_code, 401)
def test_fetch_user_valid_login(self):
username = "******"
headers = generate_auth_header(username, "test1")
response = self.app.get(
"/api/v1.0/users/1", follow_redirects=True, headers=headers
)
self.assertEqual(response.status_code, 200)
u = User.query.filter_by(username=username).first()
self.assertEqual(username, u.username)
def test_update_user_valid(self):
username = "******"
json = {"username": username, "password": "******"}
headers = generate_auth_header("user-1", "test1")
response = self.app.put(
"/api/v1.0/users/1", follow_redirects=True, headers=headers, json=json
)
self.assertEqual(response.status_code, 200)
u = User.query.filter_by(username=username).first()
self.assertEqual(username, u.username)
def test_fetch_user_invalid_user(self):
headers = generate_auth_header("user-1", "test1")
response = self.app.get(
"/api/v1.0/users/0", follow_redirects=True, headers=headers
)
self.assertEqual(response.status_code, 404)
def test_delete_user_invalid_user_id(self):
headers = generate_auth_header("user-1", "test1")
response = self.app.delete(
"/api/v1.0/users/2", follow_redirects=True, headers=headers
)
self.assertEqual(response.status_code, 401)