#!/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()
