def test_select_clause_jails_schema(self) -> None: """Given a JailsBase schema, it returns the basic select query for a table.""" query_builder = CloudSqlSchemaTableRegionFilteredQueryBuilder(JailsBase, self.fake_jails_table, self.mock_columns_to_include) expected_select = f'SELECT {self.fake_jails_table.name}.column1,{self.fake_jails_table.name}.state_code' self.assertEqual(expected_select, query_builder.select_clause())
def test___init__region_codes_none(self) -> None: """Test that no assertion is raised if both region_codes_to_include and region_codes_to_exclude are None""" CloudSqlSchemaTableRegionFilteredQueryBuilder(StateBase, self.fake_jails_table, self.mock_columns_to_include, region_codes_to_include=None, region_codes_to_exclude=None)
def test_join_clause_not_association_table(self) -> None: """No join clause for non-association table.""" query_builder = CloudSqlSchemaTableRegionFilteredQueryBuilder( SchemaType.STATE, self.fake_state_table, self.mock_columns_to_include) self.assertEqual(None, query_builder.join_clause())
def test_join_clause_no_region_codes_in_schema(self) -> None: """Given a SchemaType.JAILS schema it returns None.""" query_builder = CloudSqlSchemaTableRegionFilteredQueryBuilder( SchemaType.JAILS, self.fake_jails_table, self.mock_columns_to_include) self.assertEqual(None, query_builder.join_clause())
def test_from_clause_jails_schema(self) -> None: """Given a JailsBase schema, it returns the basic FROM query for a table.""" query_builder = CloudSqlSchemaTableRegionFilteredQueryBuilder( JailsBase, self.fake_jails_table, self.mock_columns_to_include) expected_select = f"FROM {self.fake_jails_table.name}" self.assertEqual(expected_select, query_builder.from_clause())