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