CyberStrikeAI/internal/database/audit_time_test.go

package database

import (
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"

	"go.uber.org/zap"
)

func TestBuildAuditLogsWhere_timeFilterSQL(t *testing.T) {
	since := time.Date(2026, 6, 16, 17, 2, 0, 0, time.UTC)
	until := time.Date(2026, 6, 17, 3, 3, 0, 0, time.UTC)
	where, args := buildAuditLogsWhere(ListAuditLogsFilter{Since: &since, Until: &until})
	if !strings.Contains(where, "strftime('%s', created_at) >=") {
		t.Fatalf("expected epoch comparison for since, got %q", where)
	}
	if !strings.Contains(where, "strftime('%s', created_at) <=") {
		t.Fatalf("expected epoch comparison for until, got %q", where)
	}
	if len(args) != 2 {
		t.Fatalf("expected 2 time args, got %d", len(args))
	}
	for i, arg := range args {
		s, ok := arg.(string)
		if !ok || s == "" {
			t.Fatalf("arg %d: want non-empty UTC RFC3339 string, got %v", i, arg)
		}
	}
}

func TestListAuditLogs_timeFilterMixedStorageFormats(t *testing.T) {
	root, err := os.Getwd()
	if err != nil {
		t.Skip(err)
	}
	dbPath := filepath.Join(root, "..", "..", "data", "conversations.db")
	if _, err := os.Stat(dbPath); err != nil {
		t.Skip("conversations.db not found")
	}
	db, err := NewDB(dbPath, zap.NewNop())
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()

	since, _ := ParseRFC3339Time("2026-06-16T17:02:00Z")
	until, _ := ParseRFC3339Time("2026-06-17T03:03:00Z")
	filter := ListAuditLogsFilter{Since: &since, Until: &until, Limit: 50}
	logs, err := db.ListAuditLogs(filter)
	if err != nil {
		t.Fatal(err)
	}
	for _, row := range logs {
		at := row.CreatedAt.UTC()
		if at.Before(since) || at.After(until) {
			t.Fatalf("log %s at %s outside [%s, %s]", row.ID, at, since, until)
		}
	}
}