func TestRemoveTransportsFromNode(t *testing.T) {
nodeManager := &NodeManager{Port: 7100}
defer nodeManager.CloseAll()
nodeManager.CreateNodeConfigList(10)
nodeManager.ConnectNodes()
pubKey := nodeManager.PubKeyList[4]
configFrom := nodeManager.ConfigList[pubKey]
node := nodeManager.NodesList[pubKey]
assert.Len(t, node.GetTransports(), 2, "Error expected 2 transport in the node")
config := CreateTestConfig(nodeManager.Port)
nodeManager.Port += 100
ConnectNodeToNode(configFrom, config)
AddPeersToNode(node, *config)
assert.Len(t, node.GetTransports(), 3, "Error expected 3 transport in the node")
config2 := CreateTestConfig(nodeManager.Port)
nodeManager.Port += 100
ConnectNodeToNode(configFrom, config2)
AddPeersToNode(node, *config2)
assert.Len(t, node.GetTransports(), 4, "Error expected 4 transport in the node")
transport := node.GetTransports()[0]
nodeManager.RemoveTransportsFromNode(4, transport)
assert.Len(t, node.GetTransports(), 3, "Error expected 3 transport in the node")
}
func TestAddTransportsToNode(t *testing.T) {
nodeManager := &NodeManager{Port: 5100}
defer nodeManager.CloseAll()
nodeManager.CreateNodeConfigList(10)
nodeManager.ConnectNodes()
config := CreateTestConfig(nodeManager.Port)
nodeManager.Port += 100
pubKey := nodeManager.PubKeyList[1]
configFrom := nodeManager.ConfigList[pubKey]
node := nodeManager.NodesList[pubKey]
assert.Len(t, node.GetTransports(), 2, "Error expected 2 transport in the node")
ConnectNodeToNode(configFrom, config)
AddPeersToNode(node, *config)
assert.Len(t, node.GetTransports(), 3, "Error expected 3 transport in the node")
config2 := CreateTestConfig(nodeManager.Port)
pubKey2 := nodeManager.PubKeyList[3]
configFrom2 := nodeManager.ConfigList[pubKey2]
node2 := nodeManager.NodesList[pubKey2]
assert.Len(t, node2.GetTransports(), 2, "Error expected 2 transport in the node2")
ConnectNodeToNode(configFrom2, config2)
AddPeersToNode(node2, *config2)
assert.Len(t, node2.GetTransports(), 3, "Error expected 3 transport in the node2")
}
func TestGetTransportsFromNode(t *testing.T) {
nodeManager := &NodeManager{Port: 6100}
defer nodeManager.CloseAll()
nodeManager.CreateNodeConfigList(10)
nodeManager.ConnectNodes()
pubKey := nodeManager.PubKeyList[2]
node := nodeManager.NodesList[pubKey]
assert.Len(t, node.GetTransports(), 2, "Error expected 2 transport in the node")
}
