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.
80 lines
1.6 KiB
80 lines
1.6 KiB
package nodeconfig
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/harmony-one/harmony/p2p"
|
|
)
|
|
|
|
func TestNodeConfigSingleton(t *testing.T) {
|
|
// init 3 configs
|
|
_ = GetShardConfig(2)
|
|
|
|
// get the singleton variable
|
|
c := GetShardConfig(Global)
|
|
|
|
c.SetIsLeader(true)
|
|
|
|
if !c.IsLeader() {
|
|
t.Errorf("IsLeader = %v, expected = %v", c.IsLeader(), true)
|
|
}
|
|
|
|
c.SetBeaconGroupID(p2p.GroupIDBeacon)
|
|
|
|
d := GetShardConfig(Global)
|
|
|
|
if !d.IsLeader() {
|
|
t.Errorf("IsLeader = %v, expected = %v", d.IsLeader(), true)
|
|
}
|
|
|
|
g := d.GetBeaconGroupID()
|
|
|
|
if g != p2p.GroupIDBeacon {
|
|
t.Errorf("GetBeaconGroupID = %v, expected = %v", g, p2p.GroupIDBeacon)
|
|
}
|
|
}
|
|
|
|
func TestNodeConfigMultiple(t *testing.T) {
|
|
// init 3 configs
|
|
c := GetShardConfig(2)
|
|
d := GetShardConfig(1)
|
|
e := GetShardConfig(0)
|
|
f := GetShardConfig(42)
|
|
|
|
if f != nil {
|
|
t.Errorf("expecting nil, got: %v", f)
|
|
}
|
|
|
|
if c.IsBeacon() != false {
|
|
t.Errorf("expecting the node to not be beacon yet, got: %v", c.IsBeacon())
|
|
}
|
|
|
|
c.SetIsBeacon(true)
|
|
|
|
if c.IsBeacon() != true {
|
|
t.Errorf("expecting the node to be beacon, got: %v", c.IsBeacon())
|
|
}
|
|
|
|
d.SetShardGroupID("abcd")
|
|
if d.GetShardGroupID() != "abcd" {
|
|
t.Errorf("expecting abcd, got: %v", d.GetShardGroupID())
|
|
}
|
|
|
|
e.SetClientGroupID("client")
|
|
if e.GetClientGroupID() != "client" {
|
|
t.Errorf("expecting client, got: %v", d.GetClientGroupID())
|
|
}
|
|
|
|
e.SetIsClient(false)
|
|
if e.IsClient() != false {
|
|
t.Errorf("expecting false, got: %v", e.IsClient())
|
|
}
|
|
|
|
c.SetRole(NewNode)
|
|
if c.Role() != NewNode {
|
|
t.Errorf("expecting NewNode, got: %s", c.Role())
|
|
}
|
|
if c.Role().String() != "NewNode" {
|
|
t.Errorf("expecting NewNode, got: %s", c.Role().String())
|
|
}
|
|
}
|
|
|