姚章浩
73 lines
2.5 KiB
Go
73 lines
2.5 KiB
Go
package scripts
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestDockerDeploymentContainsOnlyDobotAIApplication(t *testing.T) {
|
|
compose, err := os.ReadFile("../deploy/dobotai-dingtalk/compose.yaml")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
source := string(compose)
|
|
for _, required := range []string{"dobotai-dingtalk:", "container_name: dobotai-dingtalk", "network_mode: host", "env_file:", "read_only: true"} {
|
|
if !strings.Contains(source, required) {
|
|
t.Errorf("compose missing %q", required)
|
|
}
|
|
}
|
|
for _, forbidden := range []string{"dobotai-postgres", "postgres-data", "name: dobotai-network"} {
|
|
if strings.Contains(source, forbidden) {
|
|
t.Errorf("compose must not manage a local database resource %q", forbidden)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRootInstallerBuildsImageWithoutOperatorImageArgument(t *testing.T) {
|
|
data, err := os.ReadFile("../install-docker.sh")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
source := string(data)
|
|
for _, required := range []string{"docker build", "dobotai-dingtalk:local-", "install_docker_if_missing", "DOBOTAI_SOURCE_ROOT", "deploy/dobotai-dingtalk/install-docker.sh"} {
|
|
if !strings.Contains(source, required) {
|
|
t.Errorf("root installer missing %q", required)
|
|
}
|
|
}
|
|
if strings.Contains(source, "${DOBOTAI_IMAGE") {
|
|
t.Fatal("root installer must not require DOBOTAI_IMAGE")
|
|
}
|
|
}
|
|
|
|
func TestRDSInitializerOnlyInitializesEmptySchemaAndRejectsInconsistentSchema(t *testing.T) {
|
|
data, err := os.ReadFile("../deploy/dobotai-dingtalk/rds-schema-init.sh")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
source := string(data)
|
|
for _, required := range []string{"pg_isready", "to_regclass('public.turns')", "empty", "normal", "inconsistent", "数据库不存在", "*.up.sql", "DATABASE_URL", "postgres:16-alpine"} {
|
|
if !strings.Contains(source, required) {
|
|
t.Errorf("RDS initializer missing %q", required)
|
|
}
|
|
}
|
|
for _, forbidden := range []string{"docker compose", "dobotai-postgres", "pg_dump -Fc", "DROP DATABASE", "DROP SCHEMA"} {
|
|
if strings.Contains(source, forbidden) {
|
|
t.Errorf("RDS initializer must not manage or destroy databases: %q", forbidden)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRDSCredentialConfiguratorPromptsForExternalDatabase(t *testing.T) {
|
|
data, err := os.ReadFile("../deploy/dobotai-dingtalk/configure_env.py")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
source := string(data)
|
|
for _, required := range []string{"DATABASE_URL", "阿里云RDS", "getpass.getpass", "os.replace", "0o600"} {
|
|
if !strings.Contains(source, required) {
|
|
t.Errorf("RDS configurator missing %q", required)
|
|
}
|
|
}
|
|
}
|