mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
f0e234df1f
Agent 仅上报 host_metrics/edge_health/access_logs；业务流量与 UV 由 Server 侧访问日志聚合。新增 of_node_edge_health 与 of_access_log_hourly， 删除 request_reports/openresty 吞吐路径；API 不再暴露 traffic_reports 与 openresty_rx|tx。心跳/离线默认阈值与回填迁移一并入库。
49 lines
1.4 KiB
Go
49 lines
1.4 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package analytics
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/db"
|
|
analyticsmodel "github.com/Rain-kl/Wavelet/internal/model/analytics"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestInsertNodeEdgeHealth_EmptyNodeID(t *testing.T) {
|
|
err := InsertNodeEdgeHealth(context.Background(), analyticsmodel.NodeEdgeHealth{})
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestInsertNodeEdgeHealth_UsesEdgeHealthBatchSQL(t *testing.T) {
|
|
ctx := context.Background()
|
|
mockBatch := &mockBatch{}
|
|
mockConn := &mockConn{
|
|
batch: mockBatch,
|
|
batchQuery: analyticsmodel.NodeEdgeHealth{}.BatchInsertSQL(),
|
|
}
|
|
db.SetChConnForTest(mockConn)
|
|
t.Cleanup(func() { db.SetChConnForTest(nil) })
|
|
|
|
capturedAt := time.Now().UTC()
|
|
err := InsertNodeEdgeHealth(ctx, analyticsmodel.NodeEdgeHealth{
|
|
NodeID: "node-a",
|
|
CapturedAt: capturedAt,
|
|
Status: "",
|
|
Connections: 3,
|
|
CreatedAt: capturedAt,
|
|
})
|
|
require.NoError(t, err)
|
|
assert.True(t, mockConn.prepareCalled)
|
|
assert.Equal(t, analyticsmodel.NodeEdgeHealth{}.BatchInsertSQL(), mockConn.preparedQuery)
|
|
assert.True(t, mockBatch.sendCalled)
|
|
require.Len(t, mockBatch.rows, 1)
|
|
assert.Equal(t, "node-a", mockBatch.rows[0][1])
|
|
assert.Equal(t, "unknown", mockBatch.rows[0][3]) // status default
|
|
assert.Equal(t, int64(3), mockBatch.rows[0][4]) // connections
|
|
}
|