Function TestSQLQueryStopOnError

backend/tests/sql_stop_on_error_test.go:15–265

(t *testing.T)

)

func TestSQLQueryStopOnError(t *testing.T) {
	tests := []struct {
		name                 string
		databaseName         string
		dbType               storepb.Engine
		environment          string // Environment to create database in (defaults to prod if empty)
		prepareStatements    string
		query                string
		wantResults          int // Number of successful results before error
		wantError            bool
		wantSyntaxError      bool // Whether to expect syntax_error in detailed_error
		wantPermissionDenied bool // Whether to expect permission_denied in detailed_error
	}{
		{
			name:              "MySQL - All statements succeed",
			databaseName:      "TestStopOnError1",
			dbType:            storepb.Engine_MYSQL,
			prepareStatements: "CREATE TABLE tbl1(id INT PRIMARY KEY, name VARCHAR(64));",
			query:             "INSERT INTO tbl1 VALUES(1, 'Alice'); INSERT INTO tbl1 VALUES(2, 'Bob'); SELECT * FROM tbl1;",
			wantResults:       3, // 2 inserts + 1 select
			wantError:         false,
		},
		{
			name:              "MySQL - Second statement fails",
			databaseName:      "TestStopOnError2",
			dbType:            storepb.Engine_MYSQL,
			prepareStatements: "CREATE TABLE tbl2(id INT PRIMARY KEY, name VARCHAR(64));",
			query:             "INSERT INTO tbl2 VALUES(1, 'Alice'); INSERT INTO nonexistent VALUES(2, 'Bob'); INSERT INTO tbl2 VALUES(3, 'Charlie');",
			wantResults:       2, // First insert succeeds + error result
			wantError:         true,
		},
		{
			name:              "MySQL - First statement fails",
			databaseName:      "TestStopOnError3",
			dbType:            storepb.Engine_MYSQL,
			prepareStatements: "CREATE TABLE tbl3(id INT PRIMARY KEY, name VARCHAR(64));",
			query:             "INSERT INTO nonexistent VALUES(1, 'Alice'); INSERT INTO tbl3 VALUES(2, 'Bob');",
			wantResults:       1, // Error result only
			wantError:         true,
		},
		{
			name:              "PostgreSQL - All statements succeed",
			databaseName:      "TestStopOnError4",
			dbType:            storepb.Engine_POSTGRES,
			prepareStatements: "CREATE TABLE tbl4(id INT PRIMARY KEY, name VARCHAR(64));",
			query:             "INSERT INTO tbl4 VALUES(1, 'Alice'); INSERT INTO tbl4 VALUES(2, 'Bob'); SELECT * FROM tbl4;",
			wantResults:       3,
			wantError:         false,
		},
		{
			name:              "PostgreSQL - Second statement fails",
			databaseName:      "TestStopOnError5",
			dbType:            storepb.Engine_POSTGRES,
			prepareStatements: "CREATE TABLE tbl5(id INT PRIMARY KEY, name VARCHAR(64));",
			query:             "INSERT INTO tbl5 VALUES(1, 'Alice'); INSERT INTO nonexistent VALUES(2, 'Bob'); INSERT INTO tbl5 VALUES(3, 'Charlie');",
			wantResults:       2, // First insert succeeds + error result
			wantError:         true,
		},