root / env / lib / python2.7 / site-packages / distribute-0.6.19-py2.7.egg / setuptools / tests / test_upload_docs.py @ 1a305335
History | View | Annotate | Download (1.84 KB)
1 | 1a305335 | officers | """build_ext tests
|
---|---|---|---|
2 | """
|
||
3 | import sys, os, shutil, tempfile, unittest, site, zipfile |
||
4 | from setuptools.command.upload_docs import upload_docs |
||
5 | from setuptools.dist import Distribution |
||
6 | |||
7 | SETUP_PY = """\
|
||
8 | from setuptools import setup
|
||
9 |
|
||
10 | setup(name='foo')
|
||
11 | """
|
||
12 | |||
13 | class TestUploadDocsTest(unittest.TestCase): |
||
14 | def setUp(self): |
||
15 | self.dir = tempfile.mkdtemp()
|
||
16 | setup = os.path.join(self.dir, 'setup.py') |
||
17 | f = open(setup, 'w') |
||
18 | f.write(SETUP_PY) |
||
19 | f.close() |
||
20 | self.old_cwd = os.getcwd()
|
||
21 | os.chdir(self.dir)
|
||
22 | |||
23 | self.upload_dir = os.path.join(self.dir, 'build') |
||
24 | os.mkdir(self.upload_dir)
|
||
25 | |||
26 | # A test document.
|
||
27 | f = open(os.path.join(self.upload_dir, 'index.html'), 'w') |
||
28 | f.write("Hello world.")
|
||
29 | f.close() |
||
30 | |||
31 | # An empty folder.
|
||
32 | os.mkdir(os.path.join(self.upload_dir, 'empty')) |
||
33 | |||
34 | if sys.version >= "2.6": |
||
35 | self.old_base = site.USER_BASE
|
||
36 | site.USER_BASE = upload_docs.USER_BASE = tempfile.mkdtemp() |
||
37 | self.old_site = site.USER_SITE
|
||
38 | site.USER_SITE = upload_docs.USER_SITE = tempfile.mkdtemp() |
||
39 | |||
40 | def tearDown(self): |
||
41 | os.chdir(self.old_cwd)
|
||
42 | shutil.rmtree(self.dir)
|
||
43 | if sys.version >= "2.6": |
||
44 | shutil.rmtree(site.USER_BASE) |
||
45 | shutil.rmtree(site.USER_SITE) |
||
46 | site.USER_BASE = self.old_base
|
||
47 | site.USER_SITE = self.old_site
|
||
48 | |||
49 | def test_create_zipfile(self): |
||
50 | # Test to make sure zipfile creation handles common cases.
|
||
51 | # This explicitly includes a folder containing an empty folder.
|
||
52 | |||
53 | dist = Distribution() |
||
54 | |||
55 | cmd = upload_docs(dist) |
||
56 | cmd.upload_dir = self.upload_dir
|
||
57 | zip_file = cmd.create_zipfile() |
||
58 | |||
59 | assert zipfile.is_zipfile(zip_file)
|
||
60 | |||
61 | zip_f = zipfile.ZipFile(zip_file) # woh...
|
||
62 | |||
63 | assert zip_f.namelist() == ['index.html'] |
||
64 |