3b01594638
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package app
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"log"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.az-it.net/az/morz-infoboard/player/agent/internal/config"
|
|
)
|
|
|
|
func TestAppRunUpdatesHealthAndLogsStructuredEvents(t *testing.T) {
|
|
var logBuffer bytes.Buffer
|
|
logger := log.New(&logBuffer, "", 0)
|
|
|
|
application := newApp(config.Config{
|
|
ScreenID: "info01-dev",
|
|
ServerBaseURL: "http://127.0.0.1:8080",
|
|
MQTTBroker: "tcp://127.0.0.1:1883",
|
|
HeartbeatEvery: 1,
|
|
}, logger, time.Now)
|
|
|
|
if got, want := application.Snapshot().Status, StatusStarting; got != want {
|
|
t.Fatalf("initial status = %q, want %q", got, want)
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
errCh := make(chan error, 1)
|
|
go func() {
|
|
errCh <- application.Run(ctx)
|
|
}()
|
|
|
|
deadline := time.Now().Add(2 * time.Second)
|
|
for time.Now().Before(deadline) {
|
|
snapshot := application.Snapshot()
|
|
if snapshot.Status == StatusRunning && !snapshot.LastHeartbeatAt.IsZero() {
|
|
break
|
|
}
|
|
|
|
time.Sleep(10 * time.Millisecond)
|
|
}
|
|
|
|
snapshot := application.Snapshot()
|
|
if got, want := snapshot.Status, StatusRunning; got != want {
|
|
t.Fatalf("running status = %q, want %q", got, want)
|
|
}
|
|
|
|
if snapshot.LastHeartbeatAt.IsZero() {
|
|
t.Fatal("LastHeartbeatAt = zero, want heartbeat timestamp")
|
|
}
|
|
|
|
cancel()
|
|
|
|
select {
|
|
case err := <-errCh:
|
|
if err != nil {
|
|
t.Fatalf("Run() error = %v", err)
|
|
}
|
|
case <-time.After(2 * time.Second):
|
|
t.Fatal("Run() did not return after cancel")
|
|
}
|
|
|
|
if got, want := application.Snapshot().Status, StatusStopped; got != want {
|
|
t.Fatalf("final status = %q, want %q", got, want)
|
|
}
|
|
|
|
logs := logBuffer.String()
|
|
for _, needle := range []string{
|
|
"event=agent_configured",
|
|
"screen_id=info01-dev",
|
|
"event=heartbeat_tick",
|
|
"event=agent_stopped",
|
|
} {
|
|
if !strings.Contains(logs, needle) {
|
|
t.Fatalf("logs missing %q: %s", needle, logs)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAppSnapshotIncludesConfiguredTargets(t *testing.T) {
|
|
application := newApp(config.Config{
|
|
ScreenID: "screen-77",
|
|
ServerBaseURL: "https://backend.example",
|
|
MQTTBroker: "tcp://mqtt.example:1883",
|
|
HeartbeatEvery: 15,
|
|
}, log.New(&bytes.Buffer{}, "", 0), time.Now)
|
|
|
|
snapshot := application.Snapshot()
|
|
|
|
if got, want := snapshot.ScreenID, "screen-77"; got != want {
|
|
t.Fatalf("ScreenID = %q, want %q", got, want)
|
|
}
|
|
|
|
if got, want := snapshot.ServerBaseURL, "https://backend.example"; got != want {
|
|
t.Fatalf("ServerBaseURL = %q, want %q", got, want)
|
|
}
|
|
|
|
if got, want := snapshot.MQTTBroker, "tcp://mqtt.example:1883"; got != want {
|
|
t.Fatalf("MQTTBroker = %q, want %q", got, want)
|
|
}
|
|
|
|
if got, want := snapshot.HeartbeatEvery, 15; got != want {
|
|
t.Fatalf("HeartbeatEvery = %d, want %d", got, want)
|
|
}
|
|
}
|