From 6b7b416df9adc5c54230776e6abde769117b14a8 Mon Sep 17 00:00:00 2001
From: Vincent Pelletier <vincent@nexedi.com>
Date: Fri, 29 Oct 2010 18:33:52 +0000
Subject: [PATCH] Fix tests for r2364.

git-svn-id: https://svn.erp5.org/repos/neo/trunk@2368 71dcc9de-d417-0410-9af5-da40c76e7ee4
---
 neo/tests/storage/testReplicationHandler.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/neo/tests/storage/testReplicationHandler.py b/neo/tests/storage/testReplicationHandler.py
index d7521327..0d6aa309 100644
--- a/neo/tests/storage/testReplicationHandler.py
+++ b/neo/tests/storage/testReplicationHandler.py
@@ -86,12 +86,12 @@ class StorageReplicationHandlerTests(NeoTestBase):
             'checkTIDRange': None,
             'getTIDsFrom': None,
             'getObjectHistoryFrom': None,
+            'getCurrentRID': rid,
+            'getCurrentCriticalTID': critical_tid,
         })
-        replicator.current_partition = Mock({
-            'getRID': rid,
-            'getCriticalTID': critical_tid,
-        })
-        replicator.current_connection = conn
+        def isCurrentConnection(other_conn):
+            return other_conn is conn
+        replicator.isCurrentConnection = isCurrentConnection
         real_replicator = replicator
         class FakeApp(object):
             replicator = real_replicator
-- 
2.30.9