def basic_test_manager():
# Create a local Host to communicate with
test_host = ProcessSpawner(Host, name="test-host", addr="tcp://127.0.0.1:*")
host_cli = RPCClient(test_host.name, test_host.addr)
mgr = ProcessSpawner(Manager, name="manager", addr="tcp://127.0.0.1:*")
mcli = RPCClient(mgr.name, mgr.addr)
# test connection to host
host_name = test_host.name
mcli.connect_host(host_name, test_host.addr)
assert mcli.list_hosts() == [host_name]
# create nodegroup and nodes
assert mcli.list_nodegroups(host_name) == []
mcli.create_nodegroup(host_name, "nodegroup1")
assert mcli.list_nodegroups(host_name) == ["nodegroup1"]
assert mcli.list_nodes("nodegroup1") == []
mcli.create_node("nodegroup1", "node1", "_MyTestNode")
assert mcli.list_nodes("nodegroup1") == ["node1"]
mcli.control_node("node1", "start")
mcli.control_node("node1", "stop")
mcli.delete_node("node1")
assert mcli.list_nodes("nodegroup1") == []
# mcli.close()
# host_cli.close()
mgr.stop()
test_host.stop()
def test_host2():
process_host0 = ProcessSpawner(Host, 'host0', 'tcp://127.0.0.1:*')
client0 = RPCClient('host0', process_host0.addr)
client0.create_nodegroup('nodegroup 0.1', 'tcp://127.0.0.1:*')
client0.create_nodegroup('nodegroup 0.2', 'tcp://127.0.0.1:*')
time.sleep(2.)
client0.close_nodegroup('nodegroup 0.1')
client0.close_nodegroup('nodegroup 0.2')
time.sleep(1.)
process_host0.stop()
