Method test_close_connection

tests/test_cli.py:91–102

(self)

        )

    def test_close_connection(self):
        def wait_handler(websocket):
            with self.assertRaises(ConnectionClosed):
                websocket.recv()

        with run_server(wait_handler) as server:
            server_uri = get_uri(server)
            output = self.run_main([server_uri], "", close_input=True)
        self.assertEqual(
            remove_commands_and_prompts(output),
            add_connection_messages("", server_uri),
        )

    def test_connection_failure(self):
        output = self.run_main(["ws://localhost:54321"], expected_exit_code=1)