def test_returned_nodes_should_have_order_type_field(self):
po_node = DeliveryNodeFactory(item=PurchaseOrderItemFactory(purchase_order=(PurchaseOrderFactory())))
ro_node = DeliveryNodeFactory(item=ReleaseOrderItemFactory(release_order=(ReleaseOrderFactory())))
response = self.client.get(ENDPOINT_URL)
node_order_types = [node['order_type'] for node in response.data]
self.assertItemsEqual([po_node.type(), ro_node.type()], node_order_types)
def test_should_return_delivery_type_waybill(self):
ro_item = ReleaseOrderItemFactory()
node = DeliveryNodeFactory(item=ro_item)
self.assertEqual(node.type(), 'Waybill')
def test_should_return_delivery_type_purchase_order(self):
po_item = PurchaseOrderItemFactory()
node = DeliveryNodeFactory(item=po_item)
self.assertEqual(node.type(), 'Purchase Order')
def test_should_return_delivery_type_waybill(self):
ro_item = ReleaseOrderItemFactory()
node = DeliveryNodeFactory(item=ro_item)
self.assertEqual(node.type(), 'Waybill')
def test_should_return_delivery_type_purchase_order(self):
po_item = PurchaseOrderItemFactory()
node = DeliveryNodeFactory(item=po_item)
self.assertEqual(node.type(), 'Purchase Order')
