def test_location_link(self):
"""
Test the connection of Task and Location
"""
location = mommy.make('tasking.location', name='Nairobi', country='KE')
mocked_target_object = mommy.make('tasking.Task')
now = timezone.now()
data = {
'name': 'Cow price',
'description': 'Some description',
'start': 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,
}
data_with_location = data.copy()
data_with_location['locations'] = [location.id]
serializer_instance = TaskSerializer(data=data_with_location)
self.assertTrue(serializer_instance.is_valid())
task = serializer_instance.save()
self.assertEqual(location, task.locations.get(id=location.id))
def test_task_parent_link(self):
"""
Test the connection between a parent and child task
"""
mocked_parent_task = mommy.make('tasking.Task', name='Cow Price')
mocked_target_object = mommy.make('tasking.Task')
now = timezone.now()
data = {
'name': 'Milk Production',
'description': 'Some description',
'start': 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,
'parent': mocked_parent_task.id
}
serializer_instance = TaskSerializer(data=data)
self.assertTrue(serializer_instance.is_valid())
task = serializer_instance.save()
self.assertEqual(mocked_parent_task, task.parent)
def test_task_parent_link(self):
"""
Test the connection between a parent and child task
"""
mocked_parent_task = mommy.make("tasking.Task", name="Cow Price")
mocked_target_object = mommy.make("tasking.Task")
now = timezone.now()
data = {
"name": "Milk Production",
"description": "Some description",
"start": 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,
"parent": mocked_parent_task.id,
}
serializer_instance = TaskSerializer(data=data)
self.assertTrue(serializer_instance.is_valid())
task = serializer_instance.save()
self.assertEqual(mocked_parent_task, task.parent)
def test_validate_timing_rule(self):
"""
Test that the serializer timing_rule validation works
"""
mocked_target_object = mommy.make('tasking.Task')
data = {
'name': 'Cow price',
'description': 'Some description',
'start': timezone.now(),
'total_submission_target': 10,
'timing_rule': 'inva;lid',
'target_content_type': self.task_type.id,
'target_id': mocked_target_object.id,
}
serializer_instance = TaskSerializer(data=data)
self.assertFalse(serializer_instance.is_valid())
def test_validate_timing_rule(self):
"""
Test that the serializer timing_rule validation works
"""
mocked_target_object = mommy.make("tasking.Task")
data = {
"name": "Cow price",
"description": "Some description",
"start": timezone.now(),
"total_submission_target": 10,
"timing_rule": "inva;lid",
"target_content_type": self.task_type.id,
"target_id": mocked_target_object.id,
}
serializer_instance = TaskSerializer(data=data)
self.assertFalse(serializer_instance.is_valid())
def test_location_link(self):
"""
Test the connection of Task and Location
"""
location = mommy.make("tasking.location", name="Nairobi", country="KE")
mocked_target_object = mommy.make("tasking.Task")
now = timezone.now()
data = {
"name": "Cow price",
"description": "Some description",
"start": 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,
}
data_with_location = data.copy()
locations_input = [{
"location": location.id,
"timing_rule": "RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5",
"start": "09:00:00",
"end": "15:00:00",
}]
data_with_location["locations_input"] = locations_input
serializer_instance = TaskSerializer(data=data_with_location)
self.assertTrue(serializer_instance.is_valid())
task = serializer_instance.save()
self.assertDictContainsSubset(
locations_input[0], serializer_instance.data["task_locations"][0])
self.assertEqual(location, task.locations.get(id=location.id))
def test_create_task(self):
"""
Test that the serializer can create Task objects
"""
mocked_target_object = mommy.make('tasking.Task')
rule1 = mommy.make('tasking.SegmentRule')
rule2 = mommy.make('tasking.SegmentRule')
rrule = 'DTSTART:20180521T210000Z RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5'
data = {
'name': 'Cow price',
'description': 'Some description',
'total_submission_target': 10,
'timing_rule': rrule,
'target_content_type': self.task_type.id,
'target_id': mocked_target_object.id,
'estimated_time': 'P4DT1H15M20S',
}
data_with_segment_rules = data.copy()
data_with_segment_rules['segment_rules'] = [rule1.id, rule2.id]
serializer_instance = TaskSerializer(data=data_with_segment_rules)
self.assertTrue(serializer_instance.is_valid())
task = serializer_instance.save()
# the start and end fields are going to be from the timing rule
start = get_rrule_start(rrulestr(rrule))
end = get_rrule_end(rrulestr(rrule))
# Change estimated time to DD HH:MM:SS format since Serializer
# Changes it to such
data['estimated_time'] = '4 01:15:20'
# 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, serializer_instance.data)
# we test that we do have our segment rules
self.assertEqual(set([rule1.id, rule2.id]),
set(serializer_instance.data['segment_rules']))
# we test that submissions are equal to 0
self.assertEqual(serializer_instance.data['submission_count'], 0)
self.assertEqual(task.submissions, 0)
# Add a submission to task and assert it changes.
mocked_submission = mommy.make('tasking.Submission', task=task)
self.assertTrue(mocked_submission.task, task)
self.assertEqual(task.submissions, 1)
self.assertEqual('Cow price', task.name)
self.assertEqual('Some description', task.description)
self.assertEqual(start, task.start)
self.assertEqual(end, task.end)
self.assertEqual(10, task.total_submission_target)
# assert that the ISO 8601 String was converted to accurately
self.assertEqual(task.estimated_time, timedelta(4, 4520))
# test that the segment rules for the task are as we expect
self.assertEqual(rule1, task.segment_rules.get(id=rule1.id))
self.assertEqual(rule2, task.segment_rules.get(id=rule2.id))
expected_fields = [
'id', 'created', 'modified', 'name', 'parent', 'description',
'start', 'end', 'timing_rule', 'estimated_time',
'total_submission_target', 'user_submission_target', 'status',
'submission_count', 'target_content_type', 'target_id',
'segment_rules', 'locations', 'task_locations'
]
self.assertEqual(set(expected_fields),
set(list(serializer_instance.data.keys())))
def test_location_link_update(self):
"""
Test the connection of Task and Location
"""
location = mommy.make('tasking.location', name='Nairobi', country='KE')
location2 = mommy.make('tasking.Location')
mocked_target_object = mommy.make('tasking.Task')
now = timezone.now()
data = {
'name': 'Cow price',
'description': 'Some description',
'start': 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,
}
data_with_location = data.copy()
locations_input = [{
'location': location.id,
'timing_rule': 'RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5',
'start': '09:00:00',
'end': '15:00:00'
}, {
'location': location2.id,
'timing_rule': 'RRULE:FREQ=DAILY;INTERVAL=10;COUNT=7',
'start': '12:00:00',
'end': '19:00:00'
}]
data_with_location['locations_input'] = locations_input
serializer_instance = TaskSerializer(data=data_with_location)
self.assertTrue(serializer_instance.is_valid())
task = serializer_instance.save()
self.assertEqual(2, TaskLocation.objects.filter(task=task).count())
# delete all locations
data2 = data_with_location.copy()
data2['locations_input'] = []
serializer_instance2 = TaskSerializer(instance=task, data=data2)
self.assertTrue(serializer_instance2.is_valid())
serializer_instance2.save()
self.assertEqual(0, TaskLocation.objects.filter(task=task).count())
# add new location
data3 = data_with_location.copy()
data3['locations_input'] = [{
'location': location.id,
'timing_rule': 'RRULE:FREQ=DAILY;INTERVAL=10;COUNT=7',
'start': '02:00:00',
'end': '09:00:00'
}]
serializer_instance3 = TaskSerializer(instance=task, data=data3)
self.assertTrue(serializer_instance3.is_valid())
serializer_instance3.save()
self.assertEqual(1, TaskLocation.objects.filter(task=task).count())
def test_create_task(self):
"""
Test that the serializer can create Task objects
"""
mocked_target_object = mommy.make("tasking.Task")
rule1 = mommy.make("tasking.SegmentRule")
rule2 = mommy.make("tasking.SegmentRule")
rrule = "DTSTART:20180521T210000Z RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5"
data = {
"name": "Cow price",
"description": "Some description",
"total_submission_target": 10,
"timing_rule": rrule,
"target_content_type": self.task_type.id,
"target_id": mocked_target_object.id,
"estimated_time": "P4DT1H15M20S",
}
data_with_segment_rules = data.copy()
data_with_segment_rules["segment_rules"] = [rule1.id, rule2.id]
serializer_instance = TaskSerializer(data=data_with_segment_rules)
self.assertTrue(serializer_instance.is_valid())
task = serializer_instance.save()
# the start and end fields are going to be from the timing rule
start = get_rrule_start(rrulestr(rrule))
end = get_rrule_end(rrulestr(rrule))
# Change estimated time to DD HH:MM:SS format since Serializer
# Changes it to such
data["estimated_time"] = "4 01:15:20"
# 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, serializer_instance.data)
# we test that we do have our segment rules
self.assertEqual(set([rule1.id, rule2.id]),
set(serializer_instance.data["segment_rules"]))
# we test that submissions are equal to 0
self.assertEqual(serializer_instance.data["submission_count"], 0)
self.assertEqual(task.submissions, 0)
# Add a submission to task and assert it changes.
mocked_submission = mommy.make("tasking.Submission", task=task)
self.assertTrue(mocked_submission.task, task)
self.assertEqual(task.submissions, 1)
self.assertEqual("Cow price", task.name)
self.assertEqual("Some description", task.description)
self.assertEqual(start, task.start)
self.assertEqual(end, task.end)
self.assertEqual(10, task.total_submission_target)
# assert that the ISO 8601 String was converted to accurately
self.assertEqual(task.estimated_time, timedelta(4, 4520))
# test that the segment rules for the task are as we expect
self.assertEqual(rule1, task.segment_rules.get(id=rule1.id))
self.assertEqual(rule2, task.segment_rules.get(id=rule2.id))
expected_fields = [
"id",
"created",
"modified",
"name",
"parent",
"description",
"start",
"end",
"timing_rule",
"estimated_time",
"total_submission_target",
"user_submission_target",
"status",
"submission_count",
"target_content_type",
"target_id",
"segment_rules",
"locations",
"task_locations",
]
self.assertEqual(set(expected_fields),
set(list(serializer_instance.data.keys())))
def test_location_link_update(self):
"""
Test the connection of Task and Location
"""
location = mommy.make("tasking.location", name="Nairobi", country="KE")
location2 = mommy.make("tasking.Location")
mocked_target_object = mommy.make("tasking.Task")
now = timezone.now()
data = {
"name": "Cow price",
"description": "Some description",
"start": 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,
}
data_with_location = data.copy()
locations_input = [
{
"location": location.id,
"timing_rule": "RRULE:FREQ=DAILY;INTERVAL=10;COUNT=5",
"start": "09:00:00",
"end": "15:00:00",
},
{
"location": location2.id,
"timing_rule": "RRULE:FREQ=DAILY;INTERVAL=10;COUNT=7",
"start": "12:00:00",
"end": "19:00:00",
},
]
data_with_location["locations_input"] = locations_input
serializer_instance = TaskSerializer(data=data_with_location)
self.assertTrue(serializer_instance.is_valid())
task = serializer_instance.save()
self.assertEqual(2, TaskLocation.objects.filter(task=task).count())
# delete all locations
data2 = data_with_location.copy()
data2["locations_input"] = []
serializer_instance2 = TaskSerializer(instance=task, data=data2)
self.assertTrue(serializer_instance2.is_valid())
serializer_instance2.save()
self.assertEqual(0, TaskLocation.objects.filter(task=task).count())
# add new location
data3 = data_with_location.copy()
data3["locations_input"] = [{
"location": location.id,
"timing_rule": "RRULE:FREQ=DAILY;INTERVAL=10;COUNT=7",
"start": "02:00:00",
"end": "09:00:00",
}]
serializer_instance3 = TaskSerializer(instance=task, data=data3)
self.assertTrue(serializer_instance3.is_valid())
serializer_instance3.save()
self.assertEqual(1, TaskLocation.objects.filter(task=task).count())