GH-172: Add MRN data sources

acteng · Dec 4, 2024 · 56a1dfc · 56a1dfc
1 parent 036557d
commit 56a1dfc
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 0 deletions.
diff --git a/schemes/domain/schemes/data_sources.py b/schemes/domain/schemes/data_sources.py
@@ -26,3 +26,5 @@ class DataSource(Enum):
     LUF_QUARTERLY_UPDATE = auto()
     CRSTS_SCHEME_LIST = auto()
     CRSTS_QUARTERLY_UPDATE = auto()
+    MRN_SCHEME_LIST = auto()
+    MRN_QUARTERLY_UPDATE = auto()
diff --git a/schemes/infrastructure/database/schemes/data_sources.py b/schemes/infrastructure/database/schemes/data_sources.py
@@ -27,6 +27,8 @@ class DataSourceMapper:
         DataSource.LUF_QUARTERLY_UPDATE: 21,
         DataSource.CRSTS_SCHEME_LIST: 22,
         DataSource.CRSTS_QUARTERLY_UPDATE: 23,
+        DataSource.MRN_SCHEME_LIST: 24,
+        DataSource.MRN_QUARTERLY_UPDATE: 25,
     }
 
     def to_id(self, data_source: DataSource) -> int:

diff --git a/schemes/views/schemes/data_sources.py b/schemes/views/schemes/data_sources.py
@@ -31,6 +31,8 @@ class DataSourceRepr(Enum):
     LUF_QUARTERLY_UPDATE = "LUF quarterly update"
     CRSTS_SCHEME_LIST = "CRSTS scheme list"
     CRSTS_QUARTERLY_UPDATE = "CRSTS quarterly update"
+    MRN_SCHEME_LIST = "MRN scheme list"
+    MRN_QUARTERLY_UPDATE = "MRN quarterly update"
 
     @classmethod
     def from_domain(cls, data_source: DataSource) -> DataSourceRepr:
@@ -65,4 +67,6 @@ def _members() -> dict[DataSource, DataSourceRepr]:
             DataSource.LUF_QUARTERLY_UPDATE: DataSourceRepr.LUF_QUARTERLY_UPDATE,
             DataSource.CRSTS_SCHEME_LIST: DataSourceRepr.CRSTS_SCHEME_LIST,
             DataSource.CRSTS_QUARTERLY_UPDATE: DataSourceRepr.CRSTS_QUARTERLY_UPDATE,
+            DataSource.MRN_SCHEME_LIST: DataSourceRepr.MRN_SCHEME_LIST,
+            DataSource.MRN_QUARTERLY_UPDATE: DataSourceRepr.MRN_QUARTERLY_UPDATE,
         }
diff --git a/tests/infrastructure/database/schemes/test_data_sources.py b/tests/infrastructure/database/schemes/test_data_sources.py
@@ -30,6 +30,8 @@
         (DataSource.LUF_QUARTERLY_UPDATE, 21),
         (DataSource.CRSTS_SCHEME_LIST, 22),
         (DataSource.CRSTS_QUARTERLY_UPDATE, 23),
+        (DataSource.MRN_SCHEME_LIST, 24),
+        (DataSource.MRN_QUARTERLY_UPDATE, 25),
     ],
 )
 class TestDataSourceMapper:

diff --git a/tests/views/schemes/test_data_sources.py b/tests/views/schemes/test_data_sources.py
@@ -30,6 +30,8 @@
         (DataSource.LUF_QUARTERLY_UPDATE, "LUF quarterly update"),
         (DataSource.CRSTS_SCHEME_LIST, "CRSTS scheme list"),
         (DataSource.CRSTS_QUARTERLY_UPDATE, "CRSTS quarterly update"),
+        (DataSource.MRN_SCHEME_LIST, "MRN scheme list"),
+        (DataSource.MRN_QUARTERLY_UPDATE, "MRN quarterly update"),
     ],
 )
 class TestDataSourceRepr: