patool/tests/__init__.py

67 lines
2.4 KiB
Python
Raw Normal View History

2010-02-21 11:14:57 +00:00
# -*- coding: utf-8 -*-
# Copyright (C) 2010 Bastian Kleineidam
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import unittest
import os
import shutil
import nose
import patoolib
from distutils.spawn import find_executable
basedir = os.path.dirname(__file__)
datadir = os.path.join(basedir, 'data')
class ArchiveTest (unittest.TestCase):
"""Helper class for achive tests."""
2010-02-21 12:40:42 +00:00
def archive_commands (self, filename, cmd):
2010-02-21 11:14:57 +00:00
self.archive_list(filename, cmd)
2010-02-21 12:40:42 +00:00
self.archive_test(filename, cmd)
2010-02-21 11:14:57 +00:00
self.archive_extract(filename, cmd)
def archive_extract (self, filename, cmd):
archive = os.path.join(datadir, filename)
# create a temporary directory for extraction
tmpdir = patoolib.util.tmpdir(dir=basedir)
os.chdir(tmpdir)
try:
2010-02-21 12:40:42 +00:00
patoolib._handle_archive(archive, 'extract', program=cmd)
patoolib._handle_archive(archive, 'extract', program=cmd, force=True)
2010-02-21 11:14:57 +00:00
finally:
os.chdir(basedir)
shutil.rmtree(tmpdir)
def archive_list (self, filename, cmd):
archive = os.path.join(datadir, filename)
2010-02-21 12:40:42 +00:00
patoolib._handle_archive(archive, 'list', program=cmd)
patoolib._handle_archive(archive, 'list', program=cmd, verbose=True)
def archive_test (self, filename, cmd):
archive = os.path.join(datadir, filename)
patoolib._handle_archive(archive, 'test', program=cmd)
patoolib._handle_archive(archive, 'test', program=cmd, verbose=True)
2010-02-21 11:14:57 +00:00
def needs_cmd (cmd):
"""Decorator skipping test if given command is not available."""
def check_prog (f):
def newfunc (*args, **kwargs):
if not find_executable(cmd):
raise nose.SkipTest("command `%s' not available" % cmd)
return f(*args, **kwargs)
newfunc.func_name = f.func_name
return newfunc
return check_prog