152 lines
4.0 KiB
Go
152 lines
4.0 KiB
Go
package control_test
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neofs-node/pkg/services/control"
|
|
)
|
|
|
|
func TestHealthCheckResponse_Body_StableMarshal(t *testing.T) {
|
|
testStableMarshal(t,
|
|
generateHealthCheckResponseBody(),
|
|
new(control.HealthCheckResponse_Body),
|
|
func(m1, m2 protoMessage) bool {
|
|
return equalHealthCheckResponseBodies(
|
|
m1.(*control.HealthCheckResponse_Body),
|
|
m2.(*control.HealthCheckResponse_Body),
|
|
)
|
|
},
|
|
)
|
|
}
|
|
|
|
func generateHealthCheckResponseBody() *control.HealthCheckResponse_Body {
|
|
body := new(control.HealthCheckResponse_Body)
|
|
body.SetNetmapStatus(control.NetmapStatus_ONLINE)
|
|
body.SetHealthStatus(control.HealthStatus_SHUTTING_DOWN)
|
|
|
|
return body
|
|
}
|
|
|
|
func equalHealthCheckResponseBodies(b1, b2 *control.HealthCheckResponse_Body) bool {
|
|
return b1.GetNetmapStatus() == b2.GetNetmapStatus() &&
|
|
b1.GetHealthStatus() == b2.GetHealthStatus()
|
|
}
|
|
|
|
func TestNetmapSnapshotResponse_Body_StableMarshal(t *testing.T) {
|
|
testStableMarshal(t,
|
|
generateNetmapSnapshotResponseBody(),
|
|
new(control.NetmapSnapshotResponse_Body),
|
|
func(m1, m2 protoMessage) bool {
|
|
return equalNetmapSnapshotResponseBodies(
|
|
m1.(*control.NetmapSnapshotResponse_Body),
|
|
m2.(*control.NetmapSnapshotResponse_Body),
|
|
)
|
|
},
|
|
)
|
|
}
|
|
|
|
func generateNetmapSnapshotResponseBody() *control.NetmapSnapshotResponse_Body {
|
|
body := new(control.NetmapSnapshotResponse_Body)
|
|
body.SetNetmap(generateNetmap())
|
|
|
|
return body
|
|
}
|
|
|
|
func equalNetmapSnapshotResponseBodies(b1, b2 *control.NetmapSnapshotResponse_Body) bool {
|
|
return equalNetmaps(b1.GetNetmap(), b2.GetNetmap())
|
|
}
|
|
|
|
func TestSetNetmapStatusRequest_Body_StableMarshal(t *testing.T) {
|
|
testStableMarshal(t,
|
|
generateSetNetmapStatusRequestBody(),
|
|
new(control.SetNetmapStatusRequest_Body),
|
|
func(m1, m2 protoMessage) bool {
|
|
return equalSetnetmapStatusRequestBodies(
|
|
m1.(*control.SetNetmapStatusRequest_Body),
|
|
m2.(*control.SetNetmapStatusRequest_Body),
|
|
)
|
|
},
|
|
)
|
|
}
|
|
|
|
func generateSetNetmapStatusRequestBody() *control.SetNetmapStatusRequest_Body {
|
|
body := new(control.SetNetmapStatusRequest_Body)
|
|
body.SetStatus(control.NetmapStatus_ONLINE)
|
|
|
|
return body
|
|
}
|
|
|
|
func equalSetnetmapStatusRequestBodies(b1, b2 *control.SetNetmapStatusRequest_Body) bool {
|
|
return b1.GetStatus() == b2.GetStatus()
|
|
}
|
|
|
|
func TestListShardsResponse_Body_StableMarshal(t *testing.T) {
|
|
testStableMarshal(t,
|
|
generateListShardsResponseBody(),
|
|
new(control.ListShardsResponse_Body),
|
|
func(m1, m2 protoMessage) bool {
|
|
return equalListShardResponseBodies(
|
|
m1.(*control.ListShardsResponse_Body),
|
|
m2.(*control.ListShardsResponse_Body),
|
|
)
|
|
},
|
|
)
|
|
}
|
|
|
|
func equalListShardResponseBodies(b1, b2 *control.ListShardsResponse_Body) bool {
|
|
if len(b1.Shards) != len(b2.Shards) {
|
|
return false
|
|
}
|
|
|
|
for i := range b1.Shards {
|
|
if b1.Shards[i].GetMetabasePath() != b2.Shards[i].GetMetabasePath() ||
|
|
b1.Shards[i].GetBlobstorPath() != b2.Shards[i].GetBlobstorPath() ||
|
|
b1.Shards[i].GetWritecachePath() != b2.Shards[i].GetWritecachePath() ||
|
|
!bytes.Equal(b1.Shards[i].GetShard_ID(), b2.Shards[i].GetShard_ID()) {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func generateListShardsResponseBody() *control.ListShardsResponse_Body {
|
|
body := new(control.ListShardsResponse_Body)
|
|
body.SetShards([]*control.ShardInfo{
|
|
generateShardInfo(),
|
|
generateShardInfo(),
|
|
})
|
|
|
|
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
|
|
}
|