def test_serve_static_content(self):
# PYPI Mocked server can serve static content from disk.
def uses_local_files_for(server, url_path):
"""Test that files are served statically (eg. the output from the
server is the same than the one made by a simple file read.
"""
url = server.full_address + url_path
request = urllib2.Request(url)
response = urllib2.urlopen(request)
file = open(PYPI_DEFAULT_STATIC_PATH + "/test_pypi_server" +
url_path)
return response.read() == file.read()
server = PyPIServer(static_uri_paths=["simple", "external"],
static_filesystem_paths=["test_pypi_server"])
server.start()
# the file does not exists on the disc, so it might not be served
url = server.full_address + "/simple/unexisting_page"
request = urllib2.Request(url)
try:
urllib2.urlopen(request)
except urllib2.HTTPError,e:
self.assertEqual(e.code, 404)
def test_uses_mirrors(self):
# When the main repository seems down, try using the given mirrors"""
server = PyPIServer("foo_bar_baz")
mirror = PyPIServer("foo_bar_baz")
mirror.start() # we dont start the server here
try:
# create the index using both servers
crawler = Crawler(server.full_address + "/simple/",
hosts=('*',), timeout=1, # set the timeout to 1s for the tests
mirrors=[mirror.full_address])
# this should not raise a timeout
self.assertEqual(4, len(crawler.get_releases("foo")))
finally:
mirror.stop()
def test_records_requests(self):
# We expect that PyPIServer can log our requests
server = PyPIServer()
server.default_response_status = 200
try:
server.start()
self.assertEqual(len(server.requests), 0)
data = b'Rock Around The Bunker'
headers = {"X-test-header": "Mister Iceberg"}
request = urllib.request.Request(server.full_address, data,
headers)
urllib.request.urlopen(request)
self.assertEqual(len(server.requests), 1)
handler, request_data = server.requests[-1]
self.assertIn(data, request_data)
self.assertIn("x-test-header", handler.headers)
self.assertEqual(handler.headers["x-test-header"],
"Mister Iceberg")
finally:
server.stop()
def test_records_requests(self):
# We expect that PyPIServer can log our requests
server = PyPIServer()
server.start()
self.assertEqual(len(server.requests), 0)
data = "Rock Around The Bunker"
headers = {"X-test-header": "Mister Iceberg"}
request = urllib2.Request(server.full_address, data, headers)
urllib2.urlopen(request)
self.assertEqual(len(server.requests), 1)
handler, request_data = server.requests[-1]
self.assertIn("Rock Around The Bunker", request_data)
self.assertIn("x-test-header", handler.headers.dict)
self.assertEqual(handler.headers.dict["x-test-header"],
"Mister Iceberg")
server.stop()
def test_uses_mirrors(self):
# When the main repository seems down, try using the given mirrors"""
server = PyPIServer("foo_bar_baz")
mirror = PyPIServer("foo_bar_baz")
mirror.start() # we dont start the server here
try:
# create the index using both servers
crawler = Crawler(
server.full_address + "/simple/",
hosts=('*', ),
# set the timeout to 1s for the tests
timeout=1,
mirrors=[mirror.full_address])
# this should not raise a timeout
self.assertEqual(4, len(crawler.get_releases("foo")))
finally:
mirror.stop()
server.stop()
def test_serve_static_content(self):
# PYPI Mocked server can serve static content from disk.
def uses_local_files_for(server, url_path):
"""Test that files are served statically (eg. the output from the
server is the same than the one made by a simple file read.
"""
url = server.full_address + url_path
request = urllib.request.Request(url)
response = urllib.request.urlopen(request)
with open(PYPI_DEFAULT_STATIC_PATH + "/test_pypi_server" +
url_path) as file:
return response.read().decode() == file.read()
server = PyPIServer(static_uri_paths=["simple", "external"],
static_filesystem_paths=["test_pypi_server"])
server.start()
try:
# the file does not exists on the disc, so it might not be served
url = server.full_address + "/simple/unexisting_page"
request = urllib.request.Request(url)
try:
urllib.request.urlopen(request)
except urllib.error.HTTPError as e:
self.assertEqual(e.code, 404)
# now try serving a content that do exists
self.assertTrue(uses_local_files_for(server, "/simple/index.html"))
# and another one in another root path
self.assertTrue(
uses_local_files_for(server, "/external/index.html"))
finally:
server.stop()
def test_records_requests(self):
# We expect that PyPIServer can log our requests
server = PyPIServer()
server.default_response_status = 200
try:
server.start()
self.assertEqual(len(server.requests), 0)
data = 'Rock Around The Bunker'
headers = {"X-test-header": "Mister Iceberg"}
request = urllib2.Request(server.full_address, data, headers)
urllib2.urlopen(request)
self.assertEqual(len(server.requests), 1)
handler, request_data = server.requests[-1]
self.assertIn(data, request_data)
self.assertIn("x-test-header", handler.headers)
self.assertEqual(handler.headers["x-test-header"],
"Mister Iceberg")
finally:
server.stop()
