94 lines
3.3 KiB
Go
94 lines
3.3 KiB
Go
package handler
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/company/ai-ops/internal/domain/model"
|
|
"github.com/company/ai-ops/internal/service"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type mockMetricRepo struct{ mock.Mock }
|
|
|
|
func (m *mockMetricRepo) GetRealtime(ctx context.Context) (*model.RealtimeMetrics, error) {
|
|
args := m.Called(ctx)
|
|
return args.Get(0).(*model.RealtimeMetrics), args.Error(1)
|
|
}
|
|
func (m *mockMetricRepo) Query(ctx context.Context, req model.MetricQueryRequest) ([]model.MetricPoint, error) {
|
|
args := m.Called(ctx, req)
|
|
return args.Get(0).([]model.MetricPoint), args.Error(1)
|
|
}
|
|
func (m *mockMetricRepo) GetLatest(ctx context.Context, source, name string) (*model.MetricPoint, error) {
|
|
args := m.Called(ctx, source, name)
|
|
return args.Get(0).(*model.MetricPoint), args.Error(1)
|
|
}
|
|
|
|
type mockAlertRepo struct{ mock.Mock }
|
|
|
|
func (m *mockAlertRepo) GetOpenCount(ctx context.Context) (*model.AlertCount, error) {
|
|
args := m.Called(ctx)
|
|
return args.Get(0).(*model.AlertCount), args.Error(1)
|
|
}
|
|
func (m *mockAlertRepo) ListRules(ctx context.Context) ([]model.AlertRule, error) {
|
|
args := m.Called(ctx)
|
|
return args.Get(0).([]model.AlertRule), args.Error(1)
|
|
}
|
|
func (m *mockAlertRepo) GetRuleByID(ctx context.Context, id string) (*model.AlertRule, error) {
|
|
args := m.Called(ctx, id)
|
|
return args.Get(0).(*model.AlertRule), args.Error(1)
|
|
}
|
|
func (m *mockAlertRepo) CreateRule(ctx context.Context, rule *model.AlertRule) error {
|
|
args := m.Called(ctx, rule)
|
|
return args.Error(0)
|
|
}
|
|
func (m *mockAlertRepo) UpdateRule(ctx context.Context, rule *model.AlertRule) error {
|
|
args := m.Called(ctx, rule)
|
|
return args.Error(0)
|
|
}
|
|
func (m *mockAlertRepo) DeleteRule(ctx context.Context, id string) error {
|
|
args := m.Called(ctx, id)
|
|
return args.Error(0)
|
|
}
|
|
func (m *mockAlertRepo) ListEvents(ctx context.Context, status string, page, pageSize int) ([]model.AlertEvent, int, error) {
|
|
args := m.Called(ctx, status, page, pageSize)
|
|
return args.Get(0).([]model.AlertEvent), args.Int(1), args.Error(2)
|
|
}
|
|
func (m *mockAlertRepo) CreateEvent(ctx context.Context, event *model.AlertEvent) error {
|
|
args := m.Called(ctx, event)
|
|
return args.Error(0)
|
|
}
|
|
func (m *mockAlertRepo) CreateEventWithAggregation(ctx context.Context, event *model.AlertEvent, window time.Duration, threshold int) (*model.AlertEvent, error) {
|
|
args := m.Called(ctx, event, window, threshold)
|
|
return args.Get(0).(*model.AlertEvent), args.Error(1)
|
|
}
|
|
func (m *mockAlertRepo) UpdateEventStatus(ctx context.Context, id, status string) error {
|
|
args := m.Called(ctx, id, status)
|
|
return args.Error(0)
|
|
}
|
|
func (m *mockAlertRepo) EscalateEvent(ctx context.Context, id, newLevel string) error {
|
|
args := m.Called(ctx, id, newLevel)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func TestMetricHandler_GetRealtime(t *testing.T) {
|
|
mr := new(mockMetricRepo)
|
|
ar := new(mockAlertRepo)
|
|
svc := service.NewMetricService(mr, ar)
|
|
h := NewMetricHandler(svc)
|
|
|
|
expected := &model.RealtimeMetrics{QPS: 100, AvgLatency: 50, P99Latency: 100, ErrorRate: 0.01}
|
|
mr.On("GetRealtime", mock.Anything).Return(expected, nil)
|
|
|
|
req := httptest.NewRequest("GET", "/api/v1/ai-ops/metrics/realtime", nil)
|
|
w := httptest.NewRecorder()
|
|
h.GetRealtime(w, req)
|
|
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
assert.Contains(t, w.Body.String(), `"qps":100`)
|
|
}
|