fb182adef4
🎯 核心变更: - OpType (string) → OpCode (int32) - 20+ OpCode枚举常量 (基于DOIP/IRP标准) - 类型安全 + 性能优化 📊 影响范围: - 核心模型: Operation结构体、CBOR序列化 - 数据库: schema.go + SQL DDL (PostgreSQL/MySQL/SQLite) - 持久化: repository.go查询、cursor_worker.go - API接口: Protobuf定义 + gRPC客户端 - 测试代码: 60+ 测试文件更新 ✅ 测试结果: - 通过率: 100% (所有87个测试用例) - 总体覆盖率: 53.7% - 核心包覆盖率: logger(100%), highclient(95.3%), model(79.1%) 📝 文档: - 精简README (1056行→489行，减少54%) - 完整的OpCode枚举说明 - 三种持久化策略示例 - 数据库表结构和架构图 🔧 技术细节: - 类型转换: string(OpCode) → int32(OpCode) - SQL参数: 字符串值 → 整数值 - Protobuf: op_type string → op_code int32 - 测试断言: 字符串比较 → 常量比较 🎉 质量保证: - 零编译错误 - 100%测试通过 - PostgreSQL/Pulsar集成测试验证 - 分布式并发安全测试通过
128 lines
3.9 KiB
Go
128 lines
3.9 KiB
Go
package model_test
|
||
|
||
import (
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
"github.com/stretchr/testify/require"
|
||
|
||
"go.yandata.net/iod/iod/go-trustlog/api/model"
|
||
)
|
||
|
||
// TestSignVerifyConsistency 测试加签和验签的一致性
|
||
// 验证加签时使用的数据和验签时使用的数据是否一致.
|
||
func TestSignVerifyConsistency(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
// 生成SM2密钥对
|
||
keyPair, err := model.GenerateSM2KeyPair()
|
||
require.NoError(t, err)
|
||
|
||
// 序列化为DER格式
|
||
privateKeyDER, err := model.MarshalSM2PrivateDER(keyPair.Private)
|
||
require.NoError(t, err)
|
||
|
||
publicKeyDER, err := model.MarshalSM2PublicDER(keyPair.Public)
|
||
require.NoError(t, err)
|
||
|
||
// 创建签名配置
|
||
signConfig := model.NewSM2EnvelopeConfig(privateKeyDER, publicKeyDER)
|
||
verifyConfig := model.NewSM2VerifyConfig(publicKeyDER)
|
||
|
||
// 创建测试Operation
|
||
op := &model.Operation{
|
||
OpID: "op-test-001",
|
||
Timestamp: time.Now(),
|
||
OpSource: model.OpSourceIRP,
|
||
OpCode: model.OpCodeCreateID,
|
||
DoPrefix: "test",
|
||
DoRepository: "repo",
|
||
Doid: "test/repo/123",
|
||
ProducerID: "producer-1",
|
||
OpActor: "actor-1",
|
||
}
|
||
|
||
err = op.CheckAndInit()
|
||
require.NoError(t, err)
|
||
|
||
// 1. 加签：序列化为Envelope
|
||
envelopeData, err := model.MarshalOperation(op, signConfig)
|
||
require.NoError(t, err)
|
||
require.NotNil(t, envelopeData)
|
||
|
||
// 2. 验签：验证Envelope
|
||
verifiedEnv, err := model.VerifyEnvelopeWithConfig(envelopeData, verifyConfig)
|
||
require.NoError(t, err)
|
||
require.NotNil(t, verifiedEnv)
|
||
|
||
// 3. 验证：加签时使用的body和验签时使用的body应该一致
|
||
// 手动反序列化envelope以获取body
|
||
originalEnv, err := model.UnmarshalEnvelope(envelopeData)
|
||
require.NoError(t, err)
|
||
|
||
// 验证body一致
|
||
assert.Equal(t, originalEnv.Body, verifiedEnv.Body, "加签和验签时使用的body应该完全一致")
|
||
assert.Equal(t, originalEnv.ProducerID, verifiedEnv.ProducerID)
|
||
assert.Equal(t, originalEnv.Signature, verifiedEnv.Signature)
|
||
|
||
// 4. 验证：如果修改body，验签应该失败
|
||
// 创建完全不同的body内容
|
||
modifiedBody := []byte("completely different body content")
|
||
require.NotEqual(t, originalEnv.Body, modifiedBody, "修改后的body应该不同")
|
||
|
||
modifiedEnv := &model.Envelope{
|
||
ProducerID: originalEnv.ProducerID,
|
||
Signature: originalEnv.Signature, // 使用旧的签名
|
||
Body: modifiedBody, // 使用修改后的body
|
||
}
|
||
modifiedData, err := model.MarshalEnvelope(modifiedEnv)
|
||
require.NoError(t, err)
|
||
|
||
// 验签应该失败，因为body被修改了但签名还是旧的
|
||
_, err = model.VerifyEnvelopeWithConfig(modifiedData, verifyConfig)
|
||
require.Error(t, err, "修改body后验签应该失败")
|
||
assert.Contains(t, err.Error(), "signature verification failed")
|
||
}
|
||
|
||
// TestSignVerifyDirectData 直接测试对相同数据的签名和验证.
|
||
func TestSignVerifyDirectData(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
// 生成SM2密钥对
|
||
keyPair, err := model.GenerateSM2KeyPair()
|
||
require.NoError(t, err)
|
||
|
||
// 序列化为DER格式
|
||
privateKeyDER, err := model.MarshalSM2PrivateDER(keyPair.Private)
|
||
require.NoError(t, err)
|
||
|
||
publicKeyDER, err := model.MarshalSM2PublicDER(keyPair.Public)
|
||
require.NoError(t, err)
|
||
|
||
// 创建签名器
|
||
signer := model.NewSM2Signer(privateKeyDER, publicKeyDER)
|
||
|
||
// 测试数据
|
||
testData := []byte("test data for signing")
|
||
|
||
// 1. 签名
|
||
signature, err := signer.Sign(testData)
|
||
require.NoError(t, err)
|
||
require.NotNil(t, signature)
|
||
|
||
// 2. 验证（使用相同的数据）
|
||
valid, err := signer.Verify(testData, signature)
|
||
require.NoError(t, err)
|
||
assert.True(t, valid, "使用相同数据验证应该成功")
|
||
|
||
// 3. 验证（使用不同的数据）
|
||
modifiedData := []byte("modified test data")
|
||
valid, err = signer.Verify(modifiedData, signature)
|
||
// VerifySignature在验证失败时会返回错误，这是预期的
|
||
require.Error(t, err, "使用不同数据验证应该失败并返回错误")
|
||
assert.Contains(t, err.Error(), "signature verification failed")
|
||
assert.False(t, valid)
|
||
}
|
||
|