mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-28 05:46:36 +08:00
ryan
63a24da9ee
Result: {"status":"keep","total_issues":8,"golint_canonicalheader":0,"golint_errname":0,"golint_errorlint":1,"golint_forcetypeassert":0,"golint_gosec":0,"golint_intrange":0,"golint_modernize":3,"golint_nilnil":3,"golint_perfsprint":0,"golint_prealloc":0,"golint_recvcheck":1,"golint_usestdlibvars":0,"golint_wastedassign":0,"golint_total":8,"golint_test_testifylint":0,"golint_test_thelper":0,"golint_test_usetesting":0,"golint_test_total":0,"eslint_problems":0,"eslint_errors":0,"eslint_warnings":0,"tsc_errors":0,"measure_s":39}
59 lines
1.8 KiB
Go
59 lines
1.8 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package analytics
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
db "github.com/Rain-kl/Wavelet/internal/infra/persistence"
|
|
analyticsmodel "github.com/Rain-kl/Wavelet/internal/model/analytics"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBatchInsertNodeAccessLogs_Empty(t *testing.T) {
|
|
err := BatchInsertNodeAccessLogs(context.Background(), nil)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestBatchInsertNodeAccessLogs_UsesModelBatchSQL(t *testing.T) {
|
|
ctx := context.Background()
|
|
mockBatch := &mockBatch{}
|
|
mockConn := &mockConn{
|
|
batch: mockBatch,
|
|
batchQuery: analyticsmodel.NodeAccessLog{}.BatchInsertSQL(),
|
|
}
|
|
db.SetChConnForTest(mockConn)
|
|
t.Cleanup(func() { db.SetChConnForTest(nil) })
|
|
|
|
loggedAt := time.Now().UTC()
|
|
err := BatchInsertNodeAccessLogs(ctx, []analyticsmodel.NodeAccessLog{
|
|
{
|
|
NodeID: "node-a",
|
|
LoggedAt: loggedAt,
|
|
RemoteAddr: "1.1.1.1",
|
|
Region: "US",
|
|
Host: "example.com",
|
|
Path: "/alpha",
|
|
StatusCode: 200,
|
|
BytesSent: 2048,
|
|
CreatedAt: loggedAt,
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
assert.True(t, mockConn.prepareCalled)
|
|
assert.Equal(t, analyticsmodel.NodeAccessLog{}.BatchInsertSQL(), mockConn.preparedQuery)
|
|
assert.True(t, mockBatch.sendCalled)
|
|
require.Len(t, mockBatch.rows, 1)
|
|
assert.Equal(t, "node-a", mockBatch.rows[0][1])
|
|
require.Len(t, mockBatch.rows[0], 14)
|
|
assert.Empty(t, mockBatch.rows[0][7]) // user_agent
|
|
assert.Empty(t, mockBatch.rows[0][8]) // cache_status
|
|
assert.Equal(t, uint64(2048), mockBatch.rows[0][10]) // bytes_sent
|
|
assert.Equal(t, uint64(0), mockBatch.rows[0][11]) // request_length
|
|
assert.Equal(t, uint32(0), mockBatch.rows[0][12]) // request_time_ms
|
|
}
|