From c1ee6c55a007c17c599834e4254d72c796b1a9fe Mon Sep 17 00:00:00 2001 From: Marco Zeisler Date: Thu, 17 Jun 2021 22:05:32 +0200 Subject: [PATCH] fix datetime mocking --- components/i_feed/test_ifeed.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/components/i_feed/test_ifeed.py b/components/i_feed/test_ifeed.py index 2d8b393..d6e79f0 100644 --- a/components/i_feed/test_ifeed.py +++ b/components/i_feed/test_ifeed.py @@ -47,9 +47,8 @@ class TestVehicle(unittest.TestCase): self.assertEqual(self.v.nce, False) def test_daf_prop(self): - with patch('datetime.datetime', MyDate): - # noinspection PyUnresolvedReferences - datetime.datetime.set_timestamp(self.timestamp) + with patch('devices.vehicle.datetime', MyDate): + MyDate.set_timestamp(self.timestamp) # test if daf object created properly daf = DAF(vehicle_identification_number=self.vin, @@ -60,13 +59,8 @@ class TestVehicle(unittest.TestCase): self.assertEqual(self.v.daf, daf) def test_current_location_prop(self): - with patch('datetime.datetime', MyDate): - # noinspection PyUnresolvedReferences - datetime.datetime.set_timestamp(self.timestamp) - - # assert timestamp mocking worked - self.assertEqual(type(datetime.datetime), type(MyDate)) - self.assertEqual(self.timestamp, datetime.datetime.now()) + with patch('devices.vehicle.datetime', MyDate): + MyDate.set_timestamp(self.timestamp) # initial position should be starting point self.assertEqual(self.v.gps_location, self.starting_point)