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')