예제 #1
0
 def setUp(self):
     self.stdout = StringIO()
     self.stderr = StringIO()
     self.cmd = Command(stdout=self.stdout, stderr=self.stderr)
예제 #2
0
class RunserverTestCase(TestCase):
    """Development server command options."""

    def setUp(self):
        self.stdout = StringIO()
        self.stderr = StringIO()
        self.cmd = Command(stdout=self.stdout, stderr=self.stderr)

    def assert_option(self, name, value):
        self.assertEqual(getattr(self.cmd, name), value)

    def assert_stderr(self, message):
        self.stderr.seek(0)
        self.assertIn(message, self.stderr.read())

    def test_default_options(self):
        """Deifault options for running the server."""
        with patch.object(self.cmd, 'run'):
            self.cmd.handle()
        self.assert_option('addr', '127.0.0.1')
        self.assert_option('port', '8000')
        self.assert_option('use_ipv6', None)

    def test_set_ip(self):
        """Run server on another IP address/port."""
        with patch.object(self.cmd, 'run'):
            self.cmd.handle(addrport='1.2.3.4:5000')
        self.assert_option('addr', '1.2.3.4')
        self.assert_option('port', '5000')
        self.assert_option('use_ipv6', None)

    @patch('asyncio.get_event_loop')
    def test_run(self, mock_loop):
        """Running the server should kick off the aiohttp app in the event loop."""
        self.cmd.handle()
        mock_loop.assert_called_with()
        mock_loop.return_value.run_forever.assert_called_with()

    @patch('asyncio.set_event_loop')
    @patch('asyncio.new_event_loop')
    def test_auto_reloader(self, mock_loop, mock_set_loop):
        """Running with the reloader thread creates a new event loop."""
        # Need to setup command options and use inner_run to prevent the
        # background thread from actually kicking off.
        self.cmd.addr = '127.0.0.1'
        self.cmd.port = '8000'
        self.cmd._raw_ipv6 = False
        self.cmd.inner_run(use_reloader=True)
        mock_loop.assert_called_with()
        mock_set_loop.assert_called_with(mock_loop.return_value)
        mock_loop.return_value.run_forever.assert_called_with()

    @patch('asyncio.get_event_loop')
    def test_handle_general_socket_errors(self, mock_loop):
        """Handle socket errors when createing the server."""
        mock_loop.return_value.create_server.side_effect = OSError('OS is broken')
        with patch('os._exit') as mock_exit:
            self.cmd.handle()
            mock_exit.assert_called_with(1)
        self.assert_stderr('OS is broken')

    @patch('asyncio.get_event_loop')
    def test_handle_known_socket_errors(self, mock_loop):
        """Special case socket errors for more meaningful error messages."""
        cases = (
            (errno.EACCES, 'You don\'t have permission to access that port.'),
            (errno.EADDRINUSE, 'That port is already in use.'),
            (errno.EADDRNOTAVAIL, 'That IP address can\'t be assigned to.'),
        )
        for number, message in cases:
            error = OSError()
            error.errno = number
            mock_loop.return_value.create_server.side_effect = error
            with patch('os._exit') as mock_exit:
                self.cmd.handle()
                mock_exit.assert_called_with(1)
            self.assert_stderr(message)

    @patch('asyncio.get_event_loop')
    def test_keyboard_stop(self, mock_loop):
        """User should be able to stop the server with a KeyboardInterrupt."""
        mock_loop.return_value.run_forever.side_effect = KeyboardInterrupt
        with self.assertRaises(SystemExit):
            self.cmd.handle()