Skip to content

Commit b248180

Browse files
committed
fix: update DataSource tests for new URL provider fields
1 parent 2b635df commit b248180

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

tests/domain/models/data/test_data_source.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -348,7 +348,14 @@ def test_data_source_is_dataclass(self):
348348
"storage_path",
349349
"date",
350350
"identifier",
351-
"warmup_window"
351+
"warmup_window",
352+
"url",
353+
"date_column",
354+
"date_format",
355+
"cache",
356+
"refresh_interval",
357+
"pre_process",
358+
"post_process"
352359
]
353360

354361
self.assertEqual(set(field_names), set(expected_fields))
@@ -362,7 +369,7 @@ def test_data_source_field_defaults(self):
362369

363370
for field in fields(DataSource):
364371

365-
if field.name in ["save", "pandas"]:
372+
if field.name in ["save", "pandas", "cache"]:
366373
# These fields are expected to have a default value
367374
self.assertIsNotNone(field.default)
368375
else:

0 commit comments

Comments
 (0)