def test_get_current_user(self):
""""Test for get_current_user from users router."""
url = URL(url='/users/me')
response = self.API_V1_TEST.get(
url.to_text(),
headers={'authorization': f'bearer {self.current_user_token}'}
)
assert response.status_code == OK
assert response_snapshot(0, response.json())
def test_get_reasons(self):
""""Test for get_reasons from dates_status router."""
url = URL('/dates_status/reasons')
response = self.API_V1_TEST.get(
url.to_text(),
headers={'authorization': f'bearer {self.current_user_token}'})
self.assertEqual(response.status_code, OK)
self.assertTrue(response_snapshot(0, response.json()))
def test_delete_user(self):
""""Test for delete_user from users router."""
url = URL(url=f'/users/{self.user.id}')
response = self.API_V1_TEST.delete(
url.to_text(),
headers={'authorization': f'bearer {self.current_user_token}'}
)
assert response.status_code == OK
# checks if the user was acctually deleted from the db.
assert get_user(db=self.session, user_id=self.user.id) is None
def test_get_dates_status(self):
""""Test for get_dates_status from dates_status router."""
query = Query(start=self.START_DATE,
end=self.END_DATE,
users_id=[self.user.id])
url = URL(url='/dates_status/', query=query)
response = self.API_V1_TEST.get(
url.to_text(),
headers={'authorization': f'bearer {self.current_user_token}'})
self.assertEqual(response.status_code, OK)
self.assertTrue(response_snapshot(0, response.json()))
def _get_faked_user_token(app_test: TestClient, username: str, password: str):
query = Query(
username=username,
password=password,
scopes="personal"
)
url = URL('/api/login')
return app_test.post(url.to_text(),
data=query.to_text(),
headers={
'content-type': 'application/x-www-form-urlencoded'
}
).json()['access_token']
def test_put_dates_status(self):
""""Test for put_dates_status from dates_status router."""
url = URL('/dates_status/')
response = self.API_V1_TEST.post(
url.to_text(),
headers={'authorization': f'bearer {self.current_user_token}'},
json={
'user_id': self.user.id,
'start_date': str(self.START_DATE),
'state': self.state_not_here.name,
'reason': self.reasons["1_abc"],
})
self.assertEqual(response.status_code, OK)
self.assertTrue(response_snapshot(0, response.json()))
def test_delete_dates_status(self):
""""Test for delete_dates_status from dates_status router."""
url = URL('/dates_status/')
response = self.API_V1_TEST.delete(
url.to_text(),
json={
"start_date": str(self.START_DATE),
"end_date": str(self.END_DATE),
"user_id": self.user.id
},
headers={'authorization': f'bearer {self.current_user_token}'})
assert response.status_code == OK
# assert that the data was acctually deleted from the db.
date_data = get_dates_data(db=self.session,
user_id=self.user.id,
start_date=self.START_DATE,
end_date=self.END_DATE)
self.assertEqual(date_data, [])