Merge pull request AppScale#866 from DrOctogon/fixing-broken-unit-test

Fixing broken unit test for zookeeper transaction rety
nlake44 · Jan 21, 2014 · 60c5d9c · 60c5d9c
2 parents aa36e8d + 3f5e09f
commit 60c5d9c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/AppDB/test/unit/test_zookeeper.py b/AppDB/test/unit/test_zookeeper.py
@@ -293,7 +293,9 @@ def test_acquire_additional_lock(self):
     fake_zookeeper.should_receive('retry').with_args('create', str, makepath=bool, sequence=bool,
       ephemeral=bool, value=str, acl=None).and_return("/some/lock/path")
     fake_zookeeper.should_receive('retry').with_args('create_async', str, value=str,
-      acl=None, ephemeral=bool, makepath=bool)
+      acl=None, ephemeral=bool, makepath=bool, sequence=bool)
+    fake_zookeeper.should_receive('retry').with_args('create_async', str, value=str,
+      acl=str, ephemeral=bool, makepath=bool, sequence=bool)
     lock_list = ['path1', 'path2', 'path3'] 
     lock_list_str = zk.LOCK_LIST_SEPARATOR.join(lock_list)
     fake_zookeeper.should_receive('retry').with_args('get', str) \