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()