root/branches/mk/tests/functional/test_cleaning_up.py

from glob import glob
import os
import tempfile
from StringIO import StringIO

from _path_cheesecake import CHEESECAKE_PATH
from _helper_cheesecake import read_file_contents

try:
    import subprocess
except ImportError, ex:
    from cheesecake import subprocess


class TestCleaningUp(object):
    def setUp(self):
        self.stdout_fd, self.stdout_name = tempfile.mkstemp(prefix='functional') 
        self.stderr_fd, self.stderr_name = tempfile.mkstemp(prefix='functional')
        self.sandbox = tempfile.mkdtemp()
        self.process = subprocess.Popen('%s -n ftputil -s %s' % \
                                        (CHEESECAKE_PATH, self.sandbox),
                         stdout=self.stdout_fd,
                         stderr=self.stderr_fd,
                         shell=True)
        self.return_code = self.process.wait()

    def tearDown(self):
        os.unlink(self.stdout_name)
        os.unlink(self.stderr_name)

    def test_no_tmp(self):
        "Check that no files are left in temp by Cheesecake."
        print file(self.stdout_name).read()
        print file(self.stderr_name).read()

        # Check that Cheesecake exited sucessfully.
        assert self.return_code == 0

        # Check that Cheesecake didn't wrote anything into stderr.
        assert read_file_contents(self.stderr_name) == ''

        # Check that Cheesecake didn't left sandbox.
        assert not os.path.exists(self.sandbox)

        # Check that Cheesecake didn't left any cheesecake* files.
        assert glob(os.path.join(tempfile.gettempdir(), "cheesecake*")) == []

        # Check that Cheesecake didn't left any tmp* files
        assert glob(os.path.join(tempfile.gettempdir(), tempfile.gettempprefix() + "*")) == []