← Back to branch summary

~lool/linaro-image-tools/igep-support

« back to all changes in this revision

Viewing changes to linaro_media_create/tests/fixtures.py

  • Committer: Loïc Minier
  • Date: 2011-02-01 17:24:51 UTC
  • mfrom: (274.2.2 temp-tarball)
  • Revision ID: lool@dooz.org-20110201172451-e096jdm96ohzlpvz
Merge lp:~lool/linaro-image-tools/temp-tarball; fixes warnings when running
testsuite.

Show diffs side-by-side

added added

removed removed

Lines of Context:
linaro_media_create/tests/fixtures.py
44
44
 
45
45
class CreateTarballFixture(object):
46
46
 
47
 
    def __init__(self, dir):
48
 
        self.dir = dir
49
 
        self.tarball = os.path.join(self.dir, 'tarball.tar.gz')
 
47
    def __init__(self, basedir, reldir='tarball', filename='tarball.tar.gz'):
 
48
        self.basedir = basedir
 
49
        self.reldir = reldir
 
50
        self.tarball = os.path.join(self.basedir, filename)
50
51
 
51
52
    def setUp(self):
52
53
        # Create gzipped tar archive.
53
 
        args = ['tar', '-czf', self.tarball, self.dir]
 
54
        os.mkdir(os.path.join(self.basedir, reldir))
 
55
        args = ['tar', '-C', self.basedir, '-czf', self.tarball, self.reldir]
54
56
        proc = subprocess.Popen(args)
55
57
        proc.wait()
56
58
 
57
59
    def tearDown(self):
58
60
        if os.path.exists(self.tarball):
59
61
            os.remove(self.tarball)
 
62
        dir = os.path.join(self.basedir, subdir)
 
63
        if os.path.exists(dir):
 
64
            os.rmdir(dir)
60
65
 
61
66
    def get_tarball(self):
62
67
        return self.tarball