def test_try_connection_calls_omshell_correctly(self):
server_address = factory.make_string()
shell = Omshell(server_address, "", ipv6=self.ipv6)
# Instead of calling a real omshell, we'll just record the
# parameters passed to Popen.
recorder = FakeMethod(result=(0, b"obj: <null>"))
shell._run = recorder
shell.try_connection()
expected_script = dedent("""\
server {server}
port {port}
connect
""")
expected_script = expected_script.format(server=server_address,
port=self.port)
# Check that the 'stdin' arg contains the correct set of
# commands.
self.assertEqual(
[1, (expected_script.encode("utf-8"), )],
[recorder.call_count,
recorder.extract_args()[0]],
)
def test_try_connection_returns_False(self):
server_address = factory.make_string()
shell = Omshell(server_address, "", ipv6=self.ipv6)
# Instead of calling a real omshell, we'll just record the
# parameters passed to Popen.
recorder = FakeMethod(result=(0, factory.make_bytes()))
shell._run = recorder
self.assertFalse(shell.try_connection())