# -*- coding: utf-8 -*- # # 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 2 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, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA """Black-box tests for bzr export. """ import os import stat import sys import tarfile import zipfile from bzrlib.export import ( zip_exporter, ) from bzrlib.tests import TestSkipped from bzrlib.tests.blackbox import ExternalBase class TestExport(ExternalBase): def test_tar_export(self): tree = self.make_branch_and_tree('tar') self.build_tree(['tar/a']) tree.add('a') os.chdir('tar') self.run_bzr('ignore', 'something') tree.commit('1') self.failUnless(tree.has_filename('.bzrignore')) self.run_bzr('export', 'test.tar.gz') ball = tarfile.open('test.tar.gz') # Make sure the tarball contains 'a', but does not contain # '.bzrignore'. self.assertEqual(['test/a'], sorted(ball.getnames())) def test_tar_export_unicode(self): tree = self.make_branch_and_tree('tar') fname = u'\xe5.txt' try: self.build_tree(['tar/' + fname]) except UnicodeError: raise TestSkipped('Unable to represent path %r' % (fname,)) tree.add([fname]) tree.commit('first') os.chdir('tar') self.run_bzr('export', 'test.tar') ball = tarfile.open('test.tar') # all paths are prefixed with the base name of the tarball self.assertEqual(['test/' + fname.encode('utf8')], sorted(ball.getnames())) def test_zip_export(self): tree = self.make_branch_and_tree('zip') self.build_tree(['zip/a']) tree.add('a') os.chdir('zip') self.run_bzr('ignore', 'something') tree.commit('1') self.failUnless(tree.has_filename('.bzrignore')) self.run_bzr('export', 'test.zip') zfile = zipfile.ZipFile('test.zip') # Make sure the zipfile contains 'a', but does not contain # '.bzrignore'. self.assertEqual(['test/a'], sorted(zfile.namelist())) def test_zip_export_unicode(self): tree = self.make_branch_and_tree('zip') fname = u'\xe5.txt' try: self.build_tree(['zip/' + fname]) except UnicodeError: raise TestSkipped('Unable to represent path %r' % (fname,)) tree.add([fname]) tree.commit('first') os.chdir('zip') self.run_bzr('export', 'test.zip') zfile = zipfile.ZipFile('test.zip') # all paths are prefixed with the base name of the zipfile self.assertEqual(['test/' + fname.encode('utf8')], sorted(zfile.namelist())) def test_zip_export_directories(self): tree = self.make_branch_and_tree('zip') self.build_tree(['zip/a', 'zip/b/', 'zip/b/c', 'zip/d/']) tree.add(['a', 'b', 'b/c', 'd']) tree.commit('init') os.chdir('zip') self.run_bzr('export', 'test.zip') zfile = zipfile.ZipFile('test.zip') names = sorted(zfile.namelist()) # even on win32, zipfile.ZipFile changes all names to use # forward slashes self.assertEqual(['test/a', 'test/b/', 'test/b/c', 'test/d/'], names) file_attr = stat.S_IFREG dir_attr = stat.S_IFDIR | zip_exporter.ZIP_DIRECTORY_BIT a_info = zfile.getinfo(names[0]) self.assertEqual(file_attr, a_info.external_attr) b_info = zfile.getinfo(names[1]) self.assertEqual(dir_attr, b_info.external_attr) c_info = zfile.getinfo(names[2]) self.assertEqual(file_attr, c_info.external_attr) d_info = zfile.getinfo(names[3]) self.assertEqual(dir_attr, d_info.external_attr) def test_dir_export(self): tree = self.make_branch_and_tree('dir') self.build_tree(['dir/a']) tree.add('a') os.chdir('dir') self.run_bzr('ignore', 'something') tree.commit('1') self.failUnless(tree.has_filename('.bzrignore')) self.run_bzr('export', 'direxport') files = sorted(os.listdir('direxport')) # Make sure the exported directory contains 'a', but does not contain # '.bzrignore'. self.assertEqual(['a'], files) def example_branch(self): tree = self.make_branch_and_tree('branch') self.build_tree_contents([('branch/hello', 'foo')]) tree.add('hello') tree.commit('setup') self.build_tree_contents([('branch/goodbye', 'baz')]) tree.add('goodbye') tree.commit('setup') return tree def test_basic_directory_export(self): self.example_branch() os.chdir('branch') # Directory exports self.run_bzr('export', '../latest') self.assertEqual(['goodbye', 'hello'], sorted(os.listdir('../latest'))) self.check_file_contents('../latest/goodbye', 'baz') self.run_bzr('export', '../first', '-r', '1') self.assertEqual(['hello'], sorted(os.listdir('../first'))) self.check_file_contents('../first/hello', 'foo') # Even with .gz and .bz2 it is still a directory self.run_bzr('export', '../first.gz', '-r', '1') self.check_file_contents('../first.gz/hello', 'foo') self.run_bzr('export', '../first.bz2', '-r', '1') self.check_file_contents('../first.bz2/hello', 'foo') def test_basic_tarfile_export(self): self.example_branch() os.chdir('branch') self.run_bzr('export', '../first.tar', '-r', '1') self.failUnless(os.path.isfile('../first.tar')) tf = tarfile.open('../first.tar') try: self.assertEqual(['first/hello'], sorted(tf.getnames())) self.assertEqual('foo', tf.extractfile('first/hello').read()) finally: tf.close() self.run_bzr('export', '../first.tar.gz', '-r', '1') self.failUnless(os.path.isfile('../first.tar.gz')) self.run_bzr('export', '../first.tbz2', '-r', '1') self.failUnless(os.path.isfile('../first.tbz2')) self.run_bzr('export', '../first.tar.bz2', '-r', '1') self.failUnless(os.path.isfile('../first.tar.bz2')) self.run_bzr('export', '../first.tar.tbz2', '-r', '1') self.failUnless(os.path.isfile('../first.tar.tbz2')) tf = tarfile.open('../first.tar.tbz2', 'r:bz2') try: self.assertEqual(['first.tar/hello'], sorted(tf.getnames())) self.assertEqual('foo', tf.extractfile('first.tar/hello').read()) finally: tf.close() self.run_bzr('export', '../first2.tar', '-r', '1', '--root', 'pizza') tf = tarfile.open('../first2.tar') try: self.assertEqual(['pizza/hello'], sorted(tf.getnames())) self.assertEqual('foo', tf.extractfile('pizza/hello').read()) finally: tf.close() def test_basic_zipfile_export(self): self.example_branch() os.chdir('branch') self.run_bzr('export', '../first.zip', '-r', '1') self.failUnlessExists('../first.zip') zf = zipfile.ZipFile('../first.zip') try: self.assertEqual(['first/hello'], sorted(zf.namelist())) self.assertEqual('foo', zf.read('first/hello')) finally: zf.close() self.run_bzr('export', '../first2.zip', '-r', '1', '--root', 'pizza') zf = zipfile.ZipFile('../first2.zip') try: self.assertEqual(['pizza/hello'], sorted(zf.namelist())) self.assertEqual('foo', zf.read('pizza/hello')) finally: zf.close() self.run_bzr('export', '../first-zip', '--format=zip', '-r', '1') zf = zipfile.ZipFile('../first-zip') try: self.assertEqual(['first-zip/hello'], sorted(zf.namelist())) self.assertEqual('foo', zf.read('first-zip/hello')) finally: zf.close()