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("-"))
