package engine_test import ( "context" "os" "path/filepath" "testing" "time" foxhound "github.com/sadewadee/foxhound" "github.com/sadewadee/foxhound/engine" ) // --------------------------------------------------------------------------- // TestSaveCheckpoint // --------------------------------------------------------------------------- // TestSaveCheckpoint verifies that SaveCheckpoint writes valid JSON that can // be decoded and contains the expected field values. func TestSaveCheckpoint(t *testing.T) { h := minimalHunt("save-cp-test") dir := t.TempDir() path := filepath.Join(dir, "checkpoint.json") if err := h.SaveCheckpoint(path); err != nil { t.Fatalf("SaveCheckpoint: %v", err) } // File must exist. if _, err := os.Stat(path); err != nil { t.Fatalf("checkpoint file not created: %v", err) } cp, err := engine.LoadCheckpoint(path) if err != nil { t.Fatalf("LoadCheckpoint: %v", err) } if cp.HuntName != "save-cp-test" { t.Errorf("HuntName: want %q, got %q", "save-cp-test", cp.HuntName) } if cp.Timestamp.IsZero() { t.Error("Timestamp must not be zero") } } // --------------------------------------------------------------------------- // TestLoadCheckpoint // --------------------------------------------------------------------------- // TestLoadCheckpoint verifies that LoadCheckpoint correctly deserialises a // checkpoint written by SaveCheckpoint. func TestLoadCheckpoint(t *testing.T) { h := minimalHunt("load-cp-test") dir := t.TempDir() path := filepath.Join(dir, "cp.json") if err := h.SaveCheckpoint(path); err != nil { t.Fatalf("SaveCheckpoint: %v", err) } cp, err := engine.LoadCheckpoint(path) if err != nil { t.Fatalf("LoadCheckpoint: %v", err) } if cp == nil { t.Fatal("LoadCheckpoint returned nil") } if cp.HuntName == "" { t.Error("HuntName must not be empty after round-trip") } } // --------------------------------------------------------------------------- // TestLoadCheckpoint_FileNotFound // --------------------------------------------------------------------------- // TestLoadCheckpoint_FileNotFound verifies that LoadCheckpoint returns an // error when the file does not exist. func TestLoadCheckpoint_FileNotFound(t *testing.T) { _, err := engine.LoadCheckpoint("/tmp/foxhound-nonexistent-checkpoint-xyz.json") if err == nil { t.Error("expected error for non-existent checkpoint file, got nil") } } // --------------------------------------------------------------------------- // TestCheckpointRoundtrip // --------------------------------------------------------------------------- // TestCheckpointRoundtrip saves a checkpoint, loads it back, and verifies // every exported field matches the original snapshot. func TestCheckpointRoundtrip(t *testing.T) { h := minimalHunt("roundtrip-test") // Simulate some stats by advancing counters. h.Stats().RecordItems(7) h.Stats().RecordRequest("example.com", 10*time.Millisecond, nil, false) h.Stats().RecordRequest("example.com", 5*time.Millisecond, nil, true) dir := t.TempDir() path := filepath.Join(dir, "rt.json") if err := h.SaveCheckpoint(path); err != nil { t.Fatalf("SaveCheckpoint: %v", err) } cp, err := engine.LoadCheckpoint(path) if err != nil { t.Fatalf("LoadCheckpoint: %v", err) } if cp.HuntName != "roundtrip-test" { t.Errorf("HuntName: want roundtrip-test, got %q", cp.HuntName) } if cp.ItemsProcessed != 7 { t.Errorf("ItemsProcessed: want 7, got %d", cp.ItemsProcessed) } if cp.RequestsDone != 2 { t.Errorf("RequestsDone: want 2, got %d", cp.RequestsDone) } if cp.Timestamp.IsZero() { t.Error("Timestamp must not be zero") } if cp.ElapsedMs < 0 { t.Errorf("ElapsedMs must be >= 0, got %d", cp.ElapsedMs) } } // --------------------------------------------------------------------------- // TestAutoCheckpoint // --------------------------------------------------------------------------- // cpHTMLBody is a minimal HTML body that bypasses the captcha detector's // empty_trap heuristic (which fires when body < 500 bytes and lacks