def test_should_cancel_scheduled_flow_for_a_consignee_before_scheduling_another_one_for_that_node_line_item(
self):
node = DistributionPlanNodeFactory(consignee=self.consignee)
line_item_one = DistributionPlanLineItemFactory(
distribution_plan_node=node,
planned_distribution_date=datetime.datetime.now())
DistributionPlanLineItemFactory(
distribution_plan_node=node,
planned_distribution_date=datetime.datetime.now())
when(DistributionPlanNode.objects).get(id=node.id).thenReturn(node)
when(DistributionPlanLineItem.objects).get(
id=line_item_one.id).thenReturn(line_item_one)
when(node.consignee).build_contact().thenReturn(self.contact)
schedule_run_for(line_item_one)
task_id = line_item_one.current_node_line_item_run(
).scheduled_message_task_id
when(celery.app.control).revoke(task_id).thenReturn(None)
schedule_run_for(line_item_one)
verify(celery.app.control).revoke(task_id)
verify(fake_facade,
times=2).start_delivery_flow(sender=any(),
consignee=any(),
item_description=any())
def test_should_get_node_line_item_run_with_status_in_progress(self):
node_line_item = DistributionPlanLineItemFactory()
line_item_run = NodeLineItemRunFactory(
node_line_item=node_line_item,
status=NodeLineItemRun.STATUS.in_progress)
self.assertEqual(node_line_item.current_node_line_item_run(),
line_item_run)
def xtest_should_change_status_of_node_run_to_in_progress_when_scheduled_task_is_started(self):
node = DistributionPlanNodeFactory(consignee=self.consignee)
line_item = DistributionPlanLineItemFactory(distribution_plan_node=node,
planned_distribution_date=datetime.datetime.now())
when(DistributionPlanNode.objects).get(id=node.id).thenReturn(node)
when(DistributionPlanLineItem.objects).get(id=line_item.id).thenReturn(line_item)
when(node.consignee).build_contact().thenReturn(self.contact)
schedule_run_for(line_item)
self.assertEqual(line_item.current_node_line_item_run().status, 'in_progress')
def xtest_should_change_status_of_node_run_to_in_progress_when_scheduled_task_is_started(
self):
node = DistributionPlanNodeFactory(consignee=self.consignee)
line_item = DistributionPlanLineItemFactory(
distribution_plan_node=node,
planned_distribution_date=datetime.datetime.now())
when(DistributionPlanNode.objects).get(id=node.id).thenReturn(node)
when(DistributionPlanLineItem.objects).get(
id=line_item.id).thenReturn(line_item)
when(node.consignee).build_contact().thenReturn(self.contact)
schedule_run_for(line_item)
self.assertEqual(line_item.current_node_line_item_run().status,
'in_progress')
def test_should_cancel_scheduled_flow_for_a_consignee_before_scheduling_another_one_for_that_node_line_item(self):
node = DistributionPlanNodeFactory(consignee=self.consignee)
line_item_one = DistributionPlanLineItemFactory(distribution_plan_node=node,
planned_distribution_date=datetime.datetime.now())
DistributionPlanLineItemFactory(distribution_plan_node=node,
planned_distribution_date=datetime.datetime.now())
when(DistributionPlanNode.objects).get(id=node.id).thenReturn(node)
when(DistributionPlanLineItem.objects).get(id=line_item_one.id).thenReturn(line_item_one)
when(node.consignee).build_contact().thenReturn(self.contact)
schedule_run_for(line_item_one)
task_id = line_item_one.current_node_line_item_run().scheduled_message_task_id
when(celery.app.control).revoke(task_id).thenReturn(None)
schedule_run_for(line_item_one)
verify(celery.app.control).revoke(task_id)
verify(fake_facade, times=2).start_delivery_flow(sender=any(), consignee=any(), item_description=any())
def test_should_not_get_node_line_item_run_with_status_expired(self):
node_line_item = DistributionPlanLineItemFactory()
NodeLineItemRunFactory(node_line_item=node_line_item, status=NodeLineItemRun.STATUS.expired)
self.assertEqual(node_line_item.current_node_line_item_run(), None)
def test_should_get_node_line_item_run_with_status_in_progress(self):
node_line_item = DistributionPlanLineItemFactory()
line_item_run = NodeLineItemRunFactory(node_line_item=node_line_item, status=NodeLineItemRun.STATUS.in_progress)
self.assertEqual(node_line_item.current_node_line_item_run(), line_item_run)
def test_should_not_get_node_line_item_run_with_status_expired(self):
node_line_item = DistributionPlanLineItemFactory()
NodeLineItemRunFactory(node_line_item=node_line_item,
status=NodeLineItemRun.STATUS.expired)
self.assertEqual(node_line_item.current_node_line_item_run(), None)