diff --git a/schemes/domain/schemes/data_sources.py b/schemes/domain/schemes/data_sources.py index a63ce6cd..bd7a94a1 100644 --- 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 index 631a3c89..0e7fbc9f 100644 --- 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 index 4ceea26e..be8dcace 100644 --- 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 index 5a2be570..5970a64d 100644 --- 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 index 1012f260..e6e7953a 100644 --- 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: