~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to baz2bzr

  • Committer: abentley
  • Date: 2005-04-27 14:58:24 UTC
  • Revision ID: abentley@lappy-20050427145824-e8105f4bd5ec0ce1
Added baz2bzr file(doh!)

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#!/usr/bin/env python
 
2
try:
 
3
    import pybaz
 
4
except ImportError:
 
5
    print "This command requires PyBaz.  Please ensure that it is installed."
 
6
    import sys
 
7
    sys.exit(1)
 
8
import tempfile
 
9
import os
 
10
import os.path
 
11
import shutil
 
12
 
 
13
def test_environ():
 
14
    """
 
15
    >>> q = test_environ()
 
16
    >>> os.path.exists(q)
 
17
    True
 
18
    >>> os.path.exists(os.path.join(q, "home", ".arch-params"))
 
19
    True
 
20
    >>> teardown_environ(q)
 
21
    >>> os.path.exists(q)
 
22
    False
 
23
    """
 
24
    tdir = tempfile.mkdtemp(prefix="baz2bzr-")
 
25
    os.environ["HOME"] = os.path.join(tdir, "home")
 
26
    os.mkdir(os.environ["HOME"])
 
27
    work_dir = os.path.join(tdir, "work_dir")
 
28
    os.mkdir(work_dir)
 
29
    os.chdir(work_dir)
 
30
    lib_dir = os.path.join(tdir, "lib_dir")
 
31
    os.mkdir(lib_dir)
 
32
    pybaz.register_revision_library(lib_dir)
 
33
    arch_dir = os.path.join(tdir, "archive_dir")
 
34
    pybaz.make_archive("test@example.com", arch_dir)
 
35
    return tdir
 
36
 
 
37
def teardown_environ(tdir):
 
38
    os.chdir("/")
 
39
    shutil.rmtree(tdir)
 
40
 
 
41
import doctest
 
42
doctest.testmod()