fix(sampling): trim quotes in estimate query

Author: aarthy-dk

testgen/commands/queries/profiling_query.py

@@ -4,7 +4,7 @@
 from testgen.commands.queries.refresh_data_chars_query import CRefreshDataCharsSQL
 from testgen.commands.queries.rollup_scores_query import CRollupScoresSQL
 from testgen.common import date_service, read_template_sql_file, read_template_yaml_file
-from testgen.common.database.database_service import replace_params
+from testgen.common.database.database_service import get_flavor_service, replace_params
 from testgen.common.read_file import replace_templated_functions
 
 
@@ -121,6 +121,7 @@ def _get_params(self) -> dict:
             "CONTINGENCY_MAX_VALUES": self.contingency_max_values,
             "PROCESS_ID": self.process_id,
             "SQL_FLAVOR": self.flavor,
+            "QUOTE": get_flavor_service(self.flavor).quote_character
         }
 
     def _get_query(

testgen/template/profiling/project_update_profile_results_to_estimates.sql

@@ -24,8 +24,8 @@ set sample_ratio = :PROFILE_SAMPLE_RATIO,
     future_date_ct = ROUND(future_date_ct * :PROFILE_SAMPLE_RATIO, 0),
     boolean_true_ct = ROUND(boolean_true_ct * :PROFILE_SAMPLE_RATIO, 0)
 where profile_run_id = :PROFILE_RUN_ID
-and schema_name = split_part(:SAMPLING_TABLE, '.', 1)
-and table_name = split_part(:SAMPLING_TABLE, '.', 2)
+and schema_name = TRIM(SPLIT_PART(:SAMPLING_TABLE, '.', 1), :QUOTE)
+and table_name = TRIM(SPLIT_PART(:SAMPLING_TABLE, '.', 2), :QUOTE)
 and sample_ratio IS NULL;