def test_get_tables_from_properties(self): """Test getting selected tables from tap properties JSON""" # Load MySQL and Postgres properties JSON mysql_properties = utils.load_json( '{}/properties_mysql.json'.format(RESOURCES_DIR)) postgres_properties = utils.load_json( '{}/properties_postgres.json'.format(RESOURCES_DIR)) # Get list of selected tables # MySQL and Postgres schemas defined at different keys. get_tables_from_properties function # should detect and extract correctly mysql_tables = utils.get_tables_from_properties(mysql_properties) postgres_tables = utils.get_tables_from_properties(postgres_properties) # MySQL schema assert mysql_tables == \ [ 'mysql_source_db.address', 'mysql_source_db.order', 'mysql_source_db.weight_unit' ] assert postgres_tables == \ [ 'public.city', 'public.country' ]
def test_get_tables_from_properties_for_s3_csv(self): properties = utils.load_json( '{}/properties_s3_csv.json'.format(RESOURCES_DIR)) s3_csv_tables = utils.get_tables_from_properties(properties) # MySQL schema assert s3_csv_tables == { 'applications', 'candidate_survey_questions', 'interviews', }