MCPcopy Index your code
hub / github.com/bytebase/bytebase / createTestPostgreSQLDatabase

Method createTestPostgreSQLDatabase

backend/tests/action_test.go:615–654  ·  view source on GitHub ↗

createTestPostgreSQLDatabase creates a test PostgreSQL database instance and database

(ctx context.Context, t *testing.T)

Source from the content-addressed store, hash-verified

613
614// createTestPostgreSQLDatabase creates a test PostgreSQL database instance and database
615func (ctl *controller) createTestPostgreSQLDatabase(ctx context.Context, t *testing.T) (*v1pb.Database, *Container) {
616 a := require.New(t)
617
618 // Get PostgreSQL container
619 pgContainer, err := getPgContainer(ctx)
620 a.NoError(err)
621
622 // Create PostgreSQL instance
623 instanceResp, err := ctl.instanceServiceClient.CreateInstance(ctx, connect.NewRequest(&v1pb.CreateInstanceRequest{
624 InstanceId: generateRandomString("inst")[:8],
625 Instance: &v1pb.Instance{
626 Title: "Test PostgreSQL Instance",
627 Engine: v1pb.Engine_POSTGRES,
628 Environment: new("environments/prod"),
629 Activation: true,
630 DataSources: []*v1pb.DataSource{{
631 Type: v1pb.DataSourceType_ADMIN,
632 Host: pgContainer.host,
633 Port: pgContainer.port,
634 Username: "postgres",
635 Password: "root-password",
636 Id: "admin",
637 }},
638 },
639 }))
640 a.NoError(err)
641
642 // Create database
643 dbName := generateRandomString("db")[:8]
644 err = ctl.createDatabase(ctx, ctl.project, instanceResp.Msg, nil, dbName, "postgres")
645 a.NoError(err)
646
647 // Get database
648 databaseResp, err := ctl.databaseServiceClient.GetDatabase(ctx, connect.NewRequest(&v1pb.GetDatabaseRequest{
649 Name: fmt.Sprintf("%s/databases/%s", instanceResp.Msg.Name, dbName),
650 }))
651 a.NoError(err)
652
653 return databaseResp.Msg, pgContainer
654}
655
656func TestActionRolloutCommand(t *testing.T) {
657 t.Parallel()

Callers 3

TestActionCheckCommandFunction · 0.95

Calls 5

createDatabaseMethod · 0.95
getPgContainerFunction · 0.85
generateRandomStringFunction · 0.85
CreateInstanceMethod · 0.65
GetDatabaseMethod · 0.65

Tested by

no test coverage detected