diff --git a/backend/internal/handler/common_test.go b/backend/internal/handler/common_test.go index 15a6277e..656749d5 100644 --- a/backend/internal/handler/common_test.go +++ b/backend/internal/handler/common_test.go @@ -7,6 +7,7 @@ import ( "encoding/json" "io" "net/http" + "time" ) // --- Mock IDP Provider --- @@ -101,6 +102,15 @@ func (m *mockAuditRepo) FindByUserAndEvents(ctx context.Context, userID string, } return results, nil } + +func (m *mockAuditRepo) CountFailuresSince(ctx context.Context, since time.Time, tenantID string) (int64, error) { + return 0, nil +} + +func (m *mockAuditRepo) CountActiveSessionsSince(ctx context.Context, since time.Time, tenantID string) (int64, error) { + return 0, nil +} + func (m *mockAuditRepo) Ping(ctx context.Context) error { return nil } // --- Mock Consent Repository --- diff --git a/backend/internal/middleware/audit_middleware_test.go b/backend/internal/middleware/audit_middleware_test.go index f764b7d1..66a0f57e 100644 --- a/backend/internal/middleware/audit_middleware_test.go +++ b/backend/internal/middleware/audit_middleware_test.go @@ -8,6 +8,7 @@ import ( "net/http/httptest" "strings" "testing" + "time" "github.com/gofiber/fiber/v2" "github.com/stretchr/testify/assert" @@ -34,6 +35,14 @@ func (m *MockAuditRepository) FindByUserAndEvents(ctx context.Context, userID st return args.Get(0).([]domain.AuditLog), args.Error(1) } +func (m *MockAuditRepository) CountFailuresSince(ctx context.Context, since time.Time, tenantID string) (int64, error) { + return 0, nil +} + +func (m *MockAuditRepository) CountActiveSessionsSince(ctx context.Context, since time.Time, tenantID string) (int64, error) { + return 0, nil +} + func (m *MockAuditRepository) Ping(ctx context.Context) error { args := m.Called(ctx) return args.Error(0)