yms-daemon/main_test.go

package main

import (
	"bytes"
	"context"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"yms-daemon/internal/daemonapi"
)

func TestParseUpdateArgsAcceptsBackendAndResolvesFile(t *testing.T) {
	file := filepath.Join(t.TempDir(), "package.zip")
	if err := os.WriteFile(file, []byte("zip"), 0o600); err != nil {
		t.Fatalf("write update package: %v", err)
	}
	request, err := parseUpdateArgs([]string{"--service", "backend", "-f", file}, &bytes.Buffer{})
	if err != nil {
		t.Fatalf("parse backend update arguments: %v", err)
	}
	if request.service != "backend" || request.inputType != daemonapi.InputTypeRepackZIP || request.file != file {
		t.Fatalf("unexpected backend update arguments: %+v", request)
	}
}

func TestParseUpdateArgsAcceptsDirectNativeJAR(t *testing.T) {
	file := filepath.Join(t.TempDir(), "glory-soft-yms.jar")
	if err := os.WriteFile(file, []byte("jar"), 0o600); err != nil {
		t.Fatalf("write native backend JAR: %v", err)
	}
	request, err := parseUpdateArgs([]string{"--service", "backend", "--native-jar", file}, &bytes.Buffer{})
	if err != nil {
		t.Fatalf("parse direct native backend update arguments: %v", err)
	}
	if request.service != "backend" || request.inputType != daemonapi.InputTypeNativeJAR || request.file != file {
		t.Fatalf("unexpected direct native backend update arguments: %+v", request)
	}
}

func TestParseUpdateArgsAcceptsContainerImage(t *testing.T) {
	imageReference := "harbor.ymswell.asia/ymswell/glory-ymswell:20260813-184902-a37bf50d-v1.1.8.1"
	request, err := parseUpdateArgs([]string{"--service", "backend", "--container-image", imageReference}, &bytes.Buffer{})
	if err != nil {
		t.Fatalf("parse container backend update arguments: %v", err)
	}
	if request.service != "backend" || request.inputType != daemonapi.InputTypeContainerImage || request.imageReference != imageReference || request.file != "" {
		t.Fatalf("unexpected container backend update arguments: %+v", request)
	}
}

func TestParseUpdateArgsControlsContainerStartupLogs(t *testing.T) {
	imageReference := "harbor.ymswell.asia/ymswell/glory-ymswell:20260813-184902-a37bf50d-v1.1.8.1"
	request, err := parseUpdateArgs([]string{"--service", "backend", "--container-image", imageReference, "--no-start-log"}, &bytes.Buffer{})
	if err != nil {
		t.Fatalf("parse no-start-log container update arguments: %v", err)
	}
	if request.startLog {
		t.Fatal("--no-start-log did not disable startup logs")
	}
	request, err = parseUpdateArgs([]string{"--service", "backend", "--container-image", imageReference}, &bytes.Buffer{})
	if err != nil {
		t.Fatalf("parse default container update arguments: %v", err)
	}
	if !request.startLog {
		t.Fatal("container startup logs are not enabled by default")
	}
}

func TestParseUpdateArgsRejectsNativeJARAndContainerImageTogether(t *testing.T) {
	jarPath := filepath.Join(t.TempDir(), "glory-soft-yms.jar")
	arguments := []string{
		"--service", "backend",
		"--native-jar", jarPath,
		"--container-image", "harbor.ymswell.asia/ymswell/glory-ymswell:20260813-184902-a37bf50d-v1.1.8.1",
	}
	_, err := parseUpdateArgs(arguments, &bytes.Buffer{})
	if err == nil || !strings.Contains(err.Error(), "exactly one") {
		t.Fatalf("expected mutually exclusive backend inputs, got %v", err)
	}
}

func TestParseUpdateArgsAcceptsQuite(t *testing.T) {
	file := filepath.Join(t.TempDir(), "glory-soft-yms.jar")
	if err := os.WriteFile(file, []byte("jar"), 0o600); err != nil {
		t.Fatalf("write native backend JAR: %v", err)
	}
	request, err := parseUpdateArgs([]string{"--service", "backend", "--native-jar", file, "--quite"}, &bytes.Buffer{})
	if err != nil {
		t.Fatalf("parse quiet direct update arguments: %v", err)
	}
	if !request.quite {
		t.Fatal("--quite was not retained")
	}
}

func TestWriteUpdateProgress(t *testing.T) {
	var output bytes.Buffer
	writeUpdateProgress(&output, daemonapi.Response{State: "STARTING", Message: "Starting backend"})
	if output.String() != "STARTING       Starting backend\n" {
		t.Fatalf("unexpected update progress: %q", output.String())
	}
}

func TestParseRestartArgs(t *testing.T) {
	request, err := parseRestartArgs([]string{"--service", "backend", "--quite"}, &bytes.Buffer{})
	if err != nil {
		t.Fatalf("parse backend restart arguments: %v", err)
	}
	if request.service != serviceBackend || !request.quite {
		t.Fatalf("unexpected backend restart arguments: %+v", request)
	}
}

func TestParseRestartArgsRejectsUnsupportedInput(t *testing.T) {
	for name, arguments := range map[string][]string{
		"missing service": nil,
		"other service":   {"--service", "frontend"},
		"positional":      {"--service", "backend", "extra"},
	} {
		t.Run(name, func(t *testing.T) {
			if _, err := parseRestartArgs(arguments, &bytes.Buffer{}); err == nil {
				t.Fatal("expected restart argument rejection")
			}
		})
	}
}

func TestParseUpdateArgsRejectsIncompleteOrUnsupportedInput(t *testing.T) {
	for name, arguments := range map[string][]string{
		"missing service":      {"-f", "/tmp/package.zip"},
		"missing file":         {"--service", "backend"},
		"other service":        {"--service", "frontend", "-f", "/tmp/package.zip"},
		"positional":           {"--service", "backend", "-f", "/tmp/package.zip", "extra"},
		"both inputs":          {"--service", "backend", "-f", "/tmp/package.zip", "--native-jar", "/tmp/backend.jar"},
		"native and container": {"--service", "backend", "--native-jar", "/tmp/backend.jar", "--container-image", "repository:tag"},
	} {
		t.Run(name, func(t *testing.T) {
			if _, err := parseUpdateArgs(arguments, &bytes.Buffer{}); err == nil {
				t.Fatal("expected update argument rejection")
			}
		})
	}
}

func TestRunPrintsUsageWithoutCommand(t *testing.T) {
	var stderr bytes.Buffer
	if exitCode := run(context.Background(), nil, &bytes.Buffer{}, &stderr); exitCode != 2 {
		t.Fatalf("unexpected exit code: %d", exitCode)
	}
	if stderr.Len() == 0 {
		t.Fatal("missing usage output")
	}
}