def test_rotates_a_hundred_ports(self): app = mock.MagicMock() RetriesExceeded = streamlit.server.server.RetriesExceeded with pytest.raises(RetriesExceeded) as pytest_wrapped_e: with patch("streamlit.server.server.HTTPServer", return_value=self.get_httpserver()) as mock_server: start_listening(app) self.assertEqual(pytest_wrapped_e.type, SystemExit) self.assertEqual(pytest_wrapped_e.value.code, errno.EADDRINUSE) self.assertEqual(mock_server.listen.call_count, MAX_PORT_SEARCH_RETRIES)
def test_rotates_one_port(self, patched_server_port_is_manually_set, patched__set_option): app = mock.MagicMock() patched_server_port_is_manually_set.return_value = False with pytest.raises(RetriesExceeded): with patch("streamlit.server.server.HTTPServer", return_value=self.get_httpserver()): start_listening(app) PortRotateOneTest.which_port.assert_called_with(8502) patched__set_option.assert_called_with( "server.port", 8501, config.ConfigOption.STREAMLIT_DEFINITION)
def test_unix_socket(self): app = mock.MagicMock() config.set_option("server.address", "unix://~/fancy-test/testasd") some_socket = object() mock_server = self.get_httpserver() with patch( "streamlit.server.server.HTTPServer", return_value=mock_server), patch.object( tornado.netutil, "bind_unix_socket", return_value=some_socket) as bind_unix_socket, patch.dict( os.environ, {"HOME": "/home/superfakehomedir"}): start_listening(app) bind_unix_socket.assert_called_with( "/home/superfakehomedir/fancy-test/testasd") mock_server.add_socket.assert_called_with(some_socket)