feat: wrap the rest query statements into transaction, add helper tests

Author: agneum

engine/internal/schema/diff/diff.go

@@ -56,7 +56,7 @@ func detectNodeType(node *pg_query.Node) []*pg_query.Node {
 		fmt.Println("Select Type")
 	}
 
-	return []*pg_query.Node{node}
+	return wrapTransaction([]*pg_query.Node{node})
 }
 
 // IndexStmt processes index statement.

engine/internal/schema/diff/diff_test.go

@@ -22,9 +22,9 @@ ALTER TABLE pgbench_accounts
     ADD COLUMN test integer NOT NULL DEFAULT 0;
 `
 
-const expectedTpl = `CREATE UNIQUE INDEX CONCURRENTLY title_idx ON films USING btree (title);
+const expectedTpl = `BEGIN; CREATE UNIQUE INDEX CONCURRENTLY title_idx ON films USING btree (title); COMMIT;
 
-DROP INDEX CONCURRENTLY title_idx;
+BEGIN; DROP INDEX CONCURRENTLY title_idx; COMMIT;
 
 BEGIN; ALTER TABLE distributors ADD CONSTRAINT zipchk CHECK (char_length(zipcode) = 5) NOT VALID; COMMIT;
 BEGIN; ALTER TABLE distributors VALIDATE CONSTRAINT zipchk; COMMIT;

engine/internal/schema/diff/makefuncs_test.go

@@ -0,0 +1,26 @@
+package diff
+
+import (
+	"testing"
+
+	pg_query "github.com/pganalyze/pg_query_go/v2"
+	"github.com/stretchr/testify/require"
+)
+
+func TestMakeFuncHelpers(t *testing.T) {
+	t.Run("begin transaction statement", func(t *testing.T) {
+		beginStmt := makeBeginTransactionStmt()
+
+		transactionStmt := beginStmt.GetTransactionStmt()
+		require.NotNil(t, transactionStmt)
+		require.Equal(t, pg_query.TransactionStmtKind_TRANS_STMT_BEGIN, transactionStmt.GetKind())
+	})
+
+	t.Run("commit transaction statement", func(t *testing.T) {
+		commitStmt := makeCommitTransactionStmt()
+
+		transactionStmt := commitStmt.GetTransactionStmt()
+		require.NotNil(t, transactionStmt)
+		require.Equal(t, pg_query.TransactionStmtKind_TRANS_STMT_COMMIT, transactionStmt.GetKind())
+	})
+}