"""SQL queries for use in test utilities.""" from tests.consts import snowflake as snowflake_consts CREATE_UNIQUE_TEST_SCHEMA = """ CREATE TRANSIENT SCHEMA IF NOT EXISTS {schema_name}; """.format(schema_name=snowflake_consts.UNIQUE_TEST_SCHEMA) DROP_UNIQUE_TEST_SCHEMA = """ DROP SCHEMA IF EXISTS {schema_name}; """.format(schema_name=snowflake_consts.UNIQUE_TEST_SCHEMA) def insert_into(table_name, table_data): """Insert values into existing table.""" return """INSERT INTO {table_name} {columns} VALUES {values};""".format( table_name=table_name, values=table_data.get('values'), columns=table_data.get('columns'), ) def create_table(view_name): """Create table string format.""" return """CREATE TABLE {view_name} LIKE sony_internal.qa.{view_name}; """.format(view_name=view_name) CREATE_TABLE_US_PHYSICAL_PRODUCT_VIEW = create_table( 'US_PHYSICAL_PRODUCT_VIEW' ) CREATE_TABLE_CA_PHYSICAL_PRODUCT_VIEW = create_table( 'CA_PHYSICAL_PRODUCT_VIEW' ) CREATE_TABLE_UK_PHYSICAL_PRODUCT_VIEW = create_table( 'UK_PHYSICAL_PRODUCT_VIEW' ) CREATE_TABLE_US_PHYSICAL_RETAILER_VIEW = create_table( 'US_PHYSICAL_RETAIL_VIEW' ) CREATE_TABLE_CA_PHYSICAL_RETAILER_VIEW = create_table( 'CA_PHYSICAL_RETAIL_VIEW' ) CREATE_TABLE_PHYSICAL_PRODUCT_SALES_RTD_VIEW = create_table( 'PHYSICAL_PRODUCT_SALES_RTD_VIEW' ) CREATE_TABLE_PHYSICAL_PRODUCT_SALES_BY_MONTH_VIEW = create_table( 'PHYSICAL_PRODUCT_SALES_BY_MONTH_VIEW' ) CREATE_TABLE_PHYSICAL_CURRENT_PERIOD_VIEW = create_table( 'PHYSICAL_CURRENT_PERIOD_VIEW' ) CREATE_PHYSICAL_PRODUCT_INVENTORY_VIEW = create_table( 'PHYSICAL_PRODUCT_INVENTORY_VIEW' ) CREATE_PHYSICAL_PRODUCT_METADATA_VIEW = create_table( 'PHYSICAL_PRODUCT_METADATA_VIEW' ) CREATE_SAP_PERPETUAL_INVENTORY_VIEW = create_table('SAP_PERPETUAL_INVENTORY') CREATE_TABLE_EU_POINT_OF_SALE_BY_PRODUCT_COUNTRY_VIEW = create_table( 'EU_POINT_OF_SALE_BY_PRODUCT_COUNTRY_VIEW' ) CREATE_TABLE_EU_POINT_OF_SALE_BY_PRODUCT_VIEW = create_table( 'EU_POINT_OF_SALE_BY_PRODUCT_VIEW' ) CREATE_TABLE_EU_POINT_OF_SALE_BY_COUNTRY_VIEW = create_table( 'EU_POINT_OF_SALE_BY_COUNTRY_VIEW' )