Exemplo n.º 1
0
def test_setup_interactive(fake_inter):
    """Ensure we can build the connection pool of interactive hosts."""

    runner = Bladerunner()
    assert runner.interactive_hosts == {}

    with patch.object(runner, "interactive", return_value=fake_inter) as inter:
        runner.setup_interactive("fake_place", connect=False)

    assert fake_inter.server in runner.interactive_hosts
    inter.assert_called_once_with("fake_place", False)
Exemplo n.º 2
0
def test_setup_interactive(fake_inter):
    """Ensure we can build the connection pool of interactive hosts."""

    runner = Bladerunner()
    assert runner.interactive_hosts == {}

    with patch.object(runner, "interactive", return_value=fake_inter) as inter:
        runner.setup_interactive("fake_place", connect=False)

    assert fake_inter.server in runner.interactive_hosts
    inter.assert_called_once_with("fake_place", False)
Exemplo n.º 3
0
def test_setup_interactive_many(fake_inter):
    """We can initialize many hosts interactively at once."""

    runner = Bladerunner()
    assert runner.interactive_hosts == {}

    with patch.object(runner, "interactive", return_value=fake_inter) as inter:
        runner.setup_interactive(["fake", "place", "fake"], connect=False)

    assert inter.call_count == 2
    inter.assert_any_call("fake", False)
    inter.assert_any_call("place", False)
Exemplo n.º 4
0
def test_setup_interactive_many(fake_inter):
    """We can initialize many hosts interactively at once."""

    runner = Bladerunner()
    assert runner.interactive_hosts == {}

    with patch.object(runner, "interactive", return_value=fake_inter) as inter:
        runner.setup_interactive(["fake", "place", "fake"], connect=False)

    assert inter.call_count == 2
    inter.assert_any_call("fake", False)
    inter.assert_any_call("place", False)
Exemplo n.º 5
0
def test_end_interactive_file():
    """The end_interactive method can also be passed a file of hostnames."""

    runner = Bladerunner()
    runner.setup_interactive(
        ["one", "two", "three", "four", "five", "six"],
        connect=False,
    )

    hostfp = tempfile.mktemp()
    with open(hostfp, "w") as hostsfile:
        hostsfile.write("one\nfour\nsix\n")

    runner.end_interactive(hostfp)

    for host in runner.interactive_hosts.keys():
        assert host in ["two", "three", "five"]

    assert len(runner.interactive_hosts) == 3

    os.remove(hostfp)
Exemplo n.º 6
0
def test_interactive_hostsfile():
    """If passed a string file path as an arg, read the hosts per line."""

    hostsfp = tempfile.mktemp()
    test_hosts = ["something", "some", "thing", "else", "stuff"]

    with open(hostsfp, "w") as hostsfile:
        for test_host in test_hosts:
            hostsfile.write("{0}\n".format(test_host))

    runner = Bladerunner()
    runner.setup_interactive(hostsfp, connect=False)
    setup_hosts = runner.interactive_hosts.keys()

    for test_host in test_hosts:
        assert test_host in setup_hosts

    for setup_host in setup_hosts:
        assert setup_host in test_hosts

    os.remove(hostsfp)
Exemplo n.º 7
0
def test_end_interactive_file():
    """The end_interactive method can also be passed a file of hostnames."""

    runner = Bladerunner()
    runner.setup_interactive(
        ["one", "two", "three", "four", "five", "six"],
        connect=False,
    )

    hostfp = tempfile.mktemp()
    with open(hostfp, "w") as hostsfile:
        hostsfile.write("one\nfour\nsix\n")

    runner.end_interactive(hostfp)

    for host in runner.interactive_hosts.keys():
        assert host in ["two", "three", "five"]

    assert len(runner.interactive_hosts) == 3

    os.remove(hostfp)
Exemplo n.º 8
0
def test_interactive_hostsfile():
    """If passed a string file path as an arg, read the hosts per line."""

    hostsfp = tempfile.mktemp()
    test_hosts = ["something", "some", "thing", "else", "stuff"]

    with open(hostsfp, "w") as hostsfile:
        for test_host in test_hosts:
            hostsfile.write("{0}\n".format(test_host))

    runner = Bladerunner()
    runner.setup_interactive(hostsfp, connect=False)
    setup_hosts = runner.interactive_hosts.keys()

    for test_host in test_hosts:
        assert test_host in setup_hosts

    for setup_host in setup_hosts:
        assert setup_host in test_hosts

    os.remove(hostsfp)