testing/mozbase/mozhttpd/tests/paths.py@6474c204b198
testing/mozbase/mozhttpd/tests/paths.py
Wed, 31 Dec 2014 06:09:35 +0100
- author
- Michael Schloh von Bennewitz <michael@schloh.com>
- date
- Wed, 31 Dec 2014 06:09:35 +0100
- changeset 0
- 6474c204b198
- permissions
- -rw-r--r--
Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.
1 #!/usr/bin/env python
3 # Any copyright is dedicated to the Public Domain.
4 # http://creativecommons.org/publicdomain/zero/1.0/
6 from mozfile import TemporaryDirectory
7 import mozhttpd
8 import os
9 import unittest
10 import urllib2
12 class PathTest(unittest.TestCase):
13 def try_get(self, url, expected_contents):
14 f = urllib2.urlopen(url)
15 self.assertEqual(f.getcode(), 200)
16 self.assertEqual(f.read(), expected_contents)
18 def try_get_expect_404(self, url):
19 with self.assertRaises(urllib2.HTTPError) as cm:
20 urllib2.urlopen(url)
21 self.assertEqual(404, cm.exception.code)
23 def test_basic(self):
24 """Test that requests to docroot and a path mapping work as expected."""
25 with TemporaryDirectory() as d1, TemporaryDirectory() as d2:
26 open(os.path.join(d1, "test1.txt"), "w").write("test 1 contents")
27 open(os.path.join(d2, "test2.txt"), "w").write("test 2 contents")
28 httpd = mozhttpd.MozHttpd(port=0,
29 docroot=d1,
30 path_mappings={'/files': d2}
31 )
32 httpd.start(block=False)
33 self.try_get(httpd.get_url("/test1.txt"), "test 1 contents")
34 self.try_get(httpd.get_url("/files/test2.txt"), "test 2 contents")
35 self.try_get_expect_404(httpd.get_url("/files/test2_nope.txt"))
36 httpd.stop()
38 def test_substring_mappings(self):
39 """Test that a path mapping that's a substring of another works."""
40 with TemporaryDirectory() as d1, TemporaryDirectory() as d2:
41 open(os.path.join(d1, "test1.txt"), "w").write("test 1 contents")
42 open(os.path.join(d2, "test2.txt"), "w").write("test 2 contents")
43 httpd = mozhttpd.MozHttpd(port=0,
44 path_mappings={'/abcxyz': d1,
45 '/abc': d2,}
46 )
47 httpd.start(block=False)
48 self.try_get(httpd.get_url("/abcxyz/test1.txt"), "test 1 contents")
49 self.try_get(httpd.get_url("/abc/test2.txt"), "test 2 contents")
50 httpd.stop()
52 def test_multipart_path_mapping(self):
53 """Test that a path mapping with multiple directories works."""
54 with TemporaryDirectory() as d1:
55 open(os.path.join(d1, "test1.txt"), "w").write("test 1 contents")
56 httpd = mozhttpd.MozHttpd(port=0,
57 path_mappings={'/abc/def/ghi': d1}
58 )
59 httpd.start(block=False)
60 self.try_get(httpd.get_url("/abc/def/ghi/test1.txt"), "test 1 contents")
61 self.try_get_expect_404(httpd.get_url("/abc/test1.txt"))
62 self.try_get_expect_404(httpd.get_url("/abc/def/test1.txt"))
63 httpd.stop()
65 def test_no_docroot(self):
66 """Test that path mappings with no docroot work."""
67 with TemporaryDirectory() as d1:
68 httpd = mozhttpd.MozHttpd(port=0,
69 path_mappings={'/foo': d1})
70 httpd.start(block=False)
71 self.try_get_expect_404(httpd.get_url())
72 httpd.stop()
74 if __name__ == '__main__':
75 unittest.main()
