TTMK/web2.0-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix year issue in contact tests

Browse Source
This commit is contained in:
Aarni Halinen
2020-02-02 15:31:05 +02:00
parent e7164bb613
commit b790906b23
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
webapp/tests/test_contact.py
+9 -7
View File
@@ -62,29 +62,31 @@ def createPerson(name):
class ContactsTestCase(APITestCase):
current_year = timezone.now().year
old_year = 1970
def setUp(self):
createOccupation(timezone.now().year, role=createRoleBoard(), dummydata=1)
createOccupation(timezone.now().year, dummydata=2)
createOccupation(1970, role=createRoleBoard(), dummydata=3)
createOccupation(1970, dummydata=4)
createOccupation(self.current_year, role=createRoleBoard(), dummydata=1)
createOccupation(self.current_year, dummydata=2)
createOccupation(self.old_year, role=createRoleBoard(), dummydata=3)
createOccupation(self.old_year, dummydata=4)
def test_get(self):
response = self.client.get(f"{URL}", format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
serializer = OccupationSerializer(
Occupation.by_year(2020),
Occupation.by_year(self.current_year),
many=True
)
self.assertEqual(response.data["results"], serializer.data)
def test_get_by_year(self):
response = self.client.get(f"{URL}?year=1970", format='json')
response = self.client.get(f"{URL}?year={self.old_year}", format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
serializer = OccupationSerializer(
Occupation.by_year(1970),
Occupation.by_year(self.old_year),
many=True
)