def test_create_dose_form_route_mappings_table_with_dataset_id(
self, mock_query, mock_create_table):
# pre conditions
route_mappings_csv = os.path.join(
resources.resource_files_path,
populate_route_ids.DOSE_FORM_ROUTES_FILE + ".csv")
dose_form_route_mappings = resources.csv_to_list(route_mappings_csv)
mapping_list = populate_route_ids.get_mapping_list(
dose_form_route_mappings)
query_params = dict(
project_id=self.project_id,
dataset_id=self.dataset_id,
routes_table_id=populate_route_ids.DOSE_FORM_ROUTES_TABLE_ID,
mapping_list=mapping_list)
expected_query = populate_route_ids.INSERT_ROUTES_QUERY.format(
**query_params)
# test
populate_route_ids.create_dose_form_route_mappings_table(
self.project_id, self.dataset_id)
# post conditions
mock_query.assert_called_with(expected_query)
mock_create_table.assert_called_with(
populate_route_ids.DOSE_FORM_ROUTES_TABLE_ID,
populate_route_ids.DOSE_FORM_ROUTE_FIELDS,
drop_existing=True,
dataset_id=self.dataset_id)
def test_get_mapping_list(self):
actual = populate_route_ids.get_mapping_list(self.route_mappings_list)
expected = self.route_mappings_string
self.assertEqual(expected, actual)
