def test_directory_requested(self):
"""Map a URI containing a request for a directory."""
message, mimetype = map_uri(self.directory_requested)
contents = listdir("webroot" + self.directory_requested)
content_string = "\n".join(contents)
self.assertEqual(message, content_string)
self.assertEqual(mimetype, "text/plain")
def test_text_file_requested(self):
"""Map a URI containing a request for a text file."""
message, mimetype = map_uri(self.text_requested)
with open("webroot" + self.text_requested, "rb") as infile:
expected = infile.read()
self.assertEqual(message, expected)
self.assertEqual(mimetype, "text/plain")
def test_image_file_requested(self):
"""Map a URI containing a request for an image file."""
message, mimetype = map_uri(self.image_requested)
with open("webroot" + self.image_requested, "rb") as infile:
expected = infile.read()
self.assertEqual(message, expected)
self.assertEqual(mimetype, "image/png")
