def test_collect_views_too_narrow_view_type(self):
with self.assertRaises(ValueError):
# One of the views is only a BigQueryView, not a DirectIngestPreProcessedIngestView
_ = BigQueryViewCollector.collect_and_build_views_in_dir(
DirectIngestPreProcessedIngestView,
VIEWS_DIR_RELATIVE_PATH,
view_file_prefix_filter='good_')
def test_collect_views_narrow_view_type_ok(self):
views = BigQueryViewCollector.collect_and_build_views_in_dir(
DirectIngestPreProcessedIngestView,
VIEWS_DIR_RELATIVE_PATH,
view_file_prefix_filter='good_view_2')
self.assertCountEqual([GOOD_VIEW_2], views)
def test_file_builder_wrong_type_raises(self):
with self.assertRaises(ValueError):
_ = BigQueryViewCollector.collect_and_build_views_in_dir(
BigQueryView,
VIEWS_DIR_RELATIVE_PATH,
view_file_prefix_filter='bad_view_builder_wrong_type')
def test_collect_views(self):
views = BigQueryViewCollector.collect_and_build_views_in_dir(
BigQueryView,
VIEWS_DIR_RELATIVE_PATH,
view_file_prefix_filter='good_')
self.assertCountEqual([GOOD_VIEW_1, GOOD_VIEW_2], views)