Statistics
| Branch: | Revision:

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
"""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

    
65