1010
1111class TestDecentralizedEnvironmentalNotificationMessage (unittest .TestCase ):
1212 """Test class for the DecentralizedEnvironmentalNotificationMessage."""
13+
1314 def test__init__ (self ):
1415 """Test DecentralizedEnvironmentalNotificationMessage initialization"""
1516 decentralized_environmental_notification_message = \
@@ -22,7 +23,7 @@ def test__init__(self):
2223 # print(encoded_white)
2324 self .assertEqual (encoded_white , expected_denm )
2425
25- @patch ('time .time' )
26+ @patch ('flexstack.utils.time_service.TimeService .time' )
2627 def test_fullfill_with_denrequest (self , time_mock ):
2728 """Test DENMCoder decoding"""
2829 # Given
@@ -55,7 +56,8 @@ def test_fullfill_with_denrequest(self, time_mock):
5556 # When
5657 decentralized_environmental_notification_message = \
5758 DecentralizedEnvironmentalNotificationMessage ()
58- decentralized_environmental_notification_message .fullfill_with_denrequest (denm_request )
59+ decentralized_environmental_notification_message .fullfill_with_denrequest (
60+ denm_request )
5961
6062 # Then
6163 self .assertEqual (decentralized_environmental_notification_message .denm ['denm' ]
@@ -88,7 +90,7 @@ def test_fullfill_with_denrequest(self, time_mock):
8890 ['management' ]['stationType' ], denm_request .rhs_vehicle_type )
8991 time_mock .assert_called_once ()
9092
91- @patch ('time .time' )
93+ @patch ('flexstack.utils.time_service.TimeService .time' )
9294 def test_fullfill_with_collision_risk_warning (self , time_mock ):
9395 """Test fullfill_with_collision_risk_warning function"""
9496 # Given
@@ -115,7 +117,8 @@ def test_fullfill_with_collision_risk_warning(self, time_mock):
115117 # When
116118 decentralized_environmental_notification_message = \
117119 DecentralizedEnvironmentalNotificationMessage ()
118- decentralized_environmental_notification_message .fullfill_with_collision_risk_warning (denm_request )
120+ decentralized_environmental_notification_message .fullfill_with_collision_risk_warning (
121+ denm_request )
119122
120123 # Then
121124 self .assertEqual (decentralized_environmental_notification_message .denm ['denm' ]
@@ -156,6 +159,7 @@ def test_fullfill_with_vehicle_data(self):
156159
157160class TestDENMTransmissionManagement (unittest .TestCase ):
158161 """Test class for the DENM Transmission Management."""
162+
159163 def setUp (self ):
160164 btp_router = MagicMock ()
161165 btp_router .BTPDataRequest = MagicMock ()
@@ -195,12 +199,14 @@ def test_send_collision_risk_warning_denm(self, mock_fullfill_with_vehicle_data,
195199 self .denm_transmission_management .transmit_denm = MagicMock ()
196200
197201 # When
198- self .denm_transmission_management .send_collision_risk_warning_denm (denm_request )
202+ self .denm_transmission_management .send_collision_risk_warning_denm (
203+ denm_request )
199204
200205 # Then
201206 mock_fullfill_with_vehicle_data .assert_called_once_with (
202207 self .denm_transmission_management .vehicle_data )
203- mock_fullfill_with_collision_risk_warning .assert_called_once_with (denm_request )
208+ mock_fullfill_with_collision_risk_warning .assert_called_once_with (
209+ denm_request )
204210 self .denm_transmission_management .transmit_denm .assert_called_once ()
205211
206212 @patch .object (DecentralizedEnvironmentalNotificationMessage ,
@@ -245,4 +251,5 @@ def test_transmit_denm(self):
245251
246252 # Then
247253 self .denm_transmission_management .btp_router .btp_data_request .assert_called ()
248- self .denm_transmission_management .denm_coder .encode .assert_called_once_with (new_denm .denm )
254+ self .denm_transmission_management .denm_coder .encode .assert_called_once_with (
255+ new_denm .denm )
0 commit comments