wehub-resource-sync
a06f331eb8
CI / benchmark (push) Has been skipped
install-script / posix-syntax (push) Successful in 6m1s
CI / build-onnx (push) Failing after 6m43s
init-smoke / dry-run (push) Failing after 15m57s
security / govulncheck (push) Has been cancelled
security / trivy-fs (push) Has been cancelled
CI / test (1.26, ubuntu-latest) (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
CI / test (1.26, macos-latest) (push) Has been cancelled
CI / build-windows (push) Has been cancelled
CI / lint (push) Has been cancelled
install-script / powershell-syntax (push) Has been cancelled
install-script / install (macos-14) (push) Has been cancelled
install-script / install (ubuntu-latest) (push) Has been cancelled
75 lines
2.5 KiB
Go
75 lines
2.5 KiB
Go
package daemon
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// startWarmingTestServer brings up a daemon on a temp socket with the given
|
|
// optional Ready probe and returns the live socket path.
|
|
func startWarmingTestServer(t *testing.T, ready func() (bool, string)) string {
|
|
t.Helper()
|
|
dir, err := os.MkdirTemp("/tmp", "gx-warm")
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { _ = os.RemoveAll(dir) })
|
|
socket := filepath.Join(dir, "s")
|
|
t.Setenv("GORTEX_DAEMON_SOCKET", socket)
|
|
t.Setenv("GORTEX_DAEMON_PIDFILE", filepath.Join(dir, "p"))
|
|
|
|
srv := New(socket, "test", zap.NewNop())
|
|
srv.Controller = &fakeController{}
|
|
srv.Ready = ready
|
|
require.NoError(t, srv.Listen())
|
|
go func() { _ = srv.Serve() }()
|
|
t.Cleanup(func() { _ = srv.Shutdown() })
|
|
|
|
require.Eventually(t, func() bool { return IsRunningAt(socket) },
|
|
2*time.Second, 10*time.Millisecond)
|
|
return socket
|
|
}
|
|
|
|
// TestHandshake_StampsWarmingState proves the daemon reports its warmup state
|
|
// on the handshake ack so a connecting proxy / CLI can tell a still-filling
|
|
// graph from a ready one instead of guessing (and dead-ending on empty).
|
|
func TestHandshake_StampsWarmingState(t *testing.T) {
|
|
t.Run("warming", func(t *testing.T) {
|
|
socket := startWarmingTestServer(t, func() (bool, string) {
|
|
return false, "global_resolve"
|
|
})
|
|
client, err := DialTo(socket, Handshake{Mode: ModeControl, ClientName: "cli"})
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
assert.True(t, client.Ack.OK)
|
|
assert.True(t, client.Ack.Warming, "ack should flag a still-warming daemon")
|
|
assert.Equal(t, "global_resolve", client.Ack.WarmupPhase)
|
|
})
|
|
|
|
t.Run("ready", func(t *testing.T) {
|
|
socket := startWarmingTestServer(t, func() (bool, string) {
|
|
return true, "ready"
|
|
})
|
|
client, err := DialTo(socket, Handshake{Mode: ModeControl, ClientName: "cli"})
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
assert.True(t, client.Ack.OK)
|
|
assert.False(t, client.Ack.Warming, "ack should not flag a ready daemon as warming")
|
|
assert.Equal(t, "ready", client.Ack.WarmupPhase)
|
|
})
|
|
|
|
t.Run("no probe assumes ready", func(t *testing.T) {
|
|
socket := startWarmingTestServer(t, nil)
|
|
client, err := DialTo(socket, Handshake{Mode: ModeControl, ClientName: "cli"})
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
assert.True(t, client.Ack.OK)
|
|
assert.False(t, client.Ack.Warming, "a nil Ready probe means assume ready")
|
|
assert.Empty(t, client.Ack.WarmupPhase)
|
|
})
|
|
}
|