def test_is_url_from_allowed_origins_browser_serverAddress(self): with patch('streamlit.server.server_util.config.is_manually_set', side_effect=[True]), \ patch('streamlit.server.server_util.config.get_option', side_effect=[True, 'browser.server.address']): self.assertTrue( is_url_from_allowed_origins('browser.server.address'))
def test_is_url_from_allowed_origins_s3_url(self): with patch("streamlit.server.server_util.config.is_manually_set", side_effect=[True]), patch( "streamlit.server.server_util.config.get_option", side_effect=[True, "s3.amazon.com"], ): self.assertTrue(is_url_from_allowed_origins("s3.amazon.com"))
def test_is_url_from_allowed_origins_browser_serverAddress(self): with patch("streamlit.server.server_util.config.is_manually_set", side_effect=[True]), patch( "streamlit.server.server_util.config.get_option", side_effect=[True, "browser.server.address"], ): self.assertTrue( is_url_from_allowed_origins("browser.server.address"))
def test_is_url_from_allowed_origins_s3_bucket(self): with patch( "streamlit.server.server_util.config.get_option", side_effect=[True, "mybucket"], ): self.assertTrue(is_url_from_allowed_origins("mybucket"))
def test_is_url_from_allowed_origins_CORS_off(self): with patch("streamlit.server.server_util.config.get_option", side_effect=[False]): self.assertTrue(is_url_from_allowed_origins("does not matter"))
def test_is_url_from_allowed_origins_allowed_domains(self): self.assertTrue(is_url_from_allowed_origins("localhost")) self.assertTrue(is_url_from_allowed_origins("127.0.0.1"))
def check_origin(self, origin: str) -> bool: """Set up CORS.""" return super().check_origin(origin) or is_url_from_allowed_origins(origin)
def check_origin(self, origin): """Set up CORS.""" return is_url_from_allowed_origins(origin)
def test_is_url_from_allowed_origins_s3_url(self): with patch('streamlit.server.server_util.config.is_manually_set', side_effect=[True]), \ patch('streamlit.server.server_util.config.get_option', side_effect=[True, 's3.amazon.com']): self.assertTrue(is_url_from_allowed_origins('s3.amazon.com'))
def test_is_url_from_allowed_origins_s3_bucket(self): with patch('streamlit.server.server_util.config.get_option', side_effect=[True, 'mybucket']): self.assertTrue(is_url_from_allowed_origins('mybucket'))
def test_is_url_from_allowed_origins_allowed_domains(self): self.assertTrue(is_url_from_allowed_origins('localhost')) self.assertTrue(is_url_from_allowed_origins('127.0.0.1'))