79 lines
3.5 KiB
Go
79 lines
3.5 KiB
Go
// Copyright 2025 PingCAP, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package metering
|
|
|
|
import (
|
|
"maps"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDataEquals(t *testing.T) {
|
|
cases := []struct {
|
|
pair [2]Data
|
|
equal bool
|
|
}{
|
|
{pair: [2]Data{{taskID: 1, dataValues: dataValues{getRequests: 1}}, {taskID: 1, dataValues: dataValues{getRequests: 1}}}, equal: true},
|
|
{pair: [2]Data{{taskID: 1, dataValues: dataValues{getRequests: 1}}, {taskID: 1, dataValues: dataValues{getRequests: 2}}}, equal: false},
|
|
{pair: [2]Data{{taskID: 1, dataValues: dataValues{putRequests: 1}}, {taskID: 1, dataValues: dataValues{putRequests: 1}}}, equal: true},
|
|
{pair: [2]Data{{taskID: 1, dataValues: dataValues{putRequests: 1}}, {taskID: 1, dataValues: dataValues{putRequests: 2}}}, equal: false},
|
|
{pair: [2]Data{{taskID: 1, dataValues: dataValues{objStoreReadBytes: 1}}, {taskID: 1, dataValues: dataValues{objStoreReadBytes: 2}}}, equal: false},
|
|
{pair: [2]Data{{taskID: 1, dataValues: dataValues{objStoreWriteBytes: 1}}, {taskID: 1, dataValues: dataValues{objStoreWriteBytes: 2}}}, equal: false},
|
|
{pair: [2]Data{{taskID: 1, dataValues: dataValues{clusterReadBytes: 1}}, {taskID: 1, dataValues: dataValues{clusterReadBytes: 2}}}, equal: false},
|
|
{pair: [2]Data{{taskID: 1, dataValues: dataValues{clusterWriteBytes: 1}}, {taskID: 1, dataValues: dataValues{clusterWriteBytes: 2}}}, equal: false},
|
|
// we only compare the data fields, not taskID
|
|
{pair: [2]Data{{taskID: 1, dataValues: dataValues{getRequests: 1}}, {taskID: 2, dataValues: dataValues{getRequests: 1}}}, equal: true},
|
|
}
|
|
for i, c := range cases {
|
|
t.Run(strconv.Itoa(i), func(t *testing.T) {
|
|
require.Equal(t, c.equal, c.pair[0].equals(&c.pair[1]), "case %d failed", i)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDataCalMeterDataItem(t *testing.T) {
|
|
currData := &Data{dataValues: dataValues{getRequests: 10, putRequests: 20,
|
|
objStoreReadBytes: 300, objStoreWriteBytes: 400,
|
|
clusterReadBytes: 500, clusterWriteBytes: 600}, taskID: 1, keyspace: "ks", taskType: "tt"}
|
|
require.Nil(t, currData.calMeterDataItem(currData))
|
|
getItemFn := func(in map[string]any) map[string]any {
|
|
maps.Copy(in, map[string]any{
|
|
"version": "1",
|
|
"cluster_id": "ks",
|
|
"source_name": category,
|
|
"task_type": "tt",
|
|
"task_id": int64(1),
|
|
})
|
|
return in
|
|
}
|
|
require.EqualValues(t, getItemFn(map[string]any{
|
|
"get_requests": uint64(5),
|
|
"put_requests": uint64(15),
|
|
"obj_store_read_bytes": uint64(200),
|
|
"obj_store_write_bytes": uint64(300),
|
|
"cluster_read_bytes": uint64(300),
|
|
"cluster_write_bytes": uint64(400),
|
|
}), currData.calMeterDataItem(&Data{dataValues: dataValues{getRequests: 5, putRequests: 5,
|
|
objStoreReadBytes: 100, objStoreWriteBytes: 100,
|
|
clusterReadBytes: 200, clusterWriteBytes: 200}}))
|
|
require.EqualValues(t, getItemFn(map[string]any{
|
|
"get_requests": uint64(5),
|
|
}), currData.calMeterDataItem(&Data{dataValues: dataValues{getRequests: 5, putRequests: 20,
|
|
objStoreReadBytes: 300, objStoreWriteBytes: 400,
|
|
clusterReadBytes: 500, clusterWriteBytes: 600}}))
|
|
}
|