example_test.go

// Copyright 2026 HAProxy Technologies LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//	http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package serviceconditions

import (
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"

	"github.com/haproxytech/gopherd/internal/doctest"
)

// First run: the file is missing, the oneshot creates it, the dependent
// starts. Second run: the file exists, the oneshot is skipped, the dependent
// still starts (skip = satisfied), and status/start report the skip.
func TestServiceConditions(t *testing.T) {
	data, err := os.ReadFile("example.yml")
	if err != nil {
		t.Fatal(err)
	}
	auxFile := filepath.Join(t.TempDir(), "haproxy-aux.cfg")
	cfg := strings.ReplaceAll(string(data), "AUXFILE", auxFile)

	// Run 1: condition met (file missing) — the oneshot runs.
	d := doctest.RunConfig(t, cfg, doctest.Options{})
	d.WaitRunning("app", 5*time.Second)
	if _, err := os.Stat(auxFile); err != nil {
		t.Fatalf("aux file not created: %v", err)
	}
	if out := d.Output(); !strings.Contains(out, "oneshot aux-cfg completed") {
		t.Errorf("run 1 output missing oneshot completion:\n%s", out)
	}
	if code := d.Stop(); code != 0 {
		t.Fatalf("run 1: expected clean exit 0, got %d", code)
	}

	// Run 2: condition unmet (file exists) — the oneshot is skipped.
	d = doctest.RunConfig(t, cfg, doctest.Options{})
	d.WaitRunning("app", 5*time.Second)
	if out := d.Output(); !strings.Contains(out, "aux-cfg skipped (condition-file-missing: "+auxFile+" exists)") {
		t.Errorf("run 2 output missing skip line:\n%s", out)
	}

	if got := d.Command("status aux-cfg"); !strings.Contains(got, "skipped (condition-file-missing") {
		t.Errorf("status = %q, want skipped with reason", got)
	}
	// The overview and JSON agree with the single-service line: not "pending".
	if got := d.Command("status"); !strings.Contains(got, "aux-cfg") || !strings.Contains(got, "skipped") || strings.Contains(got, "pending") {
		t.Errorf("status overview = %q, want aux-cfg skipped", got)
	}
	if got := d.Command("status aux-cfg -o json"); !strings.Contains(got, `"state":"skipped"`) || !strings.Contains(got, `"reason":"condition-file-missing: `) {
		t.Errorf("status json = %q, want skipped with reason", got)
	}
	if got := d.Command("start aux-cfg"); !strings.Contains(got, "skipped (condition-file-missing") {
		t.Errorf("start = %q, want skip report", got)
	}

	if code := d.Stop(); code != 0 {
		t.Errorf("run 2: expected clean exit 0, got %d", code)
	}
}

// Boots one config twice: excluded by ROLE, then included via the inherited environment.
func TestServiceConditionsEnv(t *testing.T) {
	cfg := `
processes:
  - name: app
    command: /bin/sh
    args: ["-c", "exec sleep 300"]
    condition-env-equals:
      ROLE: api
      ENABLE_FEATURE: "true"
    on-failure: shutdown
    on-check-failure:
      app-health: restart
  - name: always
    command: /bin/sh
    args: ["-c", "exec sleep 300"]
    after: [app]
    on-failure: shutdown
checks:
  app-health:
    tcp: {host: 127.0.0.1, port: 1}
    period: 100ms
    timeout: 100ms
    threshold: 1
`

	// Run 1: excluded. The daemon has no such service; `always` lost its edge.
	t.Setenv("ROLE", "worker")
	t.Setenv("ENABLE_FEATURE", "true")
	d1 := doctest.RunConfig(t, cfg, doctest.Options{})
	d1.WaitRunning("always", 5*time.Second)
	if got := d1.Command("status"); strings.Contains(got, "app ") {
		t.Errorf("status lists the excluded service:\n%s", got)
	}
	if got := d1.Command("status app"); !strings.Contains(got, "error: unknown service") {
		t.Errorf("status app = %q, want unknown service", got)
	}
	if got := d1.Command("start app"); !strings.Contains(got, "error: unknown service") {
		t.Errorf("start app = %q, want unknown service", got)
	}
	if !strings.Contains(d1.Output(), `app excluded (condition-env-equals: ROLE does not match "api")`) {
		t.Errorf("daemon log lacks the exclusion line:\n%s", d1.Output())
	}
	// Its check went with it: no probe against a service that is not here.
	if got := d1.Command("status"); strings.Contains(got, "app-health") {
		t.Errorf("status still lists the orphaned check:\n%s", got)
	}
	if !strings.Contains(d1.Output(), "check app-health excluded (only used by excluded service app)") {
		t.Errorf("daemon log lacks the check exclusion line:\n%s", d1.Output())
	}
	if code := d1.Stop(); code != 0 {
		t.Errorf("expected clean exit 0, got %d", code)
	}

	// Run 2: met.
	t.Setenv("ROLE", "api")
	t.Setenv("ENABLE_FEATURE", "true")
	d2 := doctest.RunConfig(t, cfg, doctest.Options{})
	d2.WaitRunning("app", 5*time.Second)
	if gotRunning := d2.Command("status app"); !strings.Contains(gotRunning, "running") {
		t.Errorf("status = %q, want running", gotRunning)
	}
	if code := d2.Stop(); code != 0 {
		t.Errorf("expected clean exit 0, got %d", code)
	}
}