Function TestInMemoryAuditLog_PurgeEvents

pkg/security/audit_test.go:219–263

(t *testing.T)

}

func TestInMemoryAuditLog_PurgeEvents(t *testing.T) {
	auditLog := NewInMemoryAuditLog(nil)
	defer func() { _ = auditLog.Close() }()

	now := time.Now()
	yesterday := now.Add(-24 * time.Hour)

	// 添加旧事件和新事件
	_ = auditLog.LogEvent(AuditEvent{
		Type:      AuditTypeUserLogin,
		UserID:    "user1",
		Message:   "Old login",
		Timestamp: yesterday.Add(-1 * time.Hour),
	})
	_ = auditLog.LogEvent(AuditEvent{
		Type:      AuditTypeUserLogin,
		UserID:    "user2",
		Message:   "Recent login",
		Timestamp: now,
	})

	// 清除昨天之前的事件
	count, err := auditLog.PurgeEvents(context.Background(), yesterday)
	if err != nil {
		t.Fatalf("PurgeEvents failed: %v", err)
	}

	if count != 1 {
		t.Errorf("expected 1 event purged, got %d", count)
	}

	// 使用GetStatus验证剩余事件数
	status := auditLog.GetStatus()
	if status.TotalEvents != 1 {
		t.Errorf("expected 1 event remaining, got %d", status.TotalEvents)
	}

	// 使用GetEventsByUser验证user2的事件还在
	events, _ := auditLog.GetEventsByUser("user2", 10)
	if len(events) == 0 {
		t.Error("expected user2 event to remain")
	} else if events[0].UserID != "user2" {
		t.Errorf("expected user2 to remain, got %s", events[0].UserID)
	}
}

func TestInMemoryAuditLog_GetStatus(t *testing.T) {
	auditLog := NewInMemoryAuditLog(nil)