mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-01 14:45:29 +00:00
76 lines
3.2 KiB
Python
76 lines
3.2 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Any copyright is dedicated to the Public Domain.
|
|
# http://creativecommons.org/publicdomain/zero/1.0/
|
|
|
|
from mozfile import TemporaryDirectory
|
|
import mozhttpd
|
|
import os
|
|
import unittest
|
|
import urllib2
|
|
|
|
class PathTest(unittest.TestCase):
|
|
def try_get(self, url, expected_contents):
|
|
f = urllib2.urlopen(url)
|
|
self.assertEqual(f.getcode(), 200)
|
|
self.assertEqual(f.read(), expected_contents)
|
|
|
|
def try_get_expect_404(self, url):
|
|
with self.assertRaises(urllib2.HTTPError) as cm:
|
|
urllib2.urlopen(url)
|
|
self.assertEqual(404, cm.exception.code)
|
|
|
|
def test_basic(self):
|
|
"""Test that requests to docroot and a path mapping work as expected."""
|
|
with TemporaryDirectory() as d1, TemporaryDirectory() as d2:
|
|
open(os.path.join(d1, "test1.txt"), "w").write("test 1 contents")
|
|
open(os.path.join(d2, "test2.txt"), "w").write("test 2 contents")
|
|
httpd = mozhttpd.MozHttpd(port=0,
|
|
docroot=d1,
|
|
path_mappings={'/files': d2}
|
|
)
|
|
httpd.start(block=False)
|
|
self.try_get(httpd.get_url("/test1.txt"), "test 1 contents")
|
|
self.try_get(httpd.get_url("/files/test2.txt"), "test 2 contents")
|
|
self.try_get_expect_404(httpd.get_url("/files/test2_nope.txt"))
|
|
httpd.stop()
|
|
|
|
def test_substring_mappings(self):
|
|
"""Test that a path mapping that's a substring of another works."""
|
|
with TemporaryDirectory() as d1, TemporaryDirectory() as d2:
|
|
open(os.path.join(d1, "test1.txt"), "w").write("test 1 contents")
|
|
open(os.path.join(d2, "test2.txt"), "w").write("test 2 contents")
|
|
httpd = mozhttpd.MozHttpd(port=0,
|
|
path_mappings={'/abcxyz': d1,
|
|
'/abc': d2,}
|
|
)
|
|
httpd.start(block=False)
|
|
self.try_get(httpd.get_url("/abcxyz/test1.txt"), "test 1 contents")
|
|
self.try_get(httpd.get_url("/abc/test2.txt"), "test 2 contents")
|
|
httpd.stop()
|
|
|
|
def test_multipart_path_mapping(self):
|
|
"""Test that a path mapping with multiple directories works."""
|
|
with TemporaryDirectory() as d1:
|
|
open(os.path.join(d1, "test1.txt"), "w").write("test 1 contents")
|
|
httpd = mozhttpd.MozHttpd(port=0,
|
|
path_mappings={'/abc/def/ghi': d1}
|
|
)
|
|
httpd.start(block=False)
|
|
self.try_get(httpd.get_url("/abc/def/ghi/test1.txt"), "test 1 contents")
|
|
self.try_get_expect_404(httpd.get_url("/abc/test1.txt"))
|
|
self.try_get_expect_404(httpd.get_url("/abc/def/test1.txt"))
|
|
httpd.stop()
|
|
|
|
def test_no_docroot(self):
|
|
"""Test that path mappings with no docroot work."""
|
|
with TemporaryDirectory() as d1:
|
|
httpd = mozhttpd.MozHttpd(port=0,
|
|
path_mappings={'/foo': d1})
|
|
httpd.start(block=False)
|
|
self.try_get_expect_404(httpd.get_url())
|
|
httpd.stop()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|