diff --git a/flyteadmin/script/integration/k8s/integration.yaml b/flyteadmin/script/integration/k8s/integration.yaml index 4fae0bab27..5e57935316 100644 --- a/flyteadmin/script/integration/k8s/integration.yaml +++ b/flyteadmin/script/integration/k8s/integration.yaml @@ -417,6 +417,9 @@ spec: - name: flyteadmin image: flyteadmin:test imagePullPolicy: IfNotPresent + env: + - name: USE_INTEGRATION_TEST_CONFIG + value: "True" securityContext: privileged: true readinessProbe: diff --git a/flyteadmin/tests/bootstrap.go b/flyteadmin/tests/bootstrap.go index 9fd3b27026..aa1808282d 100644 --- a/flyteadmin/tests/bootstrap.go +++ b/flyteadmin/tests/bootstrap.go @@ -6,6 +6,7 @@ package tests import ( "context" "fmt" + "os" "gorm.io/gorm" @@ -19,9 +20,19 @@ const insertExecutionQueryStr = `INSERT INTO "executions" ` + `("execution_project","execution_domain","execution_name","phase","launch_plan_id","workflow_id") ` + `VALUES ('%s', '%s', '%s', '%s', '%d', '%d')` +const integrationTestConfigEnvVar = "USE_INTEGRATION_TEST_CONFIG" + var adminScope = promutils.NewScope("flyteadmin") func getDbConfig() *database.DbConfig { + if os.Getenv(integrationTestConfigEnvVar) == "True" { + return getIntegrationDbConfig() + } else { + return getSandboxDbConfig() + } +} + +func getIntegrationDbConfig() *database.DbConfig { return &database.DbConfig{ Postgres: database.PostgresConfig{ Host: "postgres", @@ -32,13 +43,15 @@ func getDbConfig() *database.DbConfig { } } -func getLocalDbConfig() *database.DbConfig { +// Run `flytectl demo start` to start the sandbox +func getSandboxDbConfig() *database.DbConfig { return &database.DbConfig{ Postgres: database.PostgresConfig{ - Host: "localhost", - Port: 5432, - DbName: "flyteadmin", - User: "postgres", + Host: "localhost", + Port: 30001, + DbName: "flyte", + Password: "postgres", + User: "postgres", }, } }