diff options
Diffstat (limited to 'internal/storage/db_test.go')
| -rw-r--r-- | internal/storage/db_test.go | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/internal/storage/db_test.go b/internal/storage/db_test.go index e34f88d..74aa30d 100644 --- a/internal/storage/db_test.go +++ b/internal/storage/db_test.go @@ -7,6 +7,7 @@ import ( "path/filepath" "testing" "testing/fstest" + "time" ) func TestOpen_ContextCanceled(t *testing.T) { @@ -464,3 +465,60 @@ func TestImportFromDir_pathIsFileNotDirectory(t *testing.T) { t.Fatal("expected error") } } + +func TestHostMetaRoundTrip(t *testing.T) { + tmpDir := t.TempDir() + dbPath := filepath.Join(tmpDir, "test.db") + db, err := Open(context.Background(), dbPath) + if err != nil { + t.Fatalf("open DB: %v", err) + } + defer db.Close() + ctx := context.Background() + if err := CreateSchema(ctx, db); err != nil { + t.Fatalf("schema: %v", err) + } + + if err := ResetHostMeta(ctx, db); err != nil { + t.Fatalf("reset: %v", err) + } + + tx, err := db.BeginTx(ctx, nil) + if err != nil { + t.Fatal(err) + } + if err := AddHostMeta(ctx, tx, "host1", 1705312200); err != nil { + t.Fatalf("add: %v", err) + } + if err := AddHostMeta(ctx, tx, "host2", 1705312300); err != nil { + t.Fatalf("add: %v", err) + } + if err := tx.Commit(); err != nil { + t.Fatal(err) + } + + meta, err := LoadHostMeta(ctx, db) + if err != nil { + t.Fatalf("load: %v", err) + } + if len(meta) != 2 { + t.Fatalf("len=%d, want 2", len(meta)) + } + if meta["host1"] != time.Unix(1705312200, 0).UTC() { + t.Fatalf("host1 time mismatch: %v", meta["host1"]) + } + if meta["host2"] != time.Unix(1705312300, 0).UTC() { + t.Fatalf("host2 time mismatch: %v", meta["host2"]) + } + + if err := ResetHostMeta(ctx, db); err != nil { + t.Fatalf("reset2: %v", err) + } + meta, err = LoadHostMeta(ctx, db) + if err != nil { + t.Fatalf("load after reset: %v", err) + } + if len(meta) != 0 { + t.Fatalf("len after reset=%d, want 0", len(meta)) + } +} |