diff --git a/erpnext/stock/doctype/delivery_trip/test_delivery_trip.py b/erpnext/stock/doctype/delivery_trip/test_delivery_trip.py index 8e467f84fba..68bcfe45629 100644 --- a/erpnext/stock/doctype/delivery_trip/test_delivery_trip.py +++ b/erpnext/stock/doctype/delivery_trip/test_delivery_trip.py @@ -22,16 +22,29 @@ class TestDeliveryTrip(ERPNextTestSuite): create_test_contact_and_address() address = create_address(driver) - self.delivery_trip = create_delivery_trip(driver, address) + self.delivery_trip = create_delivery_trip(driver, address, company=self.companies[0].name) def tearDown(self): - frappe.db.sql("delete from `tabDriver`") - frappe.db.sql("delete from `tabVehicle`") - frappe.db.sql("delete from `tabEmail Template`") - frappe.db.sql("delete from `tabDelivery Trip`") - return super().tearDown() + frappe.db.rollback() def test_delivery_trip_notify_customers(self): + # set default outgoing + outgoing = frappe.get_doc( + { + "doctype": "Email Account", + "company": self.companies[0].name, + "enable_outgoing": 1, + "default_outgoing": 1, + "awaiting_password": 1, + "auth_method": "Basic", + "password": "test", + "smtp_server": "localhost", + "stmp_port": 25, + "email_id": "test@example.in", + } + ) + outgoing.save() + notify_customers(delivery_trip=self.delivery_trip.name) self.delivery_trip.load_from_db() self.assertEqual(self.delivery_trip.email_notification_sent, 1) @@ -175,14 +188,14 @@ def create_vehicle(): vehicle.insert() -def create_delivery_trip(driver, address, contact=None): +def create_delivery_trip(driver, address, contact=None, company=None): if not contact: contact = get_contact_and_address("_Test Customer") delivery_trip = frappe.get_doc( { "doctype": "Delivery Trip", - "company": erpnext.get_default_company(), + "company": company or erpnext.get_default_company(), "departure_time": add_days(now_datetime(), 5), "driver": driver.name, "driver_address": address.name,