refactor: 重构trustlog-sdk目录结构到trustlog/go-trustlog
- 将所有trustlog-sdk文件移动到trustlog/go-trustlog/目录 - 更新README中所有import路径从trustlog-sdk改为go-trustlog - 更新cookiecutter配置文件中的项目名称 - 更新根目录.lefthook.yml以引用新位置的配置 - 添加go.sum文件到版本控制 - 删除过时的示例文件 这次重构与trustlog-server保持一致的目录结构， 为未来支持多语言SDK（Python、Java等）预留空间。
This commit is contained in:
ryan
321
api/model/record_test.go
Normal file
321
api/model/record_test.go
Normal file
@@ -0,0 +1,321 @@
|
||||
package model_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"go.yandata.net/iod/iod/trustlog-sdk/api/model"
|
||||
)
|
||||
|
||||
func TestRecord_Key(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{
|
||||
ID: "test-record-id",
|
||||
}
|
||||
assert.Equal(t, "test-record-id", rec.Key())
|
||||
}
|
||||
|
||||
func TestNewFullRecord(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
now := time.Now()
|
||||
rec, err := model.NewFullRecord(
|
||||
"test-prefix",
|
||||
"producer-1",
|
||||
now,
|
||||
"operator-1",
|
||||
[]byte("extra"),
|
||||
"log",
|
||||
)
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, rec)
|
||||
assert.NotEmpty(t, rec.ID)
|
||||
assert.Equal(t, "test-prefix", rec.DoPrefix)
|
||||
assert.Equal(t, "producer-1", rec.ProducerID)
|
||||
assert.Equal(t, now.Unix(), rec.Timestamp.Unix())
|
||||
assert.Equal(t, "operator-1", rec.Operator)
|
||||
assert.Equal(t, []byte("extra"), rec.Extra)
|
||||
assert.Equal(t, "log", rec.RCType)
|
||||
}
|
||||
|
||||
func TestNewFullRecord_Invalid(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
now := time.Now()
|
||||
// Missing required ProducerID
|
||||
rec, err := model.NewFullRecord(
|
||||
"test-prefix",
|
||||
"", // Empty ProducerID
|
||||
now,
|
||||
"operator-1",
|
||||
[]byte("extra"),
|
||||
"log",
|
||||
)
|
||||
require.Error(t, err)
|
||||
assert.Nil(t, rec)
|
||||
}
|
||||
|
||||
func TestRecord_CheckAndInit(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
rec *model.Record
|
||||
wantErr bool
|
||||
}{
|
||||
{
|
||||
name: "valid record",
|
||||
rec: &model.Record{
|
||||
DoPrefix: "test",
|
||||
ProducerID: "producer-1",
|
||||
Timestamp: time.Now(),
|
||||
Operator: "operator-1",
|
||||
Extra: []byte("extra"),
|
||||
RCType: "log",
|
||||
},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "auto generate ID",
|
||||
rec: &model.Record{
|
||||
ID: "", // Will be auto-generated
|
||||
DoPrefix: "test",
|
||||
ProducerID: "producer-1",
|
||||
Timestamp: time.Now(),
|
||||
Operator: "operator-1",
|
||||
Extra: []byte("extra"),
|
||||
RCType: "log",
|
||||
},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "missing ProducerID",
|
||||
rec: &model.Record{
|
||||
DoPrefix: "test",
|
||||
ProducerID: "", // Required field
|
||||
Timestamp: time.Now(),
|
||||
Operator: "operator-1",
|
||||
Extra: []byte("extra"),
|
||||
RCType: "log",
|
||||
},
|
||||
wantErr: true,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
err := tt.rec.CheckAndInit()
|
||||
if tt.wantErr {
|
||||
require.Error(t, err)
|
||||
} else {
|
||||
require.NoError(t, err)
|
||||
if tt.name == "auto generate ID" {
|
||||
assert.NotEmpty(t, tt.rec.ID)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestRecord_MarshalUnmarshalBinary(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
original := &model.Record{
|
||||
ID: "rec-123",
|
||||
DoPrefix: "test",
|
||||
ProducerID: "producer-1",
|
||||
Timestamp: time.Now(),
|
||||
Operator: "operator-1",
|
||||
Extra: []byte("extra"),
|
||||
RCType: "log",
|
||||
}
|
||||
|
||||
err := original.CheckAndInit()
|
||||
require.NoError(t, err)
|
||||
|
||||
// Marshal
|
||||
data, err := original.MarshalBinary()
|
||||
require.NoError(t, err)
|
||||
require.NotNil(t, data)
|
||||
|
||||
// Unmarshal
|
||||
result := &model.Record{}
|
||||
err = result.UnmarshalBinary(data)
|
||||
require.NoError(t, err)
|
||||
|
||||
// Verify
|
||||
assert.Equal(t, original.ID, result.ID)
|
||||
assert.Equal(t, original.DoPrefix, result.DoPrefix)
|
||||
assert.Equal(t, original.ProducerID, result.ProducerID)
|
||||
assert.Equal(t, original.Timestamp.Unix(), result.Timestamp.Unix())
|
||||
// 验证纳秒精度被保留
|
||||
assert.Equal(t, original.Timestamp.UnixNano(), result.Timestamp.UnixNano(),
|
||||
"时间戳的纳秒精度应该被保留")
|
||||
assert.Equal(t, original.Operator, result.Operator)
|
||||
assert.Equal(t, original.Extra, result.Extra)
|
||||
assert.Equal(t, original.RCType, result.RCType)
|
||||
}
|
||||
|
||||
func TestRecord_MarshalBinary_Empty(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{
|
||||
DoPrefix: "test",
|
||||
ProducerID: "producer-1",
|
||||
Timestamp: time.Now(),
|
||||
}
|
||||
// MarshalBinary should succeed even without CheckAndInit
|
||||
// It just serializes the data
|
||||
data, err := rec.MarshalBinary()
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, data)
|
||||
}
|
||||
|
||||
func TestRecord_UnmarshalBinary_Empty(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{}
|
||||
err := rec.UnmarshalBinary([]byte{})
|
||||
require.Error(t, err)
|
||||
}
|
||||
|
||||
func TestRecord_DoHash(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{
|
||||
ID: "rec-123",
|
||||
DoPrefix: "test",
|
||||
ProducerID: "producer-1",
|
||||
Timestamp: time.Now(),
|
||||
Operator: "operator-1",
|
||||
Extra: []byte("extra"),
|
||||
RCType: "log",
|
||||
}
|
||||
|
||||
err := rec.CheckAndInit()
|
||||
require.NoError(t, err)
|
||||
|
||||
ctx := context.Background()
|
||||
hashData, err := rec.DoHash(ctx)
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, hashData)
|
||||
assert.Equal(t, rec.ID, hashData.Key())
|
||||
assert.NotEmpty(t, hashData.Hash())
|
||||
}
|
||||
|
||||
func TestRecordHashData(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
// RecordHashData is created through DoHash, test it indirectly
|
||||
rec := &model.Record{
|
||||
ID: "rec-123",
|
||||
DoPrefix: "test",
|
||||
ProducerID: "producer-1",
|
||||
Timestamp: time.Now(),
|
||||
Operator: "operator-1",
|
||||
Extra: []byte("extra"),
|
||||
RCType: "log",
|
||||
}
|
||||
|
||||
err := rec.CheckAndInit()
|
||||
require.NoError(t, err)
|
||||
|
||||
ctx := context.Background()
|
||||
hashData, err := rec.DoHash(ctx)
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, hashData)
|
||||
assert.Equal(t, "rec-123", hashData.Key())
|
||||
assert.NotEmpty(t, hashData.Hash())
|
||||
assert.Equal(t, model.Sha256Simd, hashData.Type())
|
||||
}
|
||||
|
||||
func TestRecord_GetProducerID(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{
|
||||
ProducerID: "producer-123",
|
||||
}
|
||||
assert.Equal(t, "producer-123", rec.GetProducerID())
|
||||
}
|
||||
|
||||
func TestRecord_GetDoPrefix(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{
|
||||
DoPrefix: "test-prefix",
|
||||
}
|
||||
assert.Equal(t, "test-prefix", rec.GetDoPrefix())
|
||||
}
|
||||
|
||||
func TestRecord_WithDoPrefix(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{}
|
||||
result := rec.WithDoPrefix("test-prefix")
|
||||
assert.Equal(t, rec, result)
|
||||
assert.Equal(t, "test-prefix", rec.DoPrefix)
|
||||
}
|
||||
|
||||
func TestRecord_WithTimestamp(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{}
|
||||
now := time.Now()
|
||||
result := rec.WithTimestamp(now)
|
||||
assert.Equal(t, rec, result)
|
||||
assert.Equal(t, now.Unix(), rec.Timestamp.Unix())
|
||||
}
|
||||
|
||||
func TestRecord_WithOperator(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{}
|
||||
result := rec.WithOperator("operator-1")
|
||||
assert.Equal(t, rec, result)
|
||||
assert.Equal(t, "operator-1", rec.Operator)
|
||||
}
|
||||
|
||||
func TestRecord_WithExtra(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{}
|
||||
extra := []byte("extra-data")
|
||||
result := rec.WithExtra(extra)
|
||||
assert.Equal(t, rec, result)
|
||||
assert.Equal(t, extra, rec.Extra)
|
||||
}
|
||||
|
||||
func TestRecord_WithRCType(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{}
|
||||
result := rec.WithRCType("log")
|
||||
assert.Equal(t, rec, result)
|
||||
assert.Equal(t, "log", rec.RCType)
|
||||
}
|
||||
|
||||
func TestRecord_ChainedMethods(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
rec := &model.Record{}
|
||||
now := time.Now()
|
||||
result := rec.
|
||||
WithDoPrefix("prefix").
|
||||
WithTimestamp(now).
|
||||
WithOperator("operator").
|
||||
WithExtra([]byte("extra")).
|
||||
WithRCType("log")
|
||||
|
||||
assert.Equal(t, rec, result)
|
||||
assert.Equal(t, "prefix", rec.DoPrefix)
|
||||
assert.Equal(t, now.Unix(), rec.Timestamp.Unix())
|
||||
assert.Equal(t, "operator", rec.Operator)
|
||||
assert.Equal(t, []byte("extra"), rec.Extra)
|
||||
assert.Equal(t, "log", rec.RCType)
|
||||
}
|
||||
Reference in New Issue
Block a user