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
44
45
46
47
48
49
50
|
#!/usr/bin/env python
import os
import mozfile
import mozhttpd
import mozunit
import pytest
@pytest.fixture(name="files")
def fixture_files():
"""Return a list of tuples with name and binary_string."""
return [("small", os.urandom(128)), ("large", os.urandom(16384))]
@pytest.fixture(name="docroot")
def fixture_docroot(tmpdir, files):
"""Yield a str path to docroot."""
docroot = tmpdir.mkdir("docroot")
for name, binary_string in files:
filename = docroot.join(name)
filename.write_binary(binary_string)
yield str(docroot)
docroot.remove()
@pytest.fixture(name="httpd_url")
def fixture_httpd_url(docroot):
"""Yield the URL to a started MozHttpd server."""
httpd = mozhttpd.MozHttpd(docroot=docroot)
httpd.start()
yield httpd.get_url()
httpd.stop()
def test_basic(httpd_url, files):
"""Test that mozhttpd can serve files."""
# Retrieve file and check contents matchup
for name, binary_string in files:
retrieved_content = mozfile.load(httpd_url + name).read()
assert retrieved_content == binary_string
if __name__ == "__main__":
mozunit.main()
|
