From: Aaron Bentley <abentley@panoramicfeedback.com>
Date: Mon, 16 Jul 2007 16:22:59 +0000 (-0400)
Subject: Ensure that Arch tests have a user id set
X-Git-Tag: 1.0.0~181
X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=e8d004d66e6de5bb29cc874290d2114cb7146d51;p=be.git

Ensure that Arch tests have a user id set
---

diff --git a/libbe/arch.py b/libbe/arch.py
index b7c8a0b..624aea3 100644
--- a/libbe/arch.py
+++ b/libbe/arch.py
@@ -41,6 +41,25 @@ def invoke_client(*args, **kwargs):
         raise Exception("Command failed: %s" % error)
     return output
 
+def get_user_id():
+    try:
+        return invoke_client('my-id')
+    except Exception, e:
+        if 'no arch user id set' in e.args[0]:
+            return None
+        else:
+            raise
+
+
+def set_user_id(value):
+    invoke_client('my-id', value)
+
+
+def ensure_user_id():
+    if get_user_id() is None:
+        set_user_id('nobody <nobody@example.com>')
+ 
+
 def write_tree_settings(contents, path):
     file(os.path.join(path, "{arch}", "=tagging-method"), "wb").write(contents)
 
@@ -49,6 +68,7 @@ def init_tree(path):
 
 def temp_arch_tree(type="easy"):
     import tempfile
+    ensure_user_id()
     path = tempfile.mkdtemp()
     init_tree(path)
     if type=="easy":
diff --git a/libbe/tests.py b/libbe/tests.py
index 2662df1..a7d925d 100644
--- a/libbe/tests.py
+++ b/libbe/tests.py
@@ -36,6 +36,7 @@ class Dir:
         shutil.rmtree(self.name)
 
 def arch_dir():
+    arch.ensure_user_id()
     dir = Dir()
     arch.init_tree(dir.name)
     return dir