mirror of
https://github.com/hanzoai/rollout.git
synced 2026-10-30 18:23:14 +00:00
zeekay
0ce9caa5d3
Canonical source: ~/work/hanzo/operator/internal/rollout/. Both ~/work/hanzo/operator and ~/work/liquidity/operator (Go side) will consume this module to remove ~830 lines of duplication. Test fixtures are brand-neutral. Liquidity-specific manifest security tests stay in ~/work/liquidity/operator.
142 lines
3.2 KiB
Go
142 lines
3.2 KiB
Go
package rollout
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestServiceTag(t *testing.T) {
|
|
tests := []struct {
|
|
image string
|
|
want string
|
|
}{
|
|
{"example.com/foo:v1.19.4", "v1.19.4"},
|
|
{"example.com/bar:v1.0.0", "v1.0.0"},
|
|
{"no-tag", ""},
|
|
}
|
|
for _, tt := range tests {
|
|
s := Service{Image: tt.image}
|
|
if got := s.Tag(); got != tt.want {
|
|
t.Errorf("Tag(%q) = %q, want %q", tt.image, got, tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestValidateTag(t *testing.T) {
|
|
good := []string{"v1.0.0", "v1.19.4", "v0.0.1"}
|
|
for _, tag := range good {
|
|
s := Service{Image: "foo:" + tag}
|
|
if err := s.ValidateTag("test"); err != nil {
|
|
t.Errorf("ValidateTag(%q) unexpected error: %v", tag, err)
|
|
}
|
|
}
|
|
|
|
bad := []string{"dev", "main", "latest", "abc123", "1.0.0"}
|
|
for _, tag := range bad {
|
|
s := Service{Image: "foo:" + tag}
|
|
if err := s.ValidateTag("test"); err == nil {
|
|
t.Errorf("ValidateTag(%q) expected error, got nil", tag)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsBaseService(t *testing.T) {
|
|
for _, name := range []string{"ats", "bd", "ta"} {
|
|
if !IsBaseService(name) {
|
|
t.Errorf("IsBaseService(%q) = false, want true", name)
|
|
}
|
|
}
|
|
for _, name := range []string{"gateway", "exchange", "superadmin", "kms"} {
|
|
if IsBaseService(name) {
|
|
t.Errorf("IsBaseService(%q) = true, want false", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestK8sKind(t *testing.T) {
|
|
if s := (Service{Kind: "statefulset"}); s.K8sKind() != "sts" {
|
|
t.Errorf("K8sKind(statefulset) = %q, want sts", s.K8sKind())
|
|
}
|
|
if s := (Service{Kind: "deployment"}); s.K8sKind() != "deploy" {
|
|
t.Errorf("K8sKind(deployment) = %q, want deploy", s.K8sKind())
|
|
}
|
|
}
|
|
|
|
func TestLoadManifest(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
content := `version: "1"
|
|
context: kind-test-devnet
|
|
project: acme-devnet
|
|
domain: dev.example.com
|
|
replicas: 1
|
|
|
|
services:
|
|
ats:
|
|
image: example.com/ats:v1.19.4
|
|
namespace: acme
|
|
kind: statefulset
|
|
gateway:
|
|
image: example.com/gateway:v1.0.2
|
|
namespace: acme
|
|
kind: deployment
|
|
`
|
|
if err := os.WriteFile(filepath.Join(dir, "dev.yml"), []byte(content), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
m, err := LoadManifest(dir, Devnet)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if m.Context != "kind-test-devnet" {
|
|
t.Errorf("context = %q", m.Context)
|
|
}
|
|
if m.Replicas != 1 {
|
|
t.Errorf("replicas = %d", m.Replicas)
|
|
}
|
|
if len(m.Services) != 2 {
|
|
t.Errorf("services count = %d, want 2", len(m.Services))
|
|
}
|
|
|
|
ats := m.Services["ats"]
|
|
if ats.Kind != "statefulset" {
|
|
t.Errorf("ats kind = %q", ats.Kind)
|
|
}
|
|
if ats.Tag() != "v1.19.4" {
|
|
t.Errorf("ats tag = %q", ats.Tag())
|
|
}
|
|
if ats.Namespace != "acme" {
|
|
t.Errorf("ats namespace = %q", ats.Namespace)
|
|
}
|
|
}
|
|
|
|
func TestHealthPath(t *testing.T) {
|
|
if p := HealthPath("ats"); p != "/v1/base/health" {
|
|
t.Errorf("HealthPath(ats) = %q", p)
|
|
}
|
|
if p := HealthPath("gateway"); p != "/healthz" {
|
|
t.Errorf("HealthPath(gateway) = %q", p)
|
|
}
|
|
}
|
|
|
|
func TestHealthPort(t *testing.T) {
|
|
tests := map[string]int{
|
|
"ats": 8090,
|
|
"bd": 8090,
|
|
"ta": 8090,
|
|
"iam": 8000,
|
|
"kms": 8443,
|
|
"gateway": 8080,
|
|
"exchange": 3000,
|
|
"superadmin": 3000,
|
|
}
|
|
for name, want := range tests {
|
|
if got := HealthPort(name); got != want {
|
|
t.Errorf("HealthPort(%q) = %d, want %d", name, got, want)
|
|
}
|
|
}
|
|
}
|