Function TestSQLAdminExecuteStopOnError

backend/tests/sql_stop_on_error_test.go:267–409

(t *testing.T)

}

func TestSQLAdminExecuteStopOnError(t *testing.T) {
	tests := []struct {
		name              string
		databaseName      string
		dbType            storepb.Engine
		prepareStatements string
		query             string
		wantResults       int
		wantError         bool
	}{
		{
			name:              "MySQL AdminExecute - Second statement fails",
			databaseName:      "TestAdminStopOnError1",
			dbType:            storepb.Engine_MYSQL,
			prepareStatements: "CREATE TABLE admin_tbl1(id INT PRIMARY KEY, name VARCHAR(64));",
			query:             "INSERT INTO admin_tbl1 VALUES(1, 'Alice'); INSERT INTO nonexistent VALUES(2, 'Bob'); INSERT INTO admin_tbl1 VALUES(3, 'Charlie');",
			wantResults:       1,
			wantError:         true,
		},
		{
			name:              "PostgreSQL AdminExecute - Second statement fails",
			databaseName:      "TestAdminStopOnError2",
			dbType:            storepb.Engine_POSTGRES,
			prepareStatements: "CREATE TABLE admin_tbl2(id INT PRIMARY KEY, name VARCHAR(64));",
			query:             "INSERT INTO admin_tbl2 VALUES(1, 'Alice'); INSERT INTO nonexistent VALUES(2, 'Bob'); INSERT INTO admin_tbl2 VALUES(3, 'Charlie');",
			wantResults:       1,
			wantError:         true,
		},
	}

	t.Parallel()
	a := require.New(t)
	ctx := context.Background()
	ctl := &controller{}
	ctx, err := ctl.StartServerWithExternalPg(ctx)
	a.NoError(err)
	t.Cleanup(func() {
		ctl.Close(ctx)
	})

	mysqlContainer, err := getMySQLContainer(ctx)
	t.Cleanup(func() {
		mysqlContainer.Close(ctx)
	})
	a.NoError(err)

	pgContainer, err := getPgContainer(ctx)
	t.Cleanup(func() {
		pgContainer.Close(ctx)
	})
	a.NoError(err)

	mysqlInstanceResp, err := ctl.instanceServiceClient.CreateInstance(ctx, connect.NewRequest(&v1pb.CreateInstanceRequest{
		InstanceId: generateRandomString("instance"),
		Instance: &v1pb.Instance{
			Title:       "mysqlInstance",
			Engine:      v1pb.Engine_MYSQL,
			Environment: new("environments/prod"),