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)