def test_should_return_delivery_type_unknown(self): delivery = DeliveryFactory() self.assertEqual(delivery.type(), 'Unknown')
def test_should_return_delivery_type_waybill(self): ro_item = ReleaseOrderItemFactory() delivery = DeliveryFactory() DeliveryNodeFactory(distribution_plan=delivery, item=ro_item) self.assertEqual(delivery.type(), 'Waybill')
def test_should_return_delivery_type_purchase_order(self): po_item = PurchaseOrderItemFactory() delivery = DeliveryFactory() DeliveryNodeFactory(distribution_plan=delivery, item=po_item) self.assertEqual(delivery.type(), 'Purchase Order')