Fix test

tests/db/test_stopdam.py

@@ -138,3 +138,13 @@ def test_list_all_stops_in_stop_tree(add_model, system_1, base_pk):
     actual_pks = {stop.pk for stop in stopdam.list_all_stops_in_stop_tree(base_pk)}
 
     assert expected_pks == actual_pks
+
+
+def test_list_direction_rules(add_model, stop_1_1, stop_1_2, stop_1_3):
+    rule_1 = add_model(models.DirectionRule(stop=stop_1_1))
+    rule_2 = add_model(models.DirectionRule(stop=stop_1_2))
+    add_model(models.DirectionRule(stop=stop_1_3))
+
+    assert [rule_1, rule_2] == stopdam.list_direction_rules_for_stops(
+        [stop_1_1.pk, stop_1_2.pk]
+    )

tests/unit/services/test_stopservice.py

@@ -133,7 +133,7 @@ def test_get_in_system_by_id__stop_not_found(monkeypatch):
         stopservice.get_in_system_by_id(SYSTEM_ID, STOP_ONE_ID),
 
 
-def test_get_in_system_by_id(monkeypatch,):
+def test_get_in_system_by_id(monkeypatch):
     stop_one = models.Stop(
         pk=STOP_ONE_PK, id=STOP_ONE_ID, system=models.System(id=SYSTEM_ID),
     )
@@ -147,6 +147,7 @@ def test_get_in_system_by_id(monkeypatch,):
     monkeypatch.setattr(
         stopdam, "list_all_stops_in_stop_tree", lambda *args: [stop_one]
     )
+    monkeypatch.setattr(stopdam, "list_direction_rules_for_stops", lambda *args: [])
     monkeypatch.setattr(
         stopdam,
         "list_stop_time_updates_at_stops",
@@ -178,7 +179,7 @@ def test_get_in_system_by_id(monkeypatch,):
     expected = {
         **fake_stop_tree_response,
         "directions": [DIRECTION_NAME],
-        "stop_times": [{**fake_trip_stop_time_response},],
+        "stop_times": [{**fake_trip_stop_time_response}],
         "latitude": None,
         "longitude": None,
         "url": None,

transiter/services/stopservice.py

@@ -58,7 +58,6 @@ def get_in_system_by_id(
     """
     Get information about a specific stop.
     """
-
     stop = stopdam.get_in_system_by_id(system_id, stop_id)
     if stop is None:
         raise exceptions.IdNotFoundError