# -*- 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/>.
from tests import ArchiveTest, needs_program, needs_codec

class TestArchives (ArchiveTest):

    @needs_program('file')
    @needs_program('tar')
    def test_tar_file (self):
        self.program = 'tar'
        self.archive_commands('t.tar.foo', format="tar")

    @needs_program('file')
    @needs_codec('tar', 'gzip')
    def test_tar_gz_file (self):
        self.program = 'tar'
        self.archive_commands('t.tar.gz.foo', format="tar", encoding="gzip")
        self.archive_commands('t.tgz.foo', format="tar", encoding="gzip")

    @needs_program('file')
    @needs_codec('tar', 'compress')
    def test_tar_z (self):
        self.program = 'tar'
        self.archive_commands('t.tar.Z.foo', format="tar", encoding="compress")
        self.archive_commands('t.taz.foo', format="tar", encoding="compress")

    @needs_program('file')
    @needs_codec('tar', 'bzip2')
    def test_tar_bz2 (self):
        self.program = 'tar'
        self.archive_commands('t.tar.bz2.foo', format="tar", encoding="bzip2")
        self.archive_commands('t.tbz2.foo', format="tar", encoding="bzip2")

    # file(1) does not recognize .lzma files
    #@needs_program('file')
    #@needs_codec('tar', 'lzma')
    #def test_tar_lzma (self):
    #    self.program = 'tar'
    #    self.archive_commands('t.tar.lzma.foo', format="tar", encoding="lzma")

    @needs_program('file')
    @needs_codec('tar', 'lzip')
    def test_tar_lzip (self):
        self.program = 'tar'
        self.archive_commands('t.tar.lz.foo', format="tar", encoding="lzip")

    @needs_program('file')
    @needs_codec('tar', 'xz')
    def test_tar_xz (self):
        self.program = 'tar'
        self.archive_commands('t.tar.xz.foo', format="tar", encoding="xz")

    @needs_program('file')
    @needs_program('star')
    def test_star (self):
        self.program = 'star'
        self.archive_commands('t.tar.foo', format="tar")

    @needs_program('file')
    @needs_codec('tar', 'gzip')
    def test_star_gz (self):
        self.program = 'star'
        self.archive_commands('t.tar.gz.foo', format="tar", encoding="gzip")
        self.archive_commands('t.tgz.foo', format="tar", encoding="gzip")

    @needs_program('file')
    @needs_codec('tar', 'compress')
    def test_star_z (self):
        self.program = 'star'
        self.archive_commands('t.tar.Z.foo', format="tar", encoding="compress")
        self.archive_commands('t.taz.foo', format="tar", encoding="compress")

    @needs_program('file')
    @needs_codec('tar', 'bzip2')
    def test_star_bz2 (self):
        self.program = 'star'
        self.archive_commands('t.tar.bz2.foo', format="tar", encoding="bzip2")
        self.archive_commands('t.tbz2.foo', format="tar", encoding="bzip2")

    # file(1) does not recognize .lzma files
    #@needs_program('file')
    #@needs_codec('tar', 'lzma')
    #def test_star_lzma (self):
    #    self.program = 'star'
    #    self.archive_commands('t.tar.lzma.foo', format="tar", encoding="lzma")

    @needs_program('file')
    @needs_codec('tar', 'lzip')
    def test_star_lzip (self):
        self.program = 'star'
        self.archive_commands('t.tar.lz.foo', format="tar", encoding="lzip")

    @needs_program('file')
    @needs_codec('tar', 'xz')
    def test_star_xz (self):
        self.program = 'star'
        self.archive_commands('t.tar.xz.foo', format="tar", encoding="xz")

    @needs_program('file')
    @needs_program('bzip2')
    def test_bzip2 (self):
        self.program = 'bzip2'
        self.archive_extract('t.bz2.foo')
        self.archive_test('t.bz2.foo')
        self.archive_create('t.bz2.foo', format="bzip2", singlefile=True)

    @needs_program('file')
    @needs_program('pbzip2')
    def test_pbzip2 (self):
        self.program = 'pbzip2'
        self.archive_extract('t.bz2.foo')
        self.archive_test('t.bz2.foo')
        self.archive_create('t.bz2.foo', format="bzip2", singlefile=True)

    @needs_program('file')
    @needs_program('echo')
    def test_echo (self):
        self.program = 'echo'
        self.archive_list('t.bz2.foo')
        self.archive_list('t.Z.foo')
        # file(1) does not recognize .lzma files
        #self.archive_list('t.lzma.foo')
        self.archive_list('t.txt.lz.foo')

    @needs_program('file')
    @needs_program('unzip')
    def test_unzip (self):
        self.program = 'unzip'
        self.archive_extract('t.zip.foo')
        self.archive_list('t.zip.foo')
        self.archive_test('t.zip.foo')
        self.archive_extract('t.jar.foo')
        self.archive_list('t.jar.foo')
        self.archive_test('t.jar.foo')

    @needs_program('file')
    @needs_program('zip')
    def test_zip (self):
        self.program = 'zip'
        self.archive_create('t.zip.foo', format="zip")

    @needs_program('file')
    @needs_program('gzip')
    def test_gzip (self):
        self.program = 'gzip'
        self.archive_commands('t.gz.foo', format="gzip", singlefile=True)
        self.archive_commands('t.txt.gz.foo', format="gzip", singlefile=True)
        self.archive_extract('t.Z.foo')

    @needs_program('file')
    @needs_program('uncompress.real')
    def test_uncompress (self):
        self.program = 'uncompress.real'
        self.archive_extract('t.Z.foo')

    @needs_program('file')
    @needs_program('compress')
    def test_compress (self):
        self.program = 'compress'
        self.archive_create('t.Z.foo', format="compress", singlefile=True)

    @needs_program('file')
    @needs_program('7z')
    def test_p7zip_file (self):
        self.program = '7z'
        self.archive_commands('t.7z.foo', format="7z")
        self.archive_list('t.gz.foo')
        self.archive_list('t.bz2.foo')
        self.archive_list('t.zip.foo')
        self.archive_list('t.jar.foo')
        self.archive_list('t.Z.foo')
        self.archive_list('t.cab.foo')
        self.archive_list('t.arj.foo')
        self.archive_list('t.cpio.foo')
        self.archive_list('t.rpm.foo')
        self.archive_list('t.deb.foo')
        self.archive_extract('t.gz.foo')
        self.archive_extract('t.bz2.foo')
        self.archive_extract('t.zip.foo')
        self.archive_extract('t.jar.foo')
        self.archive_extract('t.Z.foo')
        self.archive_extract('t.cab.foo')
        self.archive_extract('t.arj.foo')
        self.archive_extract('t.cpio.foo')
        self.archive_extract('t.rpm.foo')
        self.archive_extract('t.deb.foo')
        self.archive_test('t.gz.foo')
        self.archive_test('t.bz2.foo')
        self.archive_test('t.zip.foo')
        self.archive_test('t.jar.foo')
        self.archive_test('t.Z.foo')
        self.archive_test('t.cab.foo')
        self.archive_test('t.arj.foo')
        self.archive_test('t.cpio.foo')
        self.archive_test('t.rpm.foo')
        self.archive_test('t.deb.foo')

    @needs_program('file')
    @needs_codec('7z', 'rar')
    def test_p7zip_rar (self):
        # only succeeds with the rar module for 7z installed
        self.program = '7z'
        self.archive_list('t.rar.foo')
        self.archive_extract('t.rar.foo')
        self.archive_test('t.rar.foo')

    @needs_program('file')
    @needs_program('unrar')
    def test_unrar (self):
        self.program = 'unrar'
        self.archive_list('t.rar.foo')
        self.archive_extract('t.rar.foo')

    @needs_program('file')
    @needs_program('rar')
    def test_rar (self):
        self.program = 'rar'
        self.archive_commands('t.rar.foo', format="rar")

    @needs_program('file')
    @needs_program('cabextract')
    def test_cabextract (self):
        self.program = 'cabextract'
        self.archive_list('t.cab.foo')
        self.archive_extract('t.cab.foo')

    @needs_program('file')
    @needs_program('arj')
    def test_arj (self):
        self.program = 'arj'
        self.archive_commands('t.arj.foo', format="arj")

    @needs_program('file')
    @needs_program('cpio')
    def test_cpio (self):
        self.program = 'cpio'
        self.archive_list('t.cpio.foo')
        self.archive_extract('t.cpio.foo')
        self.archive_create('t.cpio.foo', format="cpio")

    @needs_program('file')
    @needs_program('rpm')
    def test_rpm (self):
        self.program = 'rpm'
        self.archive_list('t.rpm.foo')
        # The rpm test fails on non-rpm system with missing dependencies.
        # I am too lazy to build a tiny rpm with one file
        # and no dependency.
        #self.archive_test('t.rpm.foo')

    @needs_program('file')
    @needs_program('rpm2cpio')
    @needs_program('cpio')
    def test_rpm_extract (self):
        self.program = 'rpm2cpio'
        self.archive_extract('t.rpm.foo')

    @needs_program('file')
    @needs_program('dpkg-deb')
    def test_dpkg (self):
        self.program = 'dpkg'
        self.archive_list('t.deb.foo')
        self.archive_extract('t.deb.foo')
        self.archive_test('t.deb.foo')

    @needs_program('file')
    @needs_program('lzop')
    def test_lzop (self):
        self.program = 'lzop'
        self.archive_commands('t.lzo.foo', format="lzop", singlefile=True)

    # file(1) does not recognize .lzma files
    #@needs_program('file')
    #@needs_program('lzma')
    #def test_lzma (self):
    #    self.program = 'lzma'
    #    self.archive_test('t.lzma.foo')
    #    self.archive_extract('t.lzma.foo')
    #    self.archive_create('t.lzma.foo', format="lzma", singlefile=True)

    @needs_program('file')
    @needs_program('lzip')
    def test_lzip (self):
        self.program = 'lzip'
        self.archive_test('t.txt.lz.foo')
        self.archive_extract('t.txt.lz.foo')
        self.archive_create('t.txt.lz.foo', format="lzip", singlefile=True)

    @needs_program('file')
    @needs_program('xz')
    def test_xz (self):
        self.program = 'xz'
        self.archive_test('t.xz.foo')
        self.archive_extract('t.xz.foo')
        self.archive_create('t.xz.foo', format="xz", singlefile=True)