mirror of
https://github.com/rcourtman/Pulse.git
synced 2026-02-18 00:17:39 +01:00
44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
package updatedetection
|
|
|
|
import "testing"
|
|
|
|
func TestStore_DeleteUpdateNotFound(t *testing.T) {
|
|
store := NewStore()
|
|
store.DeleteUpdate("missing")
|
|
if store.Count() != 0 {
|
|
t.Fatalf("expected empty store")
|
|
}
|
|
}
|
|
|
|
func TestStore_DeleteUpdatesForResourceMissing(t *testing.T) {
|
|
store := NewStore()
|
|
store.DeleteUpdatesForResource("missing")
|
|
if store.Count() != 0 {
|
|
t.Fatalf("expected empty store")
|
|
}
|
|
}
|
|
|
|
func TestStore_DeleteUpdatesForResourceNilUpdate(t *testing.T) {
|
|
store := NewStore()
|
|
store.byResource["res-1"] = "update-1"
|
|
store.updates["update-1"] = nil
|
|
|
|
store.DeleteUpdatesForResource("res-1")
|
|
if _, ok := store.byResource["res-1"]; ok {
|
|
t.Fatalf("expected byResource entry to be removed")
|
|
}
|
|
}
|
|
|
|
func TestStore_CountForHost(t *testing.T) {
|
|
store := NewStore()
|
|
store.UpsertUpdate(&UpdateInfo{ID: "update-1", ResourceID: "res-1", HostID: "host-1"})
|
|
store.UpsertUpdate(&UpdateInfo{ID: "update-2", ResourceID: "res-2", HostID: "host-1"})
|
|
|
|
if store.CountForHost("host-1") != 2 {
|
|
t.Fatalf("expected count 2 for host-1")
|
|
}
|
|
if store.CountForHost("missing") != 0 {
|
|
t.Fatalf("expected count 0 for missing host")
|
|
}
|
|
}
|