52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestDeviceLifecyclePersistence(t *testing.T) {
|
|
s, err := Open(filepath.Join(t.TempDir(), "commander.db"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer s.DB.Close()
|
|
|
|
if err := s.UpsertRemote("node1", "online", "amd64", "linux", 100, "{\"role\":\"APPLICATION\"}"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rows, err := s.ListDevices()
|
|
if err != nil || len(rows) != 1 {
|
|
t.Fatalf("rows=%d err=%v", len(rows), err)
|
|
}
|
|
if rows[0].Status != "online" || rows[0].Arch != "amd64" || rows[0].OS != "linux" {
|
|
t.Fatalf("unexpected row: %+v", rows[0])
|
|
}
|
|
|
|
if err := s.UpdateDeviceStatus("node1", "offline", 120); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rows, err = s.ListDevices()
|
|
if err != nil || len(rows) != 1 {
|
|
t.Fatalf("rows=%d err=%v", len(rows), err)
|
|
}
|
|
if rows[0].Status != "offline" || rows[0].OfflineAt == nil {
|
|
t.Fatalf("offline state not persisted: %+v", rows[0])
|
|
}
|
|
|
|
if err := s.RevokeDevice("node1", "test"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rows, err = s.ListDevices()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(rows) != 0 {
|
|
t.Fatalf("revoked device still present: %+v", rows)
|
|
}
|
|
revoked, err := s.IsRevoked("node1")
|
|
if err != nil || !revoked {
|
|
t.Fatalf("revoked=%v err=%v", revoked, err)
|
|
}
|
|
}
|