[#1059] control: Add unit test for SetShardMode
Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
parent
29c69f37eb
commit
fa78a37787
1 changed files with 29 additions and 0 deletions
|
@ -120,3 +120,32 @@ func generateListShardsResponseBody() *control.ListShardsResponse_Body {
|
|||
|
||||
return body
|
||||
}
|
||||
|
||||
func TestSetShardModeRequest_Body_StableMarshal(t *testing.T) {
|
||||
testStableMarshal(t,
|
||||
generateSetShardModeRequestBody(),
|
||||
new(control.SetShardModeRequest_Body),
|
||||
func(m1, m2 protoMessage) bool {
|
||||
return equalSetShardModeRequestBodies(
|
||||
m1.(*control.SetShardModeRequest_Body),
|
||||
m2.(*control.SetShardModeRequest_Body),
|
||||
)
|
||||
},
|
||||
)
|
||||
}
|
||||
|
||||
func generateSetShardModeRequestBody() *control.SetShardModeRequest_Body {
|
||||
body := new(control.SetShardModeRequest_Body)
|
||||
body.SetShardID([]byte{0, 1, 2, 3, 4})
|
||||
body.SetMode(control.ShardMode_READ_WRITE)
|
||||
|
||||
return body
|
||||
}
|
||||
|
||||
func equalSetShardModeRequestBodies(b1, b2 *control.SetShardModeRequest_Body) bool {
|
||||
if b1.GetMode() != b2.GetMode() || !bytes.Equal(b1.Shard_ID, b2.Shard_ID) {
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue