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)
