Skip to content
Snippets Groups Projects
Commit 73426726 authored by Carlos Vega's avatar Carlos Vega
Browse files

added test for get_worker_availability

parent 4ac0768d
No related branches found
No related tags found
1 merge request!171Feature/daily availability
# coding=utf-8 # coding=utf-8
import json import json
import datetime
from django.test import RequestFactory from django.test import RequestFactory
from django.urls import reverse from django.urls import reverse
...@@ -85,3 +85,32 @@ class TestWorkerApi(LoggedInWithWorkerTestCase): ...@@ -85,3 +85,32 @@ class TestWorkerApi(LoggedInWithWorkerTestCase):
for entry in entries: for entry in entries:
count += len(entry["workers"]) count += len(entry["workers"])
self.assertTrue(count > 0) self.assertTrue(count > 0)
def test_get_worker_availability(self):
today = datetime.datetime.today().replace(hour=8, minute=00, second=0, microsecond=0)
availability = Availability.objects.create(person=self.worker, day_number=today.isoweekday(),
available_from="8:00", available_till="16:00")
availability.save()
params={}
params['start_date'] = today.strftime("%Y-%m-%d-%H-%M")
params['end_date'] = (today+datetime.timedelta(hours=4)).strftime("%Y-%m-%d-%H-%M")
params['worker_id'] = self.worker.id
response = self.client.get(reverse('web.api.get_worker_availability'), data=params)
availability = json.loads(response.content)['availability']
self.assertEqual(availability, 100.0)
today = today.replace(hour=16, minute=1)
params={}
params['start_date'] = today.strftime("%Y-%m-%d-%H-%M")
params['end_date'] = (today+datetime.timedelta(hours=4)).strftime("%Y-%m-%d-%H-%M")
params['worker_id'] = self.worker.id
response = self.client.get(reverse('web.api.get_worker_availability'), data=params)
availability = json.loads(response.content)['availability']
self.assertEqual(availability, 0.0)
today = today.replace(hour=14, minute=0)
params={}
params['start_date'] = today.strftime("%Y-%m-%d-%H-%M")
params['end_date'] = (today+datetime.timedelta(hours=4)).strftime("%Y-%m-%d-%H-%M")
params['worker_id'] = self.worker.id
response = self.client.get(reverse('web.api.get_worker_availability'), data=params)
availability = json.loads(response.content)['availability']
self.assertEqual(availability, 50.0)
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment