fix: tests

Author: cka-y

api/tests/unittest/models/test_basic_feed_impl.py

@@ -95,6 +95,7 @@
     feed_name="feed_name",
     note="note",
     feed_contact_email="feed_contact_email",
+    related_links=[],
     source_info=SourceInfo(
         producer_url="producer_url",
         authentication_type=1,
@@ -149,6 +150,7 @@ def test_from_orm_empty_fields(self):
             external_ids=[],
             redirects=[],
             source_info=SourceInfo(),
+            related_links=[],
         )
         empty_result = FeedImpl.from_orm(empty_feed_orm)
         assert empty_result == expected_empty_feed

api/tests/unittest/models/test_gbfs_feed_impl.py

@@ -49,6 +49,7 @@ def test_from_orm_all_fields(self):
         expected = GbfsFeedImpl(
             id="feed_stable_1",
             system_id="sys1",
+            related_links=[],
             data_type="gbfs",
             created_at=datetime(2024, 1, 1, 10, 0, 0),
             external_ids=[],
@@ -88,6 +89,7 @@ def test_from_orm_empty_fields(self):
             redirects=[],
             locations=[],
             versions=[],
+            related_links=[],
             bounding_box=None,
             bounding_box_generated_at=None,
             source_info=SourceInfo(

api/tests/unittest/models/test_gtfs_feed_impl.py

@@ -134,6 +134,7 @@ def create_test_notice(notice_code: str, total_notices: int, severity: str):
     feed_name="feed_name",
     note="note",
     feed_contact_email="feed_contact_email",
+    related_links=[],
     source_info=SourceInfo(
         producer_url="producer_url",
         authentication_type=1,
@@ -226,6 +227,7 @@ def test_from_orm_empty_fields(self):
             provider=None,
             feed_name="",
             note=None,
+            related_links=[],
             feed_contact_email=None,
             source_info=SourceInfo(
                 producer_url=None,

api/tests/unittest/models/test_gtfs_rt_feed_impl.py

@@ -64,6 +64,7 @@
     id="id",
     data_type="gtfs_rt",
     status="active",
+    related_links=[],
     external_ids=[ExternalIdImpl(external_id="associated_id", source="source")],
     provider="provider",
     feed_name="feed_name",

api/tests/unittest/test_feeds.py

@@ -72,6 +72,7 @@
             "producer_url": "test_producer_url",
         },
         external_ids=[{"external_id": "test_associated_id", "source": "test_source"}],
+        related_links=[],
         redirects=[{"comment": "Some comment", "target_id": "test_target_id"}],
     ).model_dump_json()
 )

functions-python/tasks_executor/src/main.py

@@ -38,6 +38,7 @@
 from tasks.geojson.update_geojson_files_precision import (
     update_geojson_files_precision_handler,
 )
+from tasks.jbda_import.import_jbda_feeds import import_jbda_handler
 
 init_logger()
 LIST_COMMAND: Final[str] = "list"
@@ -77,6 +78,10 @@
         "description": "Rebuilds missing visualization files for GTFS datasets.",
         "handler": rebuild_missing_visualization_files_handler,
     },
+    "jbda_import": {
+        "description": "Imports JBDA data into the system.",
+        "handler": import_jbda_handler,
+    },
 }
 
 

functions-python/tasks_executor/tests/tasks/jbda_import/test_jbda_import.py

@@ -52,7 +52,7 @@ class _FakeSessionOK:
     FEEDS_URL = "https://api.gtfs-data.jp/v2/feeds"
     DETAIL_TMPL = "https://api.gtfs-data.jp/v2/organizations/{org_id}/feeds/{feed_id}"
 
-    def get(self, url):
+    def get(self, url, timeout=60):
         # feeds index
         if url == self.FEEDS_URL:
             return _FakeResponse(