def test_should_get_the_current_completed_line_item_run(self):
node_line_item = DistributionPlanLineItemFactory()
self.assertIsNone(node_line_item.completed_run())
line_item_run = NodeLineItemRunFactory(node_line_item=node_line_item, status='completed')
self.assertEqual(node_line_item.completed_run(), line_item_run)
class DistributionPlanLineItemTest(TestCase):
def setUp(self):
self.node_line_item = DistributionPlanLineItemFactory()
def test_should_have_all_expected_fields(self):
line_item = DistributionPlanLineItem()
fields_in_item = line_item._meta._name_map
self.assertEqual(len(line_item._meta.fields), 7)
expected_fields = [
'item_id', 'targeted_quantity', 'planned_distribution_date', 'remark', 'distribution_plan_node_id', 'id',
'track']
for field in expected_fields:
self.assertIn(field, fields_in_item)
def test_should_get_node_line_item_run_with_status_scheduled(self):
line_item_run = NodeLineItemRunFactory(node_line_item=self.node_line_item,
status=NodeLineItemRun.STATUS.scheduled)
self.assertEqual(self.node_line_item.current_run(), line_item_run)
def test_should_not_get_node_line_item_run_with_status_completed(self):
NodeLineItemRunFactory(node_line_item=self.node_line_item, status=NodeLineItemRun.STATUS.completed)
self.assertEqual(self.node_line_item.current_run(), None)
def test_should_not_get_node_line_item_run_with_status_expired(self):
NodeLineItemRunFactory(node_line_item=self.node_line_item, status=NodeLineItemRun.STATUS.expired)
self.assertEqual(self.node_line_item.current_run(), None)
def test_should_not_get_node_line_item_run_with_status_cancelled(self):
NodeLineItemRunFactory(node_line_item=self.node_line_item, status=NodeLineItemRun.STATUS.cancelled)
self.assertEqual(self.node_line_item.current_run(), None)
def test_should_get_the_completed_line_item_run(self):
self.assertIsNone(self.node_line_item.completed_run())
line_item_run = NodeLineItemRunFactory(node_line_item=self.node_line_item,
status=NodeLineItemRun.STATUS.completed)
self.assertEqual(self.node_line_item.completed_run(), line_item_run)
def test_should_get_latest_run(self):
first_run = NodeLineItemRunFactory(node_line_item=self.node_line_item)
self.assertEqual(self.node_line_item.latest_run(), first_run)
second_run = NodeLineItemRunFactory(node_line_item=self.node_line_item)
self.assertEqual(self.node_line_item.latest_run(), second_run)