[P] Add more datasource tests

This commit is contained in:
2026-03-31 12:49:26 +03:00
parent 058aac36c7
commit 92c0acda2c

View File

@@ -80,6 +80,51 @@ class TestMapLatLonToProcessedAgentData(unittest.TestCase):
self.assertIsNotNone(result) self.assertIsNotNone(result)
self.assertEqual(result, marker) self.assertEqual(result, marker)
def test_marker_is_close_enough(self):
marker = Mock()
marker.latitude = 30.521
marker.longitude = 50.452
self.instance._active_markers = [marker]
result = self.instance.map_lat_lon_to_processed_agent_data(50.45, 30.52)
self.assertIsNotNone(result)
self.assertEqual(result, marker)
def test_marker_is_too_far(self):
marker = Mock()
marker.latitude = 30.524
marker.longitude = 50.454
self.instance._active_markers = [marker]
result = self.instance.map_lat_lon_to_processed_agent_data(50.45, 30.52)
self.assertIsNone(result)
def test_return_closer_marker(self):
marker1 = Mock()
marker1.latitude = 30.521
marker1.longitude = 50.451
marker2 = Mock()
marker2.latitude = 30.524
marker2.longitude = 50.454
self.instance._active_markers = [marker1, marker2]
result = self.instance.map_lat_lon_to_processed_agent_data(50.452, 30.522)
self.assertIsNotNone(result)
self.assertEqual(result, marker1)
result = self.instance.map_lat_lon_to_processed_agent_data(50.453, 30.523)
self.assertIsNotNone(result)
self.assertEqual(result, marker2)
result = self.instance.map_lat_lon_to_processed_agent_data(50.459, 30.529)
self.assertIsNone(result)
def test_empty_markers(self): def test_empty_markers(self):
self.instance._active_markers = [] self.instance._active_markers = []