view test/filesupport.py @ 171:c0ea6cbb0fd2

Add "--debbuildopts -b" to "pbuilder build" command line to stop pbuilder from creating a source package. The .changes would otherwise contain references to that new source package instead of the one we passed to pbuilder. The checksums for the two source packages would be different so the .changes file would not match the source package that treepkg produces.
author Bernhard Herzog <bh@intevation.de>
date Mon, 23 Jun 2008 16:12:01 +0000
parents 007d7f2aa184
children e1c7cd896310
line wrap: on
line source
# Copyright (C) 2007, 2008 by Intevation GmbH
# Authors:
# Bernhard Herzog <bh@intevation.de>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with the software for details.

"""Support code for the test cases"""

import os
import shutil
from treepkg.util import writefile


def create_temp_dir():
    """Create a temporary directory for the test-suite and return its name.

    The temporary directory is always called temp and is created in the
    directory where the support module is located.

    If the temp directory already exists, just return the name.
    """
    name = os.path.abspath(os.path.join(os.path.dirname(__file__), "temp"))

    # if the directory already exists, we're done
    if os.path.isdir(name):
        return name

    # create the directory
    os.mkdir(name)
    return name


class FileTestMixin:

    """Mixin class for tests that use files in the temporary directory
    """

    def temp_file_name(self, basename, remove=False):
        """Returns the full name of the file named basename in the temp. dir.
        If the remove parameter is true, the file is removed if already exists.
        """
        filename = os.path.join(create_temp_dir(), basename)
        if remove and os.path.exists(filename):
            os.remove(filename)
        return filename

    def create_temp_file(self, basename, contents, mode = None):
        """Creates a file in the temp directory with the given contents.
        The optional parameter mode should either be None (the default)
        or an int specifying the file permissions (same format as the
        second parameter of os.chmod).  The method returns the absolute
        name of the created file.
        """
        filename = self.temp_file_name(basename)
        file = open(filename, "w")
        file.write(contents)
        file.close()
        if mode is not None:
            os.chmod(filename, mode)
        return filename

    def create_temp_dir(self, basename, remove=True):
        """Creates the directory basename in the temporary directory.
        If the optional parameter remove is true (the default), the
        directory and all its contents are deleted with shutil.rmtree.
        The method returns the absolute name of the created directory.
        """
        dirname = os.path.join(create_temp_dir(), basename)
        if remove and os.path.exists(dirname):
            shutil.rmtree(dirname)
        os.mkdir(dirname)
        return dirname

    def create_files(self, directory, filedesc):
        """Creates a hierarchy of directories and files in directory.
        The filedesc parameter should be a sequence of (name, contents)
        pairs or (name, permissions, contents) triples.  Each item of
        the sequence describes one entry of the directory.  If contents
        is an instance of list, the entry is a subdirectory and the
        contents is a list in the same format as filedesc and passed
        recursively to the create_files method.  If contents is a
        string, the new directory entry is a normal file and contents is
        the contents of the file.  The permissions if present, should be
        an int specifying the files permissions (setting permissions
        doesn't work yet for directories).
        """
        shutil.rmtree(directory, True)
        os.makedirs(directory)
        for item in filedesc:
            if len(item) == 3:
                name, permissions, contents = item
            else:
                name, contents = item
                permissions = None # use default permissions
            if isinstance(contents, list):
                # a list as contents indicates a directory
                newdir = os.path.join(directory, name)
                os.mkdir(newdir)
                self.create_files(newdir, contents)
            else:
                writefile(os.path.join(directory, name), contents, permissions)

    def checkFileContents(self, filename, contents):
        """check the contents of a file"""
        self.assertEquals(open(filename).read(), contents)
This site is hosted by Intevation GmbH (Datenschutzerklärung und Impressum | Privacy Policy and Imprint)