summaryrefslogtreecommitdiffstats
path: root/testing/mozbase/mozhttpd/tests/filelisting.py
blob: 6abea757f2a66f64b9e878ff6754c44b6a27ebc0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/usr/bin/env python

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
# You can obtain one at http://mozilla.org/MPL/2.0/.

import mozhttpd
import urllib2
import os
import unittest
import re

here = os.path.dirname(os.path.abspath(__file__))


class FileListingTest(unittest.TestCase):

    def check_filelisting(self, path=''):
        filelist = os.listdir(here)

        httpd = mozhttpd.MozHttpd(port=0, docroot=here)
        httpd.start(block=False)
        f = urllib2.urlopen("http://%s:%s/%s" % ('127.0.0.1', httpd.httpd.server_port, path))
        for line in f.readlines():
            webline = re.sub('\<[a-zA-Z0-9\-\_\.\=\"\'\/\\\%\!\@\#\$\^\&\*\(\) ]*\>',
                             '', line.strip('\n')).strip('/').strip().strip('@')

            if webline and not webline.startswith("Directory listing for"):
                self.assertTrue(webline in filelist,
                                "File %s in dir listing corresponds to a file" % webline)
                filelist.remove(webline)
        self.assertFalse(
            filelist, "Should have no items in filelist (%s) unaccounted for" % filelist)

    def test_filelist(self):
        self.check_filelisting()

    def test_filelist_params(self):
        self.check_filelisting('?foo=bar&fleem=&foo=fleem')


if __name__ == '__main__':
    unittest.main()