def test_get_source_file(self): # other files are opened as regular files. fd, filename = tempfile.mkstemp() os.close(fd) stream = _get_output_stream(filename) stream.write("foo") stream.close() f = open(filename, 'r') self.assertEquals("foo", f.read()) f.close()
def test_get_source_gz(self): fd, filename = tempfile.mkstemp(suffix=".gz") os.close(fd) stream = _get_output_stream(filename) stream.write("bla") stream.close() # files ending in .gz are automatically decompressed. f = gzip.GzipFile(filename) self.assertEquals("bla", f.read()) f.close()
def run( self, source=None, destination=None, verbose=False, git_branch="master", checkpoint=10000, marks=None, import_marks=None, export_marks=None, revision=None, plain=True, rewrite_tag_names=False, baseline=False, ): load_fastimport() from bzrlib.branch import Branch from bzrlib.plugins.fastimport import exporter if marks: import_marks = export_marks = marks # Open the source if source is None: source = "." branch = Branch.open_containing(source)[0] outf = exporter._get_output_stream(destination) exporter = exporter.BzrFastExporter( branch, outf=outf, ref="refs/heads/%s" % git_branch, checkpoint=checkpoint, import_marks_file=import_marks, export_marks_file=export_marks, revision=revision, verbose=verbose, plain_format=plain, rewrite_tags=rewrite_tag_names, baseline=baseline, ) return exporter.run()
def test_get_output_stream_stdout(self): # - returns standard out self.assertIsNot(None, _get_output_stream("-"))