Function TestTransactionMode

backend/tests/transaction_mode_test.go:18–193

TestTransactionMode tests the configurable transaction mode feature across different database engines.

(t *testing.T)

// TestTransactionMode tests the configurable transaction mode feature across different database engines.
func TestTransactionMode(t *testing.T) {
	ctx := context.Background()

	tests := []struct {
		name             string
		dbType           storepb.Engine
		containerFunc    func(context.Context, *testing.T) *testcontainer.Container
		setupScript      string
		testScriptOn     string
		testScriptOff    string
		expectRollbackOn bool // Whether we expect rollback for transaction mode on
		skipTransaction  bool // Some engines don't support transactions well
	}{
		{
			name:   "MySQL",
			dbType: storepb.Engine_MYSQL,
			containerFunc: func(ctx context.Context, t *testing.T) *testcontainer.Container {
				container, err := testcontainer.GetTestMySQLContainer(ctx)
				if err != nil {
					t.Fatalf("Failed to get MySQL container: %v", err)
				}
				return container
			},
			setupScript: `
				CREATE TABLE test_table (
					id INT PRIMARY KEY,
					value VARCHAR(100)
				);
			`,
			testScriptOn: `-- txn-mode = on
				INSERT INTO test_table (id, value) VALUES (1, 'test1');
				INSERT INTO test_table (id, value) VALUES (2, 'test2');
				INSERT INTO test_table (id, value) VALUES (1, 'duplicate'); -- This will fail
			`,
			testScriptOff: `-- txn-mode = off
				INSERT INTO test_table (id, value) VALUES (1, 'test1');
				INSERT INTO test_table (id, value) VALUES (2, 'test2');
				INSERT INTO test_table (id, value) VALUES (1, 'duplicate'); -- This will fail
			`,
			expectRollbackOn: true,
		},
		{
			name:   "PostgreSQL",
			dbType: storepb.Engine_POSTGRES,
			containerFunc: func(ctx context.Context, t *testing.T) *testcontainer.Container {
				return testcontainer.GetTestPgContainer(ctx, t)
			},
			setupScript: `
				CREATE TABLE test_table (
					id INTEGER PRIMARY KEY,
					value VARCHAR(100)
				);
			`,
			testScriptOn: `-- txn-mode = on
				INSERT INTO test_table (id, value) VALUES (1, 'test1');
				INSERT INTO test_table (id, value) VALUES (2, 'test2');
				INSERT INTO test_table (id, value) VALUES (1, 'duplicate'); -- This will fail
			`,