Rev 2: Support bzr+ssh operations. in http://bzr.arbash-meinel.com/plugins/lock_test

[John Arbash Meinel]

At http://bzr.arbash-meinel.com/plugins/lock_test

------------------------------------------------------------
revno: 2
revision-id: john at arbash-meinel.com-20100615201310-ut69xt4h02tnxv5x
parent: john at arbash-meinel.com-20100615200134-axeyrh0o6dnm0asg
committer: John Arbash Meinel <john at arbash-meinel.com>
branch nick: lock_test
timestamp: Tue 2010-06-15 15:13:10 -0500
message:
  Support bzr+ssh operations.
-------------- next part --------------
=== modified file 'lock_test.py'
--- a/lock_test.py	2010-06-15 20:01:34 +0000
+++ b/lock_test.py	2010-06-15 20:13:10 +0000
@@ -48,7 +48,13 @@
     repo.texts.insert_record_stream([root_content])
     repo.revisions.insert_record_stream([content])
     save_times = []
-    orig = repo._pack_collection._save_pack_names
+    pc = getattr(repo, '_pack_collection', None)
+    if pc is None:
+        # commit_write_group() is done via _ensure_real anyway, so just use it
+        # directly.
+        repo._ensure_real()
+        pc = repo._real_repository._pack_collection
+    orig = pc._save_pack_names
     def _do_save(*args, **kwargs):
         t1 = time.time()
         try:
@@ -57,10 +63,10 @@
             t2 = time.time()
             save_times.append(t2 - t1)
     try:
-        repo._pack_collection._save_pack_names = _do_save
+        pc._save_pack_names = _do_save
         t = time.time()
         repo.commit_write_group()
         tdelta = time.time() - t
     finally:
-        repo._pack_collection._save_pack_names = orig
+        pc._save_pack_names = orig
     return tdelta, save_times