cli/dockercompose/ai_test.go

Rama main · réplica de lectura del repositorio original.

package dockercompose //nolint:testpackage

import (
	"testing"

	"github.com/google/go-cmp/cmp"
	"github.com/nhost/be/services/mimir/model"
)

func expectedAI() *Service {
	return &Service{
		Image: "nhost/graphite:0.2.5",
		DependsOn: map[string]DependsOn{
			"auth":     {Condition: "service_healthy"},
			"graphql":  {Condition: "service_healthy"},
			"postgres": {Condition: "service_healthy"},
		},
		EntryPoint: nil,
		Command:    []string{"serve"},
		Environment: map[string]string{
			"ENV1":                        "VALUE1",
			"ENV2":                        "VALUE2",
			"GRAPHITE_BASE_URL":           "http://ai:8090",
			"GRAPHITE_WEBHOOK_SECRET":     "webhookSecret",
			"HASURA_GRAPHQL_ADMIN_SECRET": "adminSecret",
			"NHOST_STORAGE_URL":           "http://storage:5000/v1",
			"LICENSE":                     "",
			"NHOST_GRAPHQL_URL":           "http://graphql:8080/v1/graphql",
			"OPENAI_API_KEY":              "openaiApiKey",
			"OPENAI_ORG":                  "my-org",
			"POSTGRES_CONNECTION":         "postgres://postgres@postgres:5432/local?sslmode=disable",
			"SYNCH_PERIOD":                "10m",
		},
		ExtraHosts: []string{
			"host.docker.internal:host-gateway",
		},
		HealthCheck: &HealthCheck{
			Test:        []string{"CMD", "graphite", "healthcheck"},
			Timeout:     "60s",
			Interval:    "5s",
			StartPeriod: "10s",
		},
		Labels:     nil,
		Networks:   nil,
		Ports:      nil,
		Restart:    "always",
		Volumes:    nil,
		WorkingDir: nil,
	}
}

func TestAI(t *testing.T) {
	t.Parallel()

	cases := []struct {
		name     string
		cfg      func() *model.ConfigConfig
		useTlS   bool
		expected func() *Service
	}{
		{
			name:     "success",
			cfg:      getConfig,
			useTlS:   false,
			expected: expectedAI,
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			got := ai(tc.cfg())
			if diff := cmp.Diff(tc.expected(), got); diff != "" {
				t.Error(diff)
			}
		})
	}
}