Add core modules (SSH args parser, cache, resolver, NetBox client) with tests

internal/cache/cache_test.go

@@ -0,0 +1,235 @@
+package cache
+
+import (
+	"encoding/json"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+)
+
+func TestNew(t *testing.T) {
+	c := New("/tmp/test.json", 60)
+	if c == nil {
+		t.Fatal("New returned nil")
+	}
+	if c.ttl != 60*time.Second {
+		t.Errorf("ttl: got %v, want %v", c.ttl, 60*time.Second)
+	}
+}
+
+func TestLoad_MissingFile(t *testing.T) {
+	c := New("/nonexistent/path/cache.json", 60)
+	if err := c.Load(); err != nil {
+		t.Errorf("Load on missing file should not error, got: %v", err)
+	}
+}
+
+func TestLoad_InvalidJSON(t *testing.T) {
+	f := tempFile(t, []byte("not json"))
+	c := New(f, 60)
+	if err := c.Load(); err == nil {
+		t.Error("Load on invalid JSON should return an error")
+	}
+}
+
+func TestSaveAndLoad_Roundtrip(t *testing.T) {
+	path := filepath.Join(t.TempDir(), "cache.json")
+	c := New(path, 3600)
+
+	c.Upsert(Entry{Name: "host-a", IP: "10.0.0.1", Kind: "device"})
+	c.Upsert(Entry{Name: "host-b", IP: "10.0.0.2", Kind: "vm", Tags: []string{"prod"}})
+
+	if err := c.Save(); err != nil {
+		t.Fatalf("Save: %v", err)
+	}
+
+	c2 := New(path, 3600)
+	if err := c2.Load(); err != nil {
+		t.Fatalf("Load: %v", err)
+	}
+
+	e, _ := c2.Get("host-a")
+	if e.IP != "10.0.0.1" {
+		t.Errorf("host-a IP: got %q, want %q", e.IP, "10.0.0.1")
+	}
+	e2, _ := c2.Get("host-b")
+	if len(e2.Tags) != 1 || e2.Tags[0] != "prod" {
+		t.Errorf("host-b tags: got %v, want [prod]", e2.Tags)
+	}
+}
+
+func TestSave_CreatesDirectory(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "sub", "dir", "cache.json")
+	c := New(path, 60)
+	c.Upsert(Entry{Name: "x", IP: "1.2.3.4", Kind: "device"})
+	if err := c.Save(); err != nil {
+		t.Fatalf("Save: %v", err)
+	}
+	if _, err := os.Stat(path); err != nil {
+		t.Errorf("cache file not created: %v", err)
+	}
+}
+
+func TestUpsert_SetsTimestamp(t *testing.T) {
+	c := New("", 60)
+	before := time.Now()
+	c.Upsert(Entry{Name: "h", IP: "1.1.1.1", Kind: "device"})
+	e, _ := c.Get("h")
+	if e.CachedAt.Before(before) {
+		t.Error("CachedAt should be set to current time on Upsert")
+	}
+}
+
+func TestUpsert_Overwrites(t *testing.T) {
+	c := New("", 60)
+	c.Upsert(Entry{Name: "host", IP: "10.0.0.1", Kind: "device"})
+	c.Upsert(Entry{Name: "host", IP: "10.0.0.2", Kind: "device"})
+	e, _ := c.Get("host")
+	if e.IP != "10.0.0.2" {
+		t.Errorf("Upsert should overwrite: got %q, want %q", e.IP, "10.0.0.2")
+	}
+}
+
+func TestSearch_PrefixMatch(t *testing.T) {
+	c := New("", 60)
+	c.Upsert(Entry{Name: "app-server-01", IP: "10.0.0.1", Kind: "device"})
+	c.Upsert(Entry{Name: "app-server-02", IP: "10.0.0.2", Kind: "vm"})
+	c.Upsert(Entry{Name: "db-server-01", IP: "10.0.0.3", Kind: "device"})
+
+	results := c.Search("app")
+	if len(results) != 2 {
+		t.Errorf("Search(app): got %d results, want 2", len(results))
+	}
+}
+
+func TestSearch_CaseInsensitive(t *testing.T) {
+	c := New("", 60)
+	c.Upsert(Entry{Name: "App-Server", IP: "10.0.0.1", Kind: "device"})
+
+	if len(c.Search("app")) != 1 {
+		t.Error("Search should be case-insensitive")
+	}
+	if len(c.Search("APP")) != 1 {
+		t.Error("Search should be case-insensitive for uppercase")
+	}
+}
+
+func TestSearch_EmptyPrefix(t *testing.T) {
+	c := New("", 60)
+	c.Upsert(Entry{Name: "a", IP: "1.1.1.1", Kind: "device"})
+	c.Upsert(Entry{Name: "b", IP: "2.2.2.2", Kind: "vm"})
+
+	if len(c.Search("")) != 2 {
+		t.Error("Search('') should return all entries")
+	}
+}
+
+func TestSearch_NoMatch(t *testing.T) {
+	c := New("", 60)
+	c.Upsert(Entry{Name: "host", IP: "1.1.1.1", Kind: "device"})
+
+	if len(c.Search("xyz")) != 0 {
+		t.Error("Search should return empty slice when no match")
+	}
+}
+
+func TestGet_Fresh(t *testing.T) {
+	c := New("", 3600)
+	c.Upsert(Entry{Name: "host", IP: "10.0.0.1", Kind: "device"})
+
+	e, fresh := c.Get("host")
+	if !fresh {
+		t.Error("entry just inserted should be fresh")
+	}
+	if e.IP != "10.0.0.1" {
+		t.Errorf("IP: got %q, want %q", e.IP, "10.0.0.1")
+	}
+}
+
+func TestGet_Expired(t *testing.T) {
+	c := New("", 1) // 1 second TTL
+	e := Entry{Name: "host", IP: "10.0.0.1", Kind: "device", CachedAt: time.Now().Add(-2 * time.Second)}
+	c.mu.Lock()
+	c.entries["host"] = e
+	c.mu.Unlock()
+
+	_, fresh := c.Get("host")
+	if fresh {
+		t.Error("entry older than TTL should not be fresh")
+	}
+}
+
+func TestGet_ZeroTTL_AlwaysStale(t *testing.T) {
+	c := New("", 0) // TTL=0 means never fresh for connect mode
+	c.Upsert(Entry{Name: "host", IP: "10.0.0.1", Kind: "device"})
+
+	_, fresh := c.Get("host")
+	if fresh {
+		t.Error("TTL=0 should always return fresh=false")
+	}
+}
+
+func TestGet_Missing(t *testing.T) {
+	c := New("", 60)
+	_, fresh := c.Get("nonexistent")
+	if fresh {
+		t.Error("missing entry should not be fresh")
+	}
+}
+
+func TestClear(t *testing.T) {
+	c := New("", 60)
+	c.Upsert(Entry{Name: "a", IP: "1.1.1.1", Kind: "device"})
+	c.Upsert(Entry{Name: "b", IP: "2.2.2.2", Kind: "vm"})
+	c.Clear()
+
+	if len(c.All()) != 0 {
+		t.Error("Clear should remove all entries")
+	}
+}
+
+func TestAll(t *testing.T) {
+	c := New("", 60)
+	c.Upsert(Entry{Name: "a", IP: "1.1.1.1", Kind: "device"})
+	c.Upsert(Entry{Name: "b", IP: "2.2.2.2", Kind: "vm"})
+
+	all := c.All()
+	if len(all) != 2 {
+		t.Errorf("All: got %d entries, want 2", len(all))
+	}
+}
+
+func TestSave_ProducesValidJSON(t *testing.T) {
+	path := filepath.Join(t.TempDir(), "cache.json")
+	c := New(path, 60)
+	c.Upsert(Entry{Name: "host", IP: "10.0.0.1", Kind: "device", Tags: []string{"mgmt"}})
+
+	if err := c.Save(); err != nil {
+		t.Fatalf("Save: %v", err)
+	}
+
+	data, _ := os.ReadFile(path)
+	var df diskFormat
+	if err := json.Unmarshal(data, &df); err != nil {
+		t.Fatalf("saved file is not valid JSON: %v", err)
+	}
+	if len(df.Entries) != 1 {
+		t.Errorf("expected 1 entry in JSON, got %d", len(df.Entries))
+	}
+}
+
+// tempFile writes content to a temp file and returns its path.
+func tempFile(t *testing.T, content []byte) string {
+	t.Helper()
+	f, err := os.CreateTemp(t.TempDir(), "cache-*.json")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if _, err := f.Write(content); err != nil {
+		t.Fatal(err)
+	}
+	f.Close()
+	return f.Name()
+}