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())
