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