def test_create_with_bad_data(self):
"""
Test that we get appropriate errors when trying to create an object
with bad data
"""
bob_user = mommy.make("auth.User")
alice_user = mommy.make("auth.User")
mocked_target_object = mommy.make("tasking.Task")
# test bad target_id validation
bad_target_id = dict(
name="Cow price",
description="Some description",
start=timezone.now(),
total_submission_target=10,
timing_rule="RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5",
target_content_type=self.task_type.id,
target_id=1337,
)
view1 = TaskViewSet.as_view({"post": "create"})
request1 = self.factory.post("/tasks", bad_target_id)
# Need authenticated user
force_authenticate(request1, user=bob_user)
response1 = view1(request=request1)
self.assertEqual(response1.status_code, 400)
self.assertIn("target_id", response1.data.keys())
self.assertEqual(TARGET_DOES_NOT_EXIST,
str(response1.data["target_id"][0]))
# test bad content type validation
bad_content_type = dict(
name="Cow price",
description="Some description",
start=timezone.now(),
total_submission_target=10,
timing_rule="RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5",
target_content_type=999,
target_object_id=mocked_target_object.id,
)
view2 = TaskViewSet.as_view({"post": "create"})
request2 = self.factory.post("/tasks", bad_content_type)
# Need authenticated user
force_authenticate(request2, user=alice_user)
response2 = view2(request=request2)
self.assertEqual(response2.status_code, 400)
self.assertIn("target_content_type", response2.data.keys())
self.assertEqual(
'Invalid pk "999" - object does not exist.',
str(response2.data["target_content_type"][0]),
)
def test_status_filter(self):
"""
Test that you can filter by status
"""
user = mommy.make('auth.User')
mommy.make('tasking.Task', status=Task.DEACTIVATED, _quantity=7)
view = TaskViewSet.as_view({'get': 'list'})
# assert that there are no tasks with an Active Status
request = self.factory.get('/tasks', {'status': Task.ACTIVE})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 0)
self.assertEqual(Task.objects.filter(status=Task.ACTIVE).count(), 0)
# assert that there are 7 tasks with an Deactivated Status
request = self.factory.get('/tasks', {'status': Task.DEACTIVATED})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 7)
self.assertEqual(
Task.objects.filter(status=Task.DEACTIVATED).count(), 7)
# add a task with with an Active Status and assert that we get it back
mommy.make('tasking.Task', status=Task.ACTIVE)
request = self.factory.get('/tasks', {'status': Task.ACTIVE})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 1)
self.assertEqual(Task.objects.filter(status=Task.ACTIVE).count(), 1)
def test_search_filter_order(self):
"""
Test that you can search filter and order at the same time
"""
user = mommy.make('auth.User')
task = mommy.make(
'tasking.Task', name='Cattle Price', status=Task.ACTIVE)
task2 = mommy.make(
'tasking.Task', name='Cattle Price', status=Task.ACTIVE)
mommy.make('tasking.Task', name='Cattle Price', status=Task.DRAFT)
for _ in range(0, 4):
mommy.make(
'tasking.Task', name='Cattle Price', status=Task.DEACTIVATED)
view = TaskViewSet.as_view({'get': 'list'})
request = self.factory.get(
'/tasks?search={name}&status={status}&ordering={order}'.format(
name='Cattle Price', status=Task.ACTIVE, order='created'))
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(len(response.data), 2)
self.assertEqual(response.data[0]['status'], task.status)
self.assertEqual(response.data[0]['id'], task.id)
self.assertEqual(response.data[1]['status'], task2.status)
self.assertEqual(response.data[1]['id'], task2.id)
def test_end_time_filter(self):
"""
Test that you can filter by end_time
"""
user = mommy.make('auth.User')
view = TaskViewSet.as_view({'get': 'list'})
task = mommy.make('tasking.Task')
task2 = mommy.make('tasking.Task')
# make some occurrences that end at 5pm
mommy.make(
'tasking.TaskOccurrence', _quantity=5, task=task, end_time='17:00')
# make some tasks that end after 9pm
mommy.make(
'tasking.TaskOccurrence', _quantity=6, task=task2,
end_time='21:15')
# test that we can get tasks before or after a certain time
request2 = self.factory.get(
'/tasks', {'end_time__gte': '21:15'})
force_authenticate(request2, user=user)
response2 = view(request=request2)
self.assertEqual(response2.status_code, 200)
self.assertEqual(len(response2.data), 1)
self.assertEqual(response2.data[0]['id'], task2.id)
# check that we can get tasks before a certain time
request3 = self.factory.get(
'/tasks', {'end_time__lt': '21:15'})
force_authenticate(request3, user=user)
response3 = view(request=request3)
self.assertEqual(response3.status_code, 200)
self.assertEqual(len(response3.data), 1)
self.assertEqual(response3.data[0]['id'], task.id)
def test_search_filter_order(self):
"""
Test that you can search filter and order at the same time
"""
user = mommy.make("auth.User")
name = "Cattle Price"
task = mommy.make("tasking.Task", name=name, status=Task.ACTIVE)
task2 = mommy.make("tasking.Task", name=name, status=Task.ACTIVE)
mommy.make("tasking.Task", name=name, status=Task.DRAFT)
for _ in range(0, 4):
mommy.make("tasking.Task", name=name, status=Task.DEACTIVATED)
view = TaskViewSet.as_view({"get": "list"})
request = self.factory.get(
f"/tasks?search={name}&status={Task.ACTIVE}&ordering=created")
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(len(response.data), 2)
self.assertEqual(response.data[0]["status"], task.status)
self.assertEqual(response.data[0]["id"], task.id)
self.assertEqual(response.data[1]["status"], task2.status)
self.assertEqual(response.data[1]["id"], task2.id)
def test_update_task(self):
"""
Test UPDATE task
"""
user = mommy.make('auth.User')
user2 = mommy.make('auth.User')
task_data = self._create_task()
task_data2 = self._create_task()
data = {
'name': "Milk Price",
'target_content_type': self.user_type.id,
'target_id': user.id,
}
view = TaskViewSet.as_view({'patch': 'partial_update'})
request = self.factory.patch(
'/tasks/{id}'.format(id=task_data['id']), data=data)
force_authenticate(request, user=user)
response = view(request=request, pk=task_data['id'])
self.assertEqual(response.status_code, 200)
self.assertEqual('Milk Price', response.data['name'])
self.assertEqual(
self.user_type.id,
response.data['target_content_type'])
self.assertEqual(user.id, response.data['target_id'])
data2 = {
'name': "Cattle Price",
'description': 'Hello there!',
}
view2 = TaskViewSet.as_view({'patch': 'partial_update'})
request2 = self.factory.patch(
'/tasks/{id}'.format(id=task_data2['id']), data=data2)
force_authenticate(request2, user=user2)
response2 = view2(request=request2, pk=task_data2['id'])
self.assertEqual(response2.status_code, 200)
self.assertEqual('Cattle Price', response2.data['name'])
self.assertEqual(
'Hello there!',
response2.data['description'])
def test_update_task(self):
"""
Test UPDATE task
"""
user = mommy.make("auth.User")
user2 = mommy.make("auth.User")
task_data = self._create_task()
task_id = task_data["id"]
task_data2 = self._create_task()
task2_id = task_data2["id"]
data = {
"name": "Milk Price",
"target_content_type": self.user_type.id,
"target_id": user.id,
}
view = TaskViewSet.as_view({"patch": "partial_update"})
request = self.factory.patch(f"/tasks/{task_id}", data=data)
force_authenticate(request, user=user)
response = view(request=request, pk=task_id)
self.assertEqual(response.status_code, 200)
self.assertEqual("Milk Price", response.data["name"])
self.assertEqual(self.user_type.id,
response.data["target_content_type"])
self.assertEqual(user.id, response.data["target_id"])
data2 = {
"name": "Cattle Price",
"description": "Hello there!",
}
view2 = TaskViewSet.as_view({"patch": "partial_update"})
request2 = self.factory.patch(f"/tasks/{task2_id}", data=data2)
force_authenticate(request2, user=user2)
response2 = view2(request=request2, pk=task2_id)
self.assertEqual(response2.status_code, 200)
self.assertEqual("Cattle Price", response2.data["name"])
self.assertEqual("Hello there!", response2.data["description"])
def test_retrieve_task(self):
"""
Test GET /tasks/[pk] return a task matching pk.
"""
user = mommy.make('auth.User')
task_data = self._create_task()
view = TaskViewSet.as_view({'get': 'retrieve'})
request = self.factory.get('/tasks/{id}'.format(id=task_data['id']))
force_authenticate(request, user=user)
response = view(request=request, pk=task_data['id'])
self.assertEqual(response.status_code, 200)
self.assertDictEqual(response.data, task_data)
def test_list_tasks(self):
"""
Test GET /tasks listing of tasks for specific forms.
"""
user = mommy.make('auth.User')
task_data = self._create_task()
view = TaskViewSet.as_view({'get': 'list'})
request = self.factory.get('/tasks')
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertDictEqual(response.data.pop(), task_data)
def _create_task(self):
"""
Helper to create a single task
"""
mocked_target_object = mommy.make('auth.User')
rule1 = mommy.make('tasking.SegmentRule')
rule2 = mommy.make('tasking.SegmentRule')
user = mommy.make('auth.User')
data = {
'name': 'Cow price',
'description': 'Some description',
'total_submission_target': 10,
'timing_rule': 'RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5',
'target_content_type': self.user_type.id,
'target_id': mocked_target_object.id,
}
data_with_segment_rules = data.copy()
data_with_segment_rules['segment_rules'] = [rule1.id, rule2.id]
view = TaskViewSet.as_view({'post': 'create'})
request = self.factory.post('/tasks', data_with_segment_rules)
# Need authenticated user
force_authenticate(request, user=user)
response = view(request=request)
# we test that we do have our segment rules
self.assertEqual(set([rule1.id, rule2.id]),
set(response.data['segment_rules']))
self.assertEqual(response.status_code, 201, response.data)
# the order of segment_rules may have changed so a dict comparison
# may fail, we use `data` that does not include segment rules
self.assertDictContainsSubset(data, response.data)
# start and end were gotten from timing_rule
# lets check that they are correct by compating it to the start and
# end values of the Task object that was created
the_task = Task.objects.get(pk=response.data['id'])
# the start and end in the_task are UTC, we convert response.data to
# UTC so that we can compare
utc_start = parse(response.data['start']).astimezone(pytz.utc)
utc_end = parse(response.data['end']).astimezone(pytz.utc)
self.assertEqual(utc_start, the_task.start)
self.assertEqual(utc_end, the_task.end)
return response.data
def test_retrieve_task(self):
"""
Test GET /tasks/[pk] return a task matching pk.
"""
user = mommy.make("auth.User")
task_data = self._create_task()
task_id = task_data["id"]
view = TaskViewSet.as_view({"get": "retrieve"})
request = self.factory.get(f"/tasks/{task_id}")
force_authenticate(request, user=user)
response = view(request=request, pk=task_id)
self.assertEqual(response.status_code, 200)
self.assertDictEqual(response.data, task_data)
def test_date_filter(self):
"""
Test that you can filter by date
"""
user = mommy.make('auth.User')
task = mommy.make('tasking.Task')
task2 = mommy.make('tasking.Task')
# remove any autocreated task occurrences
# pylint: disable=no-member
task.taskoccurrence_set.all().delete()
task2.taskoccurrence_set.all().delete()
# make a bunch of occurrences
mommy.make(
'tasking.TaskOccurrence',
_quantity=7,
task=task,
date='2018-07-12')
# make one occurrence using a unique date
mommy.make('tasking.TaskOccurrence', task=task2, date='2017-09-09')
view = TaskViewSet.as_view({'get': 'list'})
# test that we get the task with our unique date
request = self.factory.get('/tasks',
{'date': '2017-09-09'})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 1)
self.assertEqual(response.data[0]['id'], task2.id)
# make some tasks that happen after 2018-07-12
mommy.make(
'tasking.TaskOccurrence',
_quantity=5,
task=task,
date='2018-11-11')
# test that we can get tasks before or after a certain date
request2 = self.factory.get('/tasks',
{'date__gt': '2018-07-13', 'xxx': 23})
force_authenticate(request2, user=user)
response2 = view(request=request2)
self.assertEqual(response2.status_code, 200)
self.assertEqual(len(response2.data), 1)
self.assertEqual(response2.data[0]['id'], task.id)
def test_name_search(self):
"""
Test that you can search by Name
"""
user = mommy.make("auth.User")
mommy.make("tasking.Task", name="Cattle Price")
mommy.make("tasking.Task", name="Chicken Price", _quantity=7)
view = TaskViewSet.as_view({"get": "list"})
request = self.factory.get("/tasks", {"search": "Cattle Price"})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 1)
self.assertEqual(Task.objects.filter(name="Cattle Price").count(), 1)
def test_name_search(self):
"""
Test that you can search by Name
"""
user = mommy.make('auth.User')
mommy.make('tasking.Task', name='Cattle Price')
mommy.make('tasking.Task', name='Chicken Price', _quantity=7)
view = TaskViewSet.as_view({'get': 'list'})
request = self.factory.get('/tasks', {'search': 'Cattle Price'})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 1)
self.assertEqual(Task.objects.filter(name='Cattle Price').count(), 1)
def test_date_filter(self):
"""
Test that you can filter by date
"""
user = mommy.make("auth.User")
task = mommy.make("tasking.Task")
task2 = mommy.make("tasking.Task")
# remove any autocreated task occurrences
# pylint: disable=no-member
task.taskoccurrence_set.all().delete()
task2.taskoccurrence_set.all().delete()
# make a bunch of occurrences
mommy.make("tasking.TaskOccurrence",
_quantity=7,
task=task,
date="2018-07-12")
# make one occurrence using a unique date
mommy.make("tasking.TaskOccurrence", task=task2, date="2017-09-09")
view = TaskViewSet.as_view({"get": "list"})
# test that we get the task with our unique date
request = self.factory.get("/tasks", {"date__exact": "2017-09-09"})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 1)
self.assertEqual(response.data[0]["id"], task2.id)
# make some tasks that happen after 2018-07-12
mommy.make("tasking.TaskOccurrence",
_quantity=5,
task=task,
date="2018-11-11")
# test that we can get tasks before or after a certain date
request2 = self.factory.get("/tasks", {
"date__gt": "2018-07-13",
"xxx": 23
})
force_authenticate(request2, user=user)
response2 = view(request=request2)
self.assertEqual(response2.status_code, 200)
self.assertEqual(len(response2.data), 1)
self.assertEqual(response2.data[0]["id"], task.id)
def test_task_locations(self):
"""
Test that we can create a task and add task locations
"""
user = mommy.make("auth.User")
mocked_target_object = mommy.make("auth.User")
location = mommy.make("tasking.Location")
data = {
"name":
"Coconut Quest",
"description":
"Mission impossible!",
"total_submission_target":
1,
"timing_rule":
"RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5",
"target_content_type":
self.user_type.id,
"target_id":
mocked_target_object.id,
"locations_input": [{
"location": location.id,
"timing_rule": "RRULE:FREQ=DAILY;INTERVAL=10;COUNT=7",
"start": "02:00:00",
"end": "09:00:00",
}],
}
view = TaskViewSet.as_view({"post": "create"})
# we need to give the inpiut as JSON so that locations_input is
# properly set
request = self.factory.post("/tasks",
json.dumps(data),
content_type="application/json")
# Need authenticated user
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 201, response.data)
self.assertEqual(1, len(response.data["task_locations"]))
task_location = TaskLocation.objects.get(task__id=response.data["id"],
location=location)
self.assertEqual("RRULE:FREQ=DAILY;INTERVAL=10;COUNT=7",
task_location.timing_rule)
self.assertEqual("02:00:00", task_location.start.isoformat())
self.assertEqual("09:00:00", task_location.end.isoformat())
def test_location_filter(self):
"""
Test that you can filter by location
"""
user = mommy.make('auth.User')
nairobi = mommy.make('tasking.Location', name='Nairobi')
arusha = mommy.make('tasking.Location', name='Arusha')
for _ in range(0, 7):
task = mommy.make('tasking.Task')
mommy.make(
'tasking.TaskLocation',
task=task,
location=nairobi
)
view = TaskViewSet.as_view({'get': 'list'})
# assert that there are no tasks for Arusha
request = self.factory.get('/tasks', {'locations': arusha.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 0)
self.assertEqual(Task.objects.filter(locations=arusha).count(), 0)
# assert that there are 7 tasks for Nairobi
request = self.factory.get('/tasks', {'locations': nairobi.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 7)
self.assertEqual(Task.objects.filter(locations=nairobi).count(), 7)
# add one Arusha task and assert that we get it back
task2 = mommy.make('tasking.Task')
mommy.make(
'tasking.TaskLocation',
task=task2,
location=arusha
)
request = self.factory.get('/tasks', {'locations': arusha.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 1)
self.assertEqual(Task.objects.filter(locations=arusha).count(), 1)
def test_delete_task(self):
"""
Test DELETE tasks.
"""
user = mommy.make("auth.User")
task = mommy.make("tasking.Task")
# assert that task exists
self.assertTrue(Task.objects.filter(pk=task.id).exists())
# delete task
view = TaskViewSet.as_view({"delete": "destroy"})
request = self.factory.delete(f"/tasks/{task.id}")
force_authenticate(request, user=user)
response = view(request=request, pk=task.id)
# assert that task was deleted
self.assertEqual(response.status_code, 204)
self.assertFalse(Task.objects.filter(pk=task.id).exists())
def test_delete_task(self):
"""
Test DELETE tasks.
"""
user = mommy.make('auth.User')
task = mommy.make('tasking.Task')
# assert that task exists
self.assertTrue(Task.objects.filter(pk=task.id).exists())
# delete task
view = TaskViewSet.as_view({'delete': 'destroy'})
request = self.factory.delete('/tasks/{id}'.format(id=task.id))
force_authenticate(request, user=user)
response = view(request=request, pk=task.id)
# assert that task was deleted
self.assertEqual(response.status_code, 204)
self.assertFalse(Task.objects.filter(pk=task.id).exists())
def test_task_locations(self):
"""
Test that we can create a task and add task locations
"""
user = mommy.make('auth.User')
mocked_target_object = mommy.make('auth.User')
location = mommy.make('tasking.Location')
data = {
'name': 'Coconut Quest',
'description': 'Mission impossible!',
'total_submission_target': 1,
'timing_rule': 'RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5',
'target_content_type': self.user_type.id,
'target_id': mocked_target_object.id,
'locations_input': [
{
'location': location.id,
'timing_rule': 'RRULE:FREQ=DAILY;INTERVAL=10;COUNT=7',
'start': '02:00:00',
'end': '09:00:00'
}
]
}
view = TaskViewSet.as_view({'post': 'create'})
# we need to give the inpiut as JSON so that locations_input is
# properly set
request = self.factory.post(
'/tasks', json.dumps(data), content_type='application/json')
# Need authenticated user
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 201, response.data)
self.assertEqual(1, len(response.data['task_locations']))
task_location = TaskLocation.objects.get(
task__id=response.data['id'],
location=location)
self.assertEqual('RRULE:FREQ=DAILY;INTERVAL=10;COUNT=7',
task_location.timing_rule)
self.assertEqual('02:00:00', task_location.start.isoformat())
self.assertEqual('09:00:00', task_location.end.isoformat())
def test_start_time_filter(self):
"""
Test that you can filter by start_time
"""
user = mommy.make('auth.User')
view = TaskViewSet.as_view({'get': 'list'})
task = mommy.make('tasking.Task')
task2 = mommy.make('tasking.Task')
# remove any autocreated tasks
# pylint: disable=no-member
task.taskoccurrence_set.all().delete()
task2.taskoccurrence_set.all().delete()
# make some occurrences that start at 7
mommy.make(
'tasking.TaskOccurrence', _quantity=5, task=task,
start_time='07:00')
# make some occurrences that happen after 9:00
mommy.make(
'tasking.TaskOccurrence', _quantity=6, task=task2,
start_time='09:15')
# test that we can get tasks before or after a certain time
request2 = self.factory.get(
'/tasks', {'start_time__gte': '09:15'})
force_authenticate(request2, user=user)
response2 = view(request=request2)
self.assertEqual(response2.status_code, 200)
self.assertEqual(len(response2.data), 1)
self.assertEqual(response2.data[0]['id'], task2.id)
# check that we can get tasks before a certain time
request3 = self.factory.get(
'/tasks', {'start_time__lt': '09:15'})
force_authenticate(request3, user=user)
response3 = view(request=request3)
self.assertEqual(response3.status_code, 200)
self.assertEqual(len(response3.data), 1)
self.assertEqual(response3.data[0]['id'], task.id)
def test_start_time_filter(self):
"""
Test that you can filter by start_time
"""
user = mommy.make("auth.User")
view = TaskViewSet.as_view({"get": "list"})
task = mommy.make("tasking.Task")
task2 = mommy.make("tasking.Task")
# remove any autocreated tasks
# pylint: disable=no-member
task.taskoccurrence_set.all().delete()
task2.taskoccurrence_set.all().delete()
# make some occurrences that start at 7
mommy.make("tasking.TaskOccurrence",
_quantity=5,
task=task,
start_time="07:00")
# make some occurrences that happen after 9:00
mommy.make("tasking.TaskOccurrence",
_quantity=6,
task=task2,
start_time="09:15")
# test that we can get tasks before or after a certain time
request2 = self.factory.get("/tasks", {"start_time__gte": "09:15"})
force_authenticate(request2, user=user)
response2 = view(request=request2)
self.assertEqual(response2.status_code, 200)
self.assertEqual(len(response2.data), 1)
self.assertEqual(response2.data[0]["id"], task2.id)
# check that we can get tasks before a certain time
request3 = self.factory.get("/tasks", {"start_time__lt": "09:15"})
force_authenticate(request3, user=user)
response3 = view(request=request3)
self.assertEqual(response3.status_code, 200)
self.assertEqual(len(response3.data), 1)
self.assertEqual(response3.data[0]["id"], task.id)
def test_project_filter(self):
"""
Test that you can filter by Project
"""
user = mommy.make('auth.User')
project1 = mommy.make('tasking.Project', name='Test Case Scenario')
project2 = mommy.make('tasking.Project', name='Reality Check')
for _ in range(0, 7):
task = mommy.make('tasking.Task')
project1.tasks.add(task)
view = TaskViewSet.as_view({'get': 'list'})
# assert that there are no tasks in project Reality Check
request = self.factory.get('/tasks', {'project': project2.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 0)
self.assertEqual(
Task.objects.filter(project=project2.id).count(), 0)
# assert that there are 7 tasks in project Test Case Scenario
request = self.factory.get('/tasks', {'project': project1.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 7)
self.assertEqual(
Task.objects.filter(project=project1.id).count(), 7)
# add a task to project Reality Check and assert its there
task2 = mommy.make('tasking.Task')
project2.tasks.add(task2)
request = self.factory.get('/tasks', {'project': project2.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 1)
self.assertEqual(
Task.objects.filter(project=project2.id).count(), 1)
def test_location_filter(self):
"""
Test that you can filter by location
"""
user = mommy.make("auth.User")
nairobi = mommy.make("tasking.Location", name="Nairobi")
arusha = mommy.make("tasking.Location", name="Arusha")
for _ in range(0, 7):
task = mommy.make("tasking.Task")
mommy.make("tasking.TaskLocation", task=task, location=nairobi)
view = TaskViewSet.as_view({"get": "list"})
# assert that there are no tasks for Arusha
request = self.factory.get("/tasks", {"locations": arusha.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 0)
self.assertEqual(Task.objects.filter(locations=arusha).count(), 0)
# assert that there are 7 tasks for Nairobi
request = self.factory.get("/tasks", {"locations": nairobi.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 7)
self.assertEqual(Task.objects.filter(locations=nairobi).count(), 7)
# add one Arusha task and assert that we get it back
task2 = mommy.make("tasking.Task")
mommy.make("tasking.TaskLocation", task=task2, location=arusha)
request = self.factory.get("/tasks", {"locations": arusha.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 1)
self.assertEqual(Task.objects.filter(locations=arusha).count(), 1)
def test_parent_filter(self):
"""
Test that you can filter by parent
"""
user = mommy.make('auth.User')
parent1 = mommy.make('tasking.Task')
parent2 = mommy.make('tasking.Task')
mommy.make('tasking.Task', parent=parent1, _quantity=7)
view = TaskViewSet.as_view({'get': 'list'})
# assert that there are no tasks whose parent is parent2
request = self.factory.get('/tasks', {'parent': parent2.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 0)
self.assertEqual(Task.objects.filter(parent=parent2.id).count(), 0)
# assert that there are 7 tasks whose parent is parent1
request = self.factory.get('/tasks', {'parent': parent1.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 7)
self.assertEqual(Task.objects.filter(parent=parent1.id).count(), 7)
# create a task whose parent is parent2 and assert its there
mommy.make('tasking.Task', parent=parent2)
request = self.factory.get('/tasks', {'parent': parent2.id})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 1)
self.assertEqual(Task.objects.filter(parent=parent2.id).count(), 1)
def test_end_time_filter(self):
"""
Test that you can filter by end_time
"""
user = mommy.make("auth.User")
view = TaskViewSet.as_view({"get": "list"})
task = mommy.make("tasking.Task")
task2 = mommy.make("tasking.Task")
# make some occurrences that end at 5pm
mommy.make("tasking.TaskOccurrence",
_quantity=5,
task=task,
end_time="17:00")
# make some tasks that end after 9pm
mommy.make("tasking.TaskOccurrence",
_quantity=6,
task=task2,
end_time="21:15")
# test that we can get tasks before or after a certain time
request2 = self.factory.get("/tasks", {"end_time__gte": "21:15"})
force_authenticate(request2, user=user)
response2 = view(request=request2)
self.assertEqual(response2.status_code, 200)
self.assertEqual(len(response2.data), 1)
self.assertEqual(response2.data[0]["id"], task2.id)
# check that we can get tasks before a certain time
request3 = self.factory.get("/tasks", {"end_time__lt": "21:15"})
force_authenticate(request3, user=user)
response3 = view(request=request3)
self.assertEqual(response3.status_code, 200)
self.assertEqual(len(response3.data), 1)
self.assertEqual(response3.data[0]["id"], task.id)
def test_task_sorting(self):
"""
Test that sorting works
"""
user = mommy.make("auth.User")
project1 = mommy.make("tasking.Project")
project2 = mommy.make("tasking.Project")
task1 = mommy.make(
"tasking.Task",
name="Milk Production Size",
status=Task.DRAFT,
estimated_time=timedelta(4, 4520),
)
project1.tasks.add(task1)
for _ in range(0, 7):
# create other tasks
task = mommy.make(
"tasking.Task",
name="Cow Price",
status=Task.DEACTIVATED,
estimated_time=timedelta(3, 3250),
)
mommy.make("tasking.Submission", task=task, _quantity=3)
project1.tasks.add(task)
task2 = mommy.make(
"tasking.Task",
name="Allocated land for farming",
status=Task.ACTIVE,
estimated_time=timedelta(2, 4520),
)
project2.tasks.add(task2)
# Create and add Submissions to Task1 and Task2
mommy.make("tasking.Submission", task=task1, _quantity=4)
mommy.make("tasking.Submission", task=task2, _quantity=1)
view = TaskViewSet.as_view({"get": "list"})
# order by status descending
request = self.factory.get("/tasks", {"ordering": "-status"})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.data[0]["id"], task1.id)
self.assertEqual(response.data[0]["status"], task1.status)
self.assertEqual(response.data[-1]["id"], task2.id)
self.assertEqual(response.data[-1]["status"], task2.status)
# order by created ascending
request = self.factory.get("/tasks", {"ordering": "created"})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(
parse(response.data[0]["created"]).astimezone(pytz.utc),
task1.created)
self.assertEqual(response.data[0]["id"], task1.id)
self.assertEqual(
parse(response.data[-1]["created"]).astimezone(pytz.utc),
task2.created)
self.assertEqual(response.data[-1]["id"], task2.id)
# order by name ascending
request = self.factory.get("/tasks", {"ordering": "name"})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.data[-1]["name"], task1.name)
self.assertEqual(response.data[-1]["id"], task1.id)
self.assertEqual(response.data[0]["name"], task2.name)
self.assertEqual(response.data[0]["id"], task2.id)
# order by project ascending
request = self.factory.get("/tasks", {"ordering": "project__id"})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.data[0]["name"], task1.name)
self.assertEqual(response.data[0]["id"], task1.id)
self.assertEqual(response.data[-1]["name"], task2.name)
self.assertEqual(response.data[-1]["id"], task2.id)
# order by submissions descending
request = self.factory.get("/tasks", {"ordering": "-submission_count"})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.data[0]["name"], task1.name)
self.assertEqual(response.data[0]["id"], task1.id)
self.assertEqual(response.data[-1]["name"], task2.name)
self.assertEqual(response.data[-1]["id"], task2.id)
self.assertTrue(task1.submissions > task2.submissions)
# order by Estimated Time descending
request = self.factory.get("/tasks", {"ordering": "-estimated_time"})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.data[0]["name"], task1.name)
self.assertEqual(response.data[0]["id"], task1.id)
self.assertEqual(response.data[-1]["name"], task2.name)
self.assertEqual(response.data[-1]["id"], task2.id)
self.assertTrue(task1.estimated_time > task2.estimated_time)
def test_authentication_required(self):
"""
Test that authentication is required for all viewset actions
"""
mocked_target_object = mommy.make("tasking.Task")
task_data = self._create_task()
task_id = task_data["id"]
task = mommy.make("tasking.Task")
user = mommy.make("auth.User")
# test that you need authentication for creating a task
good_data = {
"name": "Cow price",
"description": "Some description",
"start": timezone.now(),
"total_submission_target": 10,
"timing_rule": "RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5",
"target_content_type": self.task_type.id,
"target_id": mocked_target_object.id,
}
view = TaskViewSet.as_view({"post": "create"})
request = self.factory.post("/tasks", good_data)
response = view(request=request)
self.assertEqual(response.status_code, 403)
self.assertEqual(
"Authentication credentials were not provided.",
str(response.data["detail"]),
)
# test that you need authentication for retrieving a task
view2 = TaskViewSet.as_view({"get": "retrieve"})
request2 = self.factory.get(f"/tasks/{task_id}")
response2 = view2(request=request2, pk=task_id)
self.assertEqual(response2.status_code, 403)
self.assertEqual(
"Authentication credentials were not provided.",
str(response2.data["detail"]),
)
# test that you need authentication for listing a task
view3 = TaskViewSet.as_view({"get": "list"})
request3 = self.factory.get("/tasks")
response3 = view3(request=request3)
self.assertEqual(response3.status_code, 403)
self.assertEqual(
"Authentication credentials were not provided.",
str(response3.data["detail"]),
)
# test that you need authentication for deleting a task
self.assertTrue(Task.objects.filter(pk=task.id).exists())
view4 = TaskViewSet.as_view({"delete": "destroy"})
request4 = self.factory.delete(f"/tasks/{task.id}")
response4 = view4(request=request4, pk=task.id)
self.assertEqual(response4.status_code, 403)
self.assertEqual(
"Authentication credentials were not provided.",
str(response4.data["detail"]),
)
# test that you need authentication for updating a task
data = {
"name": "Milk Price",
"target_content_type": self.user_type.id,
"target_id": user.id,
}
view5 = TaskViewSet.as_view({"patch": "partial_update"})
request5 = self.factory.patch(f"/tasks/{task_id}", data=data)
response5 = view5(request=request5, pk=task_id)
self.assertEqual(response5.status_code, 403)
self.assertEqual(
"Authentication credentials were not provided.",
str(response5.data["detail"]),
)
def test_task_sorting(self):
"""
Test that sorting works
"""
user = mommy.make('auth.User')
project1 = mommy.make('tasking.Project')
project2 = mommy.make('tasking.Project')
task1 = mommy.make(
'tasking.Task',
name='Milk Production Size',
status=Task.DRAFT,
estimated_time=timedelta(4, 4520))
project1.tasks.add(task1)
for _ in range(0, 7):
# create other tasks
task = mommy.make(
'tasking.Task',
name='Cow Price',
status=Task.DEACTIVATED,
estimated_time=timedelta(3, 3250)
)
mommy.make('tasking.Submission', task=task, _quantity=3)
project1.tasks.add(task)
task2 = mommy.make(
'tasking.Task',
name='Allocated land for farming',
status=Task.ACTIVE,
estimated_time=timedelta(2, 4520))
project2.tasks.add(task2)
# Create and add Submissions to Task1 and Task2
mommy.make('tasking.Submission', task=task1, _quantity=4)
mommy.make('tasking.Submission', task=task2, _quantity=1)
view = TaskViewSet.as_view({'get': 'list'})
# order by status descending
request = self.factory.get('/tasks', {'ordering': '-status'})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(response.data[0]['id'], task1.id)
self.assertEqual(response.data[0]['status'], task1.status)
self.assertEqual(response.data[-1]['id'], task2.id)
self.assertEqual(response.data[-1]['status'], task2.status)
# order by created ascending
request = self.factory.get('/tasks', {'ordering': 'created'})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(
parse(response.data[0]['created']).astimezone(pytz.utc),
task1.created)
self.assertEqual(response.data[0]['id'], task1.id)
self.assertEqual(
parse(response.data[-1]['created']).astimezone(pytz.utc),
task2.created)
self.assertEqual(response.data[-1]['id'], task2.id)
# order by name ascending
request = self.factory.get('/tasks', {'ordering': 'name'})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(
response.data[-1]['name'], task1.name)
self.assertEqual(response.data[-1]['id'], task1.id)
self.assertEqual(
response.data[0]['name'], task2.name)
self.assertEqual(response.data[0]['id'], task2.id)
# order by project ascending
request = self.factory.get('/tasks', {'ordering': 'project__id'})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(
response.data[0]['name'], task1.name)
self.assertEqual(response.data[0]['id'], task1.id)
self.assertEqual(
response.data[-1]['name'], task2.name)
self.assertEqual(response.data[-1]['id'], task2.id)
# order by submissions descending
request = self.factory.get('/tasks', {'ordering': '-submission_count'})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(
response.data[0]['name'], task1.name)
self.assertEqual(response.data[0]['id'], task1.id)
self.assertEqual(
response.data[-1]['name'], task2.name)
self.assertEqual(response.data[-1]['id'], task2.id)
self.assertTrue(task1.submissions > task2.submissions)
# order by Estimated Time descending
request = self.factory.get('/tasks', {'ordering': '-estimated_time'})
force_authenticate(request, user=user)
response = view(request=request)
self.assertEqual(
response.data[0]['name'], task1.name)
self.assertEqual(response.data[0]['id'], task1.id)
self.assertEqual(
response.data[-1]['name'], task2.name)
self.assertEqual(response.data[-1]['id'], task2.id)
self.assertTrue(task1.estimated_time > task2.estimated_time)
def test_authentication_required(self):
"""
Test that authentication is required for all viewset actions
"""
mocked_target_object = mommy.make('tasking.Task')
task_data = self._create_task()
task = mommy.make('tasking.Task')
user = mommy.make('auth.User')
# test that you need authentication for creating a task
good_data = {
'name': 'Cow price',
'description': 'Some description',
'start': timezone.now(),
'total_submission_target': 10,
'timing_rule': 'RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5',
'target_content_type': self.task_type.id,
'target_id': mocked_target_object.id,
}
view = TaskViewSet.as_view({'post': 'create'})
request = self.factory.post('/tasks', good_data)
response = view(request=request)
self.assertEqual(response.status_code, 403)
self.assertEqual(
'Authentication credentials were not provided.',
six.text_type(response.data['detail']))
# test that you need authentication for retrieving a task
view2 = TaskViewSet.as_view({'get': 'retrieve'})
request2 = self.factory.get('/tasks/{id}'.format(id=task_data['id']))
response2 = view2(request=request2, pk=task_data['id'])
self.assertEqual(response2.status_code, 403)
self.assertEqual(
'Authentication credentials were not provided.',
six.text_type(response2.data['detail']))
# test that you need authentication for listing a task
view3 = TaskViewSet.as_view({'get': 'list'})
request3 = self.factory.get('/tasks')
response3 = view3(request=request3)
self.assertEqual(response3.status_code, 403)
self.assertEqual(
'Authentication credentials were not provided.',
six.text_type(response3.data['detail']))
# test that you need authentication for deleting a task
self.assertTrue(Task.objects.filter(pk=task.id).exists())
view4 = TaskViewSet.as_view({'delete': 'destroy'})
request4 = self.factory.delete('/tasks/{id}'.format(id=task.id))
response4 = view4(request=request4, pk=task.id)
self.assertEqual(response4.status_code, 403)
self.assertEqual(
'Authentication credentials were not provided.',
six.text_type(response4.data['detail']))
# test that you need authentication for updating a task
data = {
'name': "Milk Price",
'target_content_type': self.user_type.id,
'target_id': user.id,
}
view5 = TaskViewSet.as_view({'patch': 'partial_update'})
request5 = self.factory.patch(
'/tasks/{id}'.format(id=task_data['id']), data=data)
response5 = view5(request=request5, pk=task_data['id'])
self.assertEqual(response5.status_code, 403)
self.assertEqual(
'Authentication credentials were not provided.',
six.text_type(response5.data['detail']))