test_serve.py
from . import serve
def test_make_hosts_file():
hosts = serve.make_hosts_file({
"domains": {"www": "www.foo.bar.test", "www1": "www1.foo.bar.test"},
"not_domains": {"aaa": "aaa.foo.bar.test", "bbb": "bbb.foo.bar.test"}
}, "127.1.1.1")
lines = hosts.split("\n")
assert "127.1.1.1\twww.foo.bar.test" in lines
assert "127.1.1.1\twww1.foo.bar.test" in lines
assert "0.0.0.0\taaa.foo.bar.test" in lines
assert "0.0.0.0\tbbb.foo.bar.test" in lines