bug #765: Fixed function parsing issue in case function's complete bo…

…dy is in one single line (#768)

- Also, added an automation test named pg-hasura-ecommerce for this fix

.github/workflows/pg-migtests.yml

@@ -120,3 +120,6 @@ jobs:
     - name: "TEST: pg-sample-schema-emp"
       run: migtests/scripts/run-test.sh pg-tests/pg-sample-employee
 
+    - name: "TEST: pg-hasura-ecommerce"
+      run: migtests/scripts/run-test.sh pg-hasura-ecommerce
+

migtests/lib/yb.py

@@ -151,10 +151,10 @@ def invalid_index_present(self, table_name, schema_name="public"):
 		return False
 
 
-	def fetch_all_triggers(self, schema_name="public") -> List[str]:
+	def fetch_all_triggers(self, schema_name="public") -> set[str]:
 		cur = self.conn.cursor()
 		cur.execute(f"SELECT trigger_name FROM information_schema.triggers WHERE trigger_schema = '{schema_name}'")
-		return [trigger[0] for trigger in cur.fetchall()]
+		return set(cur.fetchall())
 
 	def fetch_all_procedures(self, schema_name="public") -> List[str]:
 		cur = self.conn.cursor()
@@ -179,4 +179,14 @@ def fetch_all_function_names(self, schema_name="public") -> List[str]:
 	def fetch_all_table_rows(self, table_name, schema_name="public") -> set[str]:
 		cur = self.conn.cursor()
 		cur.execute(f"SELECT * FROM {schema_name}.{table_name}")
+		return set(cur.fetchall())
+
+	def fetch_all_pg_extension(self, schema_name = "public") -> set[str]:
+		cur = self.conn.cursor()
+		cur.execute(f"SELECT extname FROM pg_extension")
+		return set(cur.fetchall())
+
+	def fetch_all_schemas(self) -> set[str]:
+		cur = self.conn.cursor()
+		cur.execute(f"SELECT schema_name FROM information_schema.schemata where schema_name !~ '^pg_' and schema_name <> 'information_schema'")
 		return set(cur.fetchall())

migtests/scripts/run-test.sh

@@ -68,6 +68,12 @@ main() {
 	export_data
 	ls -l ${EXPORT_DIR}/data
 
+	step "Fix data."
+	if [ -x "${TEST_DIR}/fix-data" ]
+	then
+		"${TEST_DIR}/fix-data" ${EXPORT_DIR}
+	fi
+
 	step "Create target database."
 	run_ysql yugabyte "DROP DATABASE IF EXISTS ${TARGET_DB_NAME};"
 	run_ysql yugabyte "CREATE DATABASE ${TARGET_DB_NAME}"