You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.5 KiB
59 lines
1.5 KiB
package utils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/harmony-one/harmony/p2p"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// Tests for TestConvertFixedDataIntoByteArray.
|
|
func TestConvertFixedDataIntoByteArray(t *testing.T) {
|
|
res := ConvertFixedDataIntoByteArray(int16(3))
|
|
if len(res) != 2 {
|
|
t.Errorf("Conversion incorrect.")
|
|
}
|
|
res = ConvertFixedDataIntoByteArray(int32(3))
|
|
if len(res) != 4 {
|
|
t.Errorf("Conversion incorrect.")
|
|
}
|
|
}
|
|
|
|
// Tests for TestAllocateShard.
|
|
func TestAllocateShard(t *testing.T) {
|
|
num, success := AllocateShard(1, 1)
|
|
assert.Equal(t, num, 1, "error")
|
|
assert.True(t, success, "error")
|
|
|
|
num, success = AllocateShard(2, 1)
|
|
assert.False(t, success, "error")
|
|
assert.Equal(t, num, 1, "error")
|
|
|
|
num, success = AllocateShard(1, 2)
|
|
assert.True(t, success, "error")
|
|
assert.Equal(t, num, 1, "error")
|
|
|
|
num, success = AllocateShard(5, 3)
|
|
assert.False(t, success, "error")
|
|
assert.Equal(t, num, 2, "error")
|
|
|
|
num, success = AllocateShard(6, 3)
|
|
assert.False(t, success, "error")
|
|
assert.Equal(t, num, 3, "error")
|
|
}
|
|
|
|
// Test for GenKey
|
|
func TestGenKey(t *testing.T) {
|
|
GenKey("3.3.3.3", "3456")
|
|
}
|
|
|
|
// Test for GetUniqueIDFromPeer
|
|
func TestGetUniqueIDFromPeer(t *testing.T) {
|
|
peer := p2p.Peer{IP: "1.1.1.1", Port: "123"}
|
|
assert.Equal(t, GetUniqueIDFromPeer(peer), uint32(1111123), "should be equal to 1111123")
|
|
}
|
|
|
|
// Test for GetUniqueIDFromIPPort
|
|
func TestGetUniqueIDFromIPPort(t *testing.T) {
|
|
assert.Equal(t, GetUniqueIDFromIPPort("1.1.1.1", "123"), uint32(1111123), "should be equal to 1111123")
|
|
}
|
|
|