mirror of
https://github.com/hanzozt/zt.git
synced 2026-10-30 15:04:55 +00:00
zeekay
241625e7d9
- Repo names: ziti-* → zt-* - Go modules: github.com/hanzozt/ziti → github.com/hanzozt/zt - Docker images: hanzozt/ziti-* → hanzozt/zt-* - C/Java/JS symbols: ziti_* → zt_* - File and directory names updated
73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
package model
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hanzozt/zt/v2/common/pb/cmd_pb"
|
|
"github.com/hanzozt/zt/v2/controller/command"
|
|
"github.com/hanzozt/zt/v2/controller/models"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestProtobufFactory(t *testing.T) {
|
|
ctx := NewTestContext(t)
|
|
defer ctx.Cleanup()
|
|
|
|
req := require.New(t)
|
|
|
|
service := &Service{
|
|
BaseEntity: models.BaseEntity{
|
|
Id: "one",
|
|
},
|
|
Name: "two",
|
|
TerminatorStrategy: "smartrouting",
|
|
}
|
|
|
|
createCmd := &command.CreateEntityCommand[*Service]{
|
|
Creator: ctx.GetManagers().Service,
|
|
Entity: service,
|
|
}
|
|
|
|
b, err := createCmd.Encode()
|
|
req.NoError(err)
|
|
|
|
val, err := ctx.GetManagers().Command.Decoders.Decode(b)
|
|
req.NoError(err)
|
|
msg, ok := val.(*command.CreateEntityCommand[*Service])
|
|
req.True(ok)
|
|
req.NoError(err)
|
|
req.Equal(service.Id, msg.Entity.Id)
|
|
req.Equal(service.Name, msg.Entity.Name)
|
|
}
|
|
|
|
func BenchmarkRegisterCommand(t *testing.B) {
|
|
ctx := NewTestContext(t)
|
|
defer ctx.Cleanup()
|
|
|
|
req := require.New(t)
|
|
|
|
service := &Service{
|
|
BaseEntity: models.BaseEntity{
|
|
Id: "one",
|
|
},
|
|
Name: "two",
|
|
TerminatorStrategy: "smartrouting",
|
|
}
|
|
|
|
createCmd := &command.CreateEntityCommand[*Service]{
|
|
Creator: ctx.GetManagers().Service,
|
|
Entity: service,
|
|
}
|
|
|
|
b, err := createCmd.Encode()
|
|
req.NoError(err)
|
|
|
|
cmdType := int32(cmd_pb.CommandType_CreateEntityType)
|
|
decoder := ctx.GetManagers().Command.Decoders.GetDecoder(cmdType)
|
|
|
|
for i := 0; i < t.N; i++ {
|
|
_, err = decoder.Decode(cmdType, b)
|
|
req.NoError(err)
|
|
}
|
|
}
|