def test_main_manage(self, mock__args):
_args = {
"config_file": "/etc/directord/config.yaml",
"zmq_shared_key": None,
"zmq_curve_encryption": False,
"debug": False,
"driver": "grpcd",
"export_jobs": None,
"export_nodes": None,
"job_info": None,
"job_port": 5555,
"backend_port": 5556,
"heartbeat_interval": 60,
"socket_path": "/var/run/directord.sock",
"stream": False,
"cache_path": "/var/cache/directord",
"list_jobs": False,
"list_nodes": True,
"mode": "manage",
"purge_jobs": False,
"purge_nodes": False,
"identity": None,
"grpc_port": 5558,
"grpc_server_address": "127.0.0.1",
}
parsed_args = namedtuple("NameSpace", _args.keys())(*_args.values())
mock__args.return_value = [parsed_args, mock.MagicMock()]
with patch("directord.user.Manage.run", autospec=True) as d:
d.return_value = {}
main.main()
def test_main_orchestrate(self, mock__args):
_args = {
"config_file": "/etc/directord/config.yaml",
"zmq_shared_key": None,
"zmq_curve_encryption": False,
"debug": False,
"driver": "grpcd",
"finger_print": False,
"job_port": 5555,
"backend_port": 5556,
"ignore_cache": False,
"heartbeat_interval": 60,
"socket_path": "/var/run/directord.sock",
"stream": False,
"cache_path": "/var/cache/directord",
"mode": "orchestrate",
"target": None,
"wait": False,
"orchestrate_files": ["file1 file2"],
"poll": False,
"restrict": None,
"identity": None,
}
parsed_args = namedtuple("NameSpace", _args.keys())(*_args.values())
mock__args.return_value = [parsed_args, mock.MagicMock()]
with patch("directord.mixin.Mixin.run_orchestration", autospec=True):
main.main()
def test_main_server(self, mock__args):
_args = {
"config_file": "/etc/directord/config.yaml",
"zmq_shared_key": None,
"zmq_curve_encryption": False,
"debug": False,
"driver": "grpcd",
"job_port": 5555,
"backend_port": 5556,
"heartbeat_interval": 60,
"socket_path": "/var/run/directord.sock",
"stream": False,
"cache_path": "/var/cache/directord",
"mode": "server",
"identity": None,
}
parsed_args = namedtuple("NameSpace", _args.keys())(*_args.values())
mock__args.return_value = [parsed_args, mock.MagicMock()]
with patch("directord.server.Server", autospec=True):
main.main()
def test_main_client(self, mock__args, mock__client):
_args = {
"config_file": "/etc/directord/config.yaml",
"zmq_shared_key": None,
"zmq_curve_encryption": False,
"debug": False,
"driver": "grpcd",
"job_port": 5555,
"backend_port": 5556,
"heartbeat_interval": 60,
"socket_path": "/var/run/directord.sock",
"stream": False,
"cache_path": "/var/cache/directord",
"mode": "client",
"identity": "client1",
}
parsed_args = namedtuple("NameSpace", _args.keys())(*_args.values())
mock__args.return_value = [parsed_args, mock.MagicMock()]
main.main()
mock__client.assert_called_once_with(args=parsed_args)
from directord import main main.main()