Function test_deploy_ssh_custom_port

tests/test_worker.py:410–421

(hq_env: HqEnv)

def test_deploy_ssh_custom_port(hq_env: HqEnv):
    hq_env.start_server()
    path = "nodefile.txt"
    with open(path, "w") as f:
        print("localhost:1234", file=f)

    with hq_env.mock.mock_program_with_code("ssh", ssh_code_dump_args_to_file("command.txt")) as mock:
        hq_env.command(["worker", "deploy-ssh", path])
        with open("command.txt") as f:
            cmdline = json.load(f)
            cmdline = cmdline[: cmdline.index("--")]
            assert cmdline == [str(mock), "localhost", "-t", "-t", "-p", "1234"]


def prepare_localhost_nodefile(count: int = 1) -> str: