diff options
Diffstat (limited to 'internal/storage/db_test.go')
| -rw-r--r-- | internal/storage/db_test.go | 169 |
1 files changed, 169 insertions, 0 deletions
diff --git a/internal/storage/db_test.go b/internal/storage/db_test.go index 5415235..b05b4c7 100644 --- a/internal/storage/db_test.go +++ b/internal/storage/db_test.go @@ -6,6 +6,7 @@ import ( "os" "path/filepath" "testing" + "testing/fstest" ) func TestOpen_ContextCanceled(t *testing.T) { @@ -32,3 +33,171 @@ func TestOpen_PingFailsOnDirectoryPath(t *testing.T) { t.Fatal("expected error opening sqlite at directory path") } } + +func TestOpen_createsDatabaseFile(t *testing.T) { + tmpDir := t.TempDir() + dbPath := filepath.Join(tmpDir, "test.db") + + db, err := Open(context.Background(), dbPath) + if err != nil { + t.Fatalf("failed to open DB: %v", err) + } + defer db.Close() + + if db == nil { + t.Error("expected non-nil database") + } +} + +func TestCreateSchema(t *testing.T) { + tmpDir := t.TempDir() + dbPath := filepath.Join(tmpDir, "test.db") + + db, err := Open(context.Background(), dbPath) + if err != nil { + t.Fatalf("failed to open DB: %v", err) + } + defer db.Close() + + ctx := context.Background() + err = CreateSchema(ctx, db) + if err != nil { + t.Fatalf("failed to create schema: %v", err) + } + + _, err = db.ExecContext(ctx, "SELECT 1 FROM record LIMIT 1") + if err != nil { + t.Fatalf("failed to query record table: %v", err) + } +} + +func TestResetRecords(t *testing.T) { + tmpDir := t.TempDir() + dbPath := filepath.Join(tmpDir, "test.db") + + db, err := Open(context.Background(), dbPath) + if err != nil { + t.Fatalf("failed to open DB: %v", err) + } + defer db.Close() + + ctx := context.Background() + if err := CreateSchema(ctx, db); err != nil { + t.Fatalf("failed to create schema: %v", err) + } + + _, err = db.ExecContext(ctx, + "INSERT INTO record (host, uptime_sec, boot_time, os, os_kernel_name, os_kernel_major) VALUES (?, ?, ?, ?, ?, ?)", + "host1", 1000, 2000, "Linux 5.10", "Linux", "Linux 5...") + if err != nil { + t.Fatalf("failed to insert record: %v", err) + } + + err = ResetRecords(ctx, db) + if err != nil { + t.Fatalf("failed to reset records: %v", err) + } + + var count int + err = db.QueryRowContext(ctx, "SELECT COUNT(*) FROM record").Scan(&count) + if err != nil { + t.Fatalf("failed to count records: %v", err) + } + + if count != 0 { + t.Errorf("expected 0 records after reset, got %d", count) + } +} + +func TestImportFromDir(t *testing.T) { + tmpDir := t.TempDir() + + recordsFile := filepath.Join(tmpDir, "testhost.records") + content := []byte("86400:1000000:Linux 5.10.0-test\n" + + "86400:1000001:Linux 5.10.0-test\n" + + "86400:1000002:Linux 5.10.0-test\n") + + if err := os.WriteFile(recordsFile, content, 0644); err != nil { + t.Fatalf("failed to create test file: %v", err) + } + + dbPath := filepath.Join(tmpDir, "test.db") + db, err := Open(context.Background(), dbPath) + if err != nil { + t.Fatalf("failed to open DB: %v", err) + } + defer db.Close() + + ctx := context.Background() + if err := CreateSchema(ctx, db); err != nil { + t.Fatalf("failed to create schema: %v", err) + } + + err = ImportFromDir(ctx, db, tmpDir) + if err != nil { + t.Fatalf("failed to import records: %v", err) + } + + var count int + err = db.QueryRowContext(ctx, "SELECT COUNT(*) FROM record").Scan(&count) + if err != nil { + t.Fatalf("failed to count records: %v", err) + } + + if count != 3 { + t.Errorf("expected 3 records after import, got %d", count) + } +} + +func TestImportFromFS_MapFS(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) + } + m := fstest.MapFS{ + "testhost.records": &fstest.MapFile{ + Data: []byte("86400:1000000:Linux 5.10.0-test\n" + + "86400:1000001:Linux 5.10.0-test\n" + + "86400:1000002:Linux 5.10.0-test\n"), + Mode: 0o644, + }, + } + if err := ImportFromFS(ctx, db, m); err != nil { + t.Fatalf("ImportFromFS: %v", err) + } + var count int + if err := db.QueryRowContext(ctx, "SELECT COUNT(*) FROM record").Scan(&count); err != nil { + t.Fatal(err) + } + if count != 3 { + t.Errorf("count = %d, want 3", count) + } +} + +func TestImportFromDir_invalidPath(t *testing.T) { + tmpDir := t.TempDir() + dbPath := filepath.Join(tmpDir, "test.db") + + db, err := Open(context.Background(), dbPath) + if err != nil { + t.Fatalf("failed to open DB: %v", err) + } + defer db.Close() + + ctx := context.Background() + if err := CreateSchema(ctx, db); err != nil { + t.Fatalf("failed to create schema: %v", err) + } + + err = ImportFromDir(ctx, db, "/nonexistent/path") + if err == nil { + t.Error("expected error for non-existent directory") + } +} |