127 lines
2.8 KiB
Go
127 lines
2.8 KiB
Go
package process
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"dappco.re/go/core"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestRegistry_Register_Good(t *testing.T) {
|
|
dir := t.TempDir()
|
|
reg := NewRegistry(dir)
|
|
|
|
started := time.Now().UTC().Truncate(time.Second)
|
|
entry := DaemonEntry{
|
|
Code: "myapp",
|
|
Daemon: "worker",
|
|
PID: os.Getpid(),
|
|
Health: "healthy",
|
|
Project: "test-project",
|
|
Binary: "/usr/bin/worker",
|
|
Started: started,
|
|
}
|
|
|
|
err := reg.Register(entry)
|
|
require.NoError(t, err)
|
|
|
|
got, ok := reg.Get("myapp", "worker")
|
|
require.True(t, ok)
|
|
assert.Equal(t, "myapp", got.Code)
|
|
assert.Equal(t, "worker", got.Daemon)
|
|
assert.Equal(t, os.Getpid(), got.PID)
|
|
assert.Equal(t, "healthy", got.Health)
|
|
assert.Equal(t, "test-project", got.Project)
|
|
assert.Equal(t, "/usr/bin/worker", got.Binary)
|
|
assert.Equal(t, started, got.Started)
|
|
}
|
|
|
|
func TestRegistry_Unregister_Good(t *testing.T) {
|
|
dir := t.TempDir()
|
|
reg := NewRegistry(dir)
|
|
|
|
entry := DaemonEntry{
|
|
Code: "myapp",
|
|
Daemon: "server",
|
|
PID: os.Getpid(),
|
|
}
|
|
|
|
err := reg.Register(entry)
|
|
require.NoError(t, err)
|
|
|
|
// File should exist
|
|
path := core.JoinPath(dir, "myapp-server.json")
|
|
_, err = os.Stat(path)
|
|
require.NoError(t, err)
|
|
|
|
err = reg.Unregister("myapp", "server")
|
|
require.NoError(t, err)
|
|
|
|
// File should be gone
|
|
_, err = os.Stat(path)
|
|
assert.True(t, os.IsNotExist(err))
|
|
}
|
|
|
|
func TestRegistry_List_Good(t *testing.T) {
|
|
dir := t.TempDir()
|
|
reg := NewRegistry(dir)
|
|
|
|
err := reg.Register(DaemonEntry{Code: "app1", Daemon: "web", PID: os.Getpid()})
|
|
require.NoError(t, err)
|
|
err = reg.Register(DaemonEntry{Code: "app2", Daemon: "api", PID: os.Getpid()})
|
|
require.NoError(t, err)
|
|
|
|
entries, err := reg.List()
|
|
require.NoError(t, err)
|
|
assert.Len(t, entries, 2)
|
|
}
|
|
|
|
func TestRegistry_PruneStale_Good(t *testing.T) {
|
|
dir := t.TempDir()
|
|
reg := NewRegistry(dir)
|
|
|
|
err := reg.Register(DaemonEntry{Code: "dead", Daemon: "proc", PID: 999999999})
|
|
require.NoError(t, err)
|
|
|
|
// File should exist before listing
|
|
path := core.JoinPath(dir, "dead-proc.json")
|
|
_, err = os.Stat(path)
|
|
require.NoError(t, err)
|
|
|
|
entries, err := reg.List()
|
|
require.NoError(t, err)
|
|
assert.Empty(t, entries)
|
|
|
|
// Stale file should be removed
|
|
_, err = os.Stat(path)
|
|
assert.True(t, os.IsNotExist(err))
|
|
}
|
|
|
|
func TestRegistry_GetMissing_Bad(t *testing.T) {
|
|
dir := t.TempDir()
|
|
reg := NewRegistry(dir)
|
|
|
|
got, ok := reg.Get("nope", "missing")
|
|
assert.Nil(t, got)
|
|
assert.False(t, ok)
|
|
}
|
|
|
|
func TestRegistry_CreateDirectory_Good(t *testing.T) {
|
|
dir := core.JoinPath(t.TempDir(), "nested", "deep", "daemons")
|
|
reg := NewRegistry(dir)
|
|
|
|
err := reg.Register(DaemonEntry{Code: "app", Daemon: "srv", PID: os.Getpid()})
|
|
require.NoError(t, err)
|
|
|
|
info, err := os.Stat(dir)
|
|
require.NoError(t, err)
|
|
assert.True(t, info.IsDir())
|
|
}
|
|
|
|
func TestRegistry_Default_Good(t *testing.T) {
|
|
reg := DefaultRegistry()
|
|
assert.NotNil(t, reg)
|
|
}
|