Commit 40453c6c authored by Grégory Wisniewski's avatar Grégory Wisniewski

Fix getCellListFor(OID|TID), now distinct methods.

git-svn-id: https://svn.erp5.org/repos/neo/trunk@1397 71dcc9de-d417-0410-9af5-da40c76e7ee4
parent aece80b3
...@@ -183,17 +183,17 @@ class ClientApplicationTests(NeoTestBase): ...@@ -183,17 +183,17 @@ class ClientApplicationTests(NeoTestBase):
tid = self.makeTID() tid = self.makeTID()
# cache cleared -> result from ZODB # cache cleared -> result from ZODB
self.assertTrue(oid not in mq) self.assertTrue(oid not in mq)
app.pt = Mock({ 'getCellListForID': (), }) app.pt = Mock({ 'getCellListForOID': (), })
app.local_var.history = (oid, [(tid, 0)]) app.local_var.history = (oid, [(tid, 0)])
self.assertEquals(app.getSerial(oid), tid) self.assertEquals(app.getSerial(oid), tid)
self.assertEquals(len(app.pt.mockGetNamedCalls('getCellListForID')), 1) self.assertEquals(len(app.pt.mockGetNamedCalls('getCellListForOID')), 1)
# fill the cache -> hit # fill the cache -> hit
mq.store(oid, (tid, '')) mq.store(oid, (tid, ''))
self.assertTrue(oid in mq) self.assertTrue(oid in mq)
app.pt = Mock({ 'getCellListForID': (), }) app.pt = Mock({ 'getCellListForOID': (), })
app.getSerial(oid) app.getSerial(oid)
self.assertEquals(app.getSerial(oid), tid) self.assertEquals(app.getSerial(oid), tid)
self.assertEquals(len(app.pt.mockGetNamedCalls('getCellListForID')), 0) self.assertEquals(len(app.pt.mockGetNamedCalls('getCellListForOID')), 0)
def test_load(self): def test_load(self):
app = self.getApp() app = self.getApp()
...@@ -266,7 +266,7 @@ class ClientApplicationTests(NeoTestBase): ...@@ -266,7 +266,7 @@ class ClientApplicationTests(NeoTestBase):
'getAddress': ('127.0.0.1', 0), 'getAddress': ('127.0.0.1', 0),
'fakeReceived': packet, 'fakeReceived': packet,
}) })
app.pt = Mock({ 'getCellListForID': (cell, ), }) app.pt = Mock({ 'getCellListForOID': (cell, ), })
app.cp = Mock({ 'getConnForCell' : conn}) app.cp = Mock({ 'getConnForCell' : conn})
app.local_var.asked_object = -1 app.local_var.asked_object = -1
self.assertRaises(NEOStorageNotFoundError, app.loadSerial, oid, tid2) self.assertRaises(NEOStorageNotFoundError, app.loadSerial, oid, tid2)
...@@ -303,7 +303,7 @@ class ClientApplicationTests(NeoTestBase): ...@@ -303,7 +303,7 @@ class ClientApplicationTests(NeoTestBase):
'getAddress': ('127.0.0.1', 0), 'getAddress': ('127.0.0.1', 0),
'fakeReceived': packet, 'fakeReceived': packet,
}) })
app.pt = Mock({ 'getCellListForID': (cell, ), }) app.pt = Mock({ 'getCellListForOID': (cell, ), })
app.cp = Mock({ 'getConnForCell' : conn}) app.cp = Mock({ 'getConnForCell' : conn})
app.local_var.asked_object = -1 app.local_var.asked_object = -1
self.assertRaises(NEOStorageNotFoundError, app.loadBefore, oid, tid2) self.assertRaises(NEOStorageNotFoundError, app.loadBefore, oid, tid2)
...@@ -386,10 +386,10 @@ class ClientApplicationTests(NeoTestBase): ...@@ -386,10 +386,10 @@ class ClientApplicationTests(NeoTestBase):
# check partition_id and an empty cell list -> NEOStorageError # check partition_id and an empty cell list -> NEOStorageError
app.local_var.txn = txn app.local_var.txn = txn
app.local_var.tid = tid app.local_var.tid = tid
app.pt = Mock({ 'getCellListForID': (), }) app.pt = Mock({ 'getCellListForOID': (), })
app.num_partitions = 2 app.num_partitions = 2
self.assertRaises(NEOStorageError, app.store, oid, tid, '', None, txn) self.assertRaises(NEOStorageError, app.store, oid, tid, '', None, txn)
calls = app.pt.mockGetNamedCalls('getCellListForID') calls = app.pt.mockGetNamedCalls('getCellListForOID')
self.assertEquals(len(calls), 1) self.assertEquals(len(calls), 1)
self.assertEquals(calls[0].getParam(0), oid) # oid=11 self.assertEquals(calls[0].getParam(0), oid) # oid=11
...@@ -410,7 +410,7 @@ class ClientApplicationTests(NeoTestBase): ...@@ -410,7 +410,7 @@ class ClientApplicationTests(NeoTestBase):
'getAddress': 'FakeServer', 'getAddress': 'FakeServer',
'getState': 'FakeState', 'getState': 'FakeState',
}) })
app.pt = Mock({ 'getCellListForID': (cell, cell, )}) app.pt = Mock({ 'getCellListForOID': (cell, cell, )})
app.cp = Mock({ 'getConnForCell': ReturnValues(None, conn)}) app.cp = Mock({ 'getConnForCell': ReturnValues(None, conn)})
app.dispatcher = Mock({}) app.dispatcher = Mock({})
app.local_var.object_stored = (oid, tid) app.local_var.object_stored = (oid, tid)
...@@ -440,7 +440,7 @@ class ClientApplicationTests(NeoTestBase): ...@@ -440,7 +440,7 @@ class ClientApplicationTests(NeoTestBase):
'getAddress': 'FakeServer', 'getAddress': 'FakeServer',
'getState': 'FakeState', 'getState': 'FakeState',
}) })
app.pt = Mock({ 'getCellListForID': (cell, cell, ) }) app.pt = Mock({ 'getCellListForOID': (cell, cell, ) })
app.dispatcher = Mock({}) app.dispatcher = Mock({})
app.conflict_serial = None # reset by hand app.conflict_serial = None # reset by hand
app.local_var.object_stored = () app.local_var.object_stored = ()
...@@ -479,7 +479,7 @@ class ClientApplicationTests(NeoTestBase): ...@@ -479,7 +479,7 @@ class ClientApplicationTests(NeoTestBase):
'getAddress': 'FakeServer', 'getAddress': 'FakeServer',
'getState': 'FakeState', 'getState': 'FakeState',
}) })
app.pt = Mock({ 'getCellListForID': (cell, cell, ) }) app.pt = Mock({ 'getCellListForTID': (cell, cell, ) })
app.cp = Mock({ 'getConnForCell': ReturnValues(None, conn), }) app.cp = Mock({ 'getConnForCell': ReturnValues(None, conn), })
app.dispatcher = Mock() app.dispatcher = Mock()
app.tpc_begin(txn, tid) app.tpc_begin(txn, tid)
...@@ -507,7 +507,7 @@ class ClientApplicationTests(NeoTestBase): ...@@ -507,7 +507,7 @@ class ClientApplicationTests(NeoTestBase):
'getAddress': 'FakeServer', 'getAddress': 'FakeServer',
'getState': 'FakeState', 'getState': 'FakeState',
}) })
app.pt = Mock({ 'getCellListForID': (cell, cell, ) }) app.pt = Mock({ 'getCellListForTID': (cell, cell, ) })
app.cp = Mock({ 'getConnForCell': ReturnValues(None, conn), }) app.cp = Mock({ 'getConnForCell': ReturnValues(None, conn), })
app.dispatcher = Mock() app.dispatcher = Mock()
app.tpc_begin(txn, tid) app.tpc_begin(txn, tid)
...@@ -526,7 +526,7 @@ class ClientApplicationTests(NeoTestBase): ...@@ -526,7 +526,7 @@ class ClientApplicationTests(NeoTestBase):
self.assertFalse(app.local_var.txn is txn) self.assertFalse(app.local_var.txn is txn)
conn = Mock() conn = Mock()
cell = Mock() cell = Mock()
app.pt = Mock({'getCellListForID': (cell, cell)}) app.pt = Mock({'getCellListForTID': (cell, cell)})
app.cp = Mock({'getConnForCell': ReturnValues(None, cell)}) app.cp = Mock({'getConnForCell': ReturnValues(None, cell)})
app.tpc_abort(txn) app.tpc_abort(txn)
# no packet sent # no packet sent
...@@ -549,7 +549,10 @@ class ClientApplicationTests(NeoTestBase): ...@@ -549,7 +549,10 @@ class ClientApplicationTests(NeoTestBase):
cell1 = Mock({ 'getNode': 'NODE1', '__hash__': 1 }) cell1 = Mock({ 'getNode': 'NODE1', '__hash__': 1 })
cell2 = Mock({ 'getNode': 'NODE2', '__hash__': 2 }) cell2 = Mock({ 'getNode': 'NODE2', '__hash__': 2 })
conn1, conn2 = Mock({ 'getNextId': 1, }), Mock({ 'getNextId': 2, }) conn1, conn2 = Mock({ 'getNextId': 1, }), Mock({ 'getNextId': 2, })
app.pt = Mock({ 'getCellListForID': ReturnValues((cell1, ), (cell1, ), (cell1, cell2)), }) app.pt = Mock({
'getCellListForOID': ReturnValues((cell1, ), (cell1, )),
'getCellListForTID': (cell1, cell2),
})
app.cp = Mock({ 'getConnForCell': ReturnValues(conn1, conn2), }) app.cp = Mock({ 'getConnForCell': ReturnValues(conn1, conn2), })
# fake data # fake data
app.local_var.data_dict = {oid1: '', oid2: ''} app.local_var.data_dict = {oid1: '', oid2: ''}
...@@ -774,8 +777,8 @@ class ClientApplicationTests(NeoTestBase): ...@@ -774,8 +777,8 @@ class ClientApplicationTests(NeoTestBase):
object_cells = [ Mock({}), ] object_cells = [ Mock({}), ]
history_cells = [ Mock({}), Mock({}) ] history_cells = [ Mock({}), Mock({}) ]
app.pt = Mock({ app.pt = Mock({
'getCellListForID': ReturnValues(object_cells, history_cells, 'getCellListForOID': object_cells,
history_cells), 'getCellListForTID': ReturnValues(history_cells, history_cells),
}) })
app.cp = Mock({ 'getConnForCell': conn}) app.cp = Mock({ 'getConnForCell': conn})
# start test here # start test here
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment