tests/test_commands.py

   1 # Copyright (C) 2010 Canonical Ltd
   2 #
   3 # This program is free software; you can redistribute it and/or modify
   4 # it under the terms of the GNU General Public License as published by
   5 # the Free Software Foundation; either version 2 of the License, or
   6 # (at your option) any later version.
   7 #
   8 # This program is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 # GNU General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU General Public License
  14 # along with this program; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  16
  17 """Test the command implementations."""
  18
  19 import os
  20 import tempfile
  21 import gzip
  22
  23 from bzrlib import tests
  24 from bzrlib.tests.blackbox import ExternalBase
  25
  26 from bzrlib.plugins.fastimport.cmds import (
  27     _get_source_stream,
  28     )
  29
  30 from bzrlib.plugins.fastimport.tests import (
  31     FastimportFeature,
  32     )
  33
  34
  35 class TestSourceStream(tests.TestCase):
  36
  37     _test_needs_features = [FastimportFeature]
  38
  39     def test_get_source_stream_stdin(self):
  40         # - returns standard in
  41         self.assertIsNot(None, _get_source_stream("-"))
  42
  43     def test_get_source_gz(self):
  44         # files ending in .gz are automatically decompressed.
  45         fd, filename = tempfile.mkstemp(suffix=".gz")
  46         f = gzip.GzipFile(fileobj=os.fdopen(fd, "w"), mode='w')
  47         f.write("bla")
  48         f.close()
  49         stream = _get_source_stream(filename)
  50         self.assertIsNot("bla", stream.read())
  51
  52     def test_get_source_file(self):
  53         # other files are opened as regular files.
  54         fd, filename = tempfile.mkstemp()
  55         f = os.fdopen(fd, 'w')
  56         f.write("bla")
  57         f.close()
  58         stream = _get_source_stream(filename)
  59         self.assertIsNot("bla", stream.read())
  60
  61
  62 class TestFastExport(ExternalBase):
  63
  64     def test_empty(self):
  65         self.make_branch_and_tree("br")
  66         self.assertEquals("", self.run_bzr("fast-export br")[0])
  67
  68     def test_pointless(self):
  69         tree = self.make_branch_and_tree("br")
  70         tree.commit("pointless")
  71         data = self.run_bzr("fast-export br")[0]
  72         self.assertTrue(data.startswith('commit refs/heads/master\nmark :1\ncommitter'))
  73
  74     def test_file(self):
  75         tree = self.make_branch_and_tree("br")
  76         tree.commit("pointless")
  77         data = self.run_bzr("fast-export br br.fi")[0]
  78         self.assertEquals("", data)
  79         self.failUnlessExists("br.fi")
  80
  81
  82 simple_fast_import_stream = """commit refs/heads/master
  83 mark :1
  84 committer Jelmer Vernooij <jelmer@samba.org> 1299718135 +0100
  85 data 7
  86 initial
  87
  88 """
  89
  90 class TestFastImportInfo(ExternalBase):
  91
  92     def test_simple(self):
  93         self.build_tree_contents([('simple.fi', simple_fast_import_stream)])
  94         output = self.run_bzr("fast-import-info simple.fi")[0]
  95         self.assertEquals(output, """Command counts:
  96 \t0\tblob
  97 \t0\tcheckpoint
  98 \t1\tcommit
  99 \t0\tfeature
 100 \t0\tprogress
 101 \t0\treset
 102 \t0\ttag
 103 File command counts:
 104 \t0\tfilemodify
 105 \t0\tfiledelete
 106 \t0\tfilecopy
 107 \t0\tfilerename
 108 \t0\tfiledeleteall
 109 Parent counts:
 110 \t1\tparents-0
 111 \t0\ttotal revisions merged
 112 Commit analysis:
 113 \tno\texecutables
 114 \tno\tseparate authors found
 115 \tno\tsymlinks
 116 \tno\tblobs referenced by SHA
 117 Head analysis:
 118 \t[':1']\trefs/heads/master
 119 Merges:
 120 """)