Commit c0537bd3 authored by Grégory Wisniewski's avatar Grégory Wisniewski

Test for client/handler.py updated according to commit #416


git-svn-id: https://svn.erp5.org/repos/neo/branches/prototype3@419 71dcc9de-d417-0410-9af5-da40c76e7ee4
parent 677a2d41
...@@ -232,12 +232,14 @@ class ClientEventHandlerTest(unittest.TestCase): ...@@ -232,12 +232,14 @@ class ClientEventHandlerTest(unittest.TestCase):
client_handler = ClientAnswerEventHandler(app, dispatcher) client_handler = ClientAnswerEventHandler(app, dispatcher)
conn = self.getConnection() conn = self.getConnection()
uuid = self.getUUID() uuid = self.getUUID()
app.uuid = 'C' * 16
client_handler.handleAcceptNodeIdentification(conn, None, CLIENT_NODE_TYPE, client_handler.handleAcceptNodeIdentification(conn, None, CLIENT_NODE_TYPE,
uuid, '127.0.0.1', 10010, uuid, '127.0.0.1', 10010,
0, 0) 0, 0, INVALID_UUID)
self.assertEquals(len(conn.mockGetNamedCalls('close')), 1) self.assertEquals(len(conn.mockGetNamedCalls('close')), 1)
self.assertEquals(app.storage_node, None) self.assertEquals(app.storage_node, None)
self.assertEquals(app.pt, None) self.assertEquals(app.pt, None)
self.assertEquals(app.uuid, 'C' * 16)
def test_masterAcceptNodeIdentification(self): def test_masterAcceptNodeIdentification(self):
node = Mock({'setUUID': None}) node = Mock({'setUUID': None})
...@@ -253,9 +255,11 @@ class ClientEventHandlerTest(unittest.TestCase): ...@@ -253,9 +255,11 @@ class ClientEventHandlerTest(unittest.TestCase):
client_handler = ClientAnswerEventHandler(app, dispatcher) client_handler = ClientAnswerEventHandler(app, dispatcher)
conn = self.getConnection() conn = self.getConnection()
uuid = self.getUUID() uuid = self.getUUID()
your_uuid = 'C' * 16
app.uuid = INVALID_UUID
client_handler.handleAcceptNodeIdentification(conn, None, MASTER_NODE_TYPE, client_handler.handleAcceptNodeIdentification(conn, None, MASTER_NODE_TYPE,
uuid, '127.0.0.1', 10010, uuid, '127.0.0.1', 10010,
10, 2) 10, 2, your_uuid)
self.assertEquals(len(conn.mockGetNamedCalls('close')), 0) self.assertEquals(len(conn.mockGetNamedCalls('close')), 0)
self.assertEquals(len(conn.mockGetNamedCalls('setUUID')), 1) self.assertEquals(len(conn.mockGetNamedCalls('setUUID')), 1)
setUUID_call_list = node.mockGetNamedCalls('setUUID') setUUID_call_list = node.mockGetNamedCalls('setUUID')
...@@ -263,6 +267,7 @@ class ClientEventHandlerTest(unittest.TestCase): ...@@ -263,6 +267,7 @@ class ClientEventHandlerTest(unittest.TestCase):
self.assertEquals(setUUID_call_list[0].getParam(0), uuid) self.assertEquals(setUUID_call_list[0].getParam(0), uuid)
self.assertEquals(app.storage_node, None) self.assertEquals(app.storage_node, None)
self.assertTrue(app.pt is not None) self.assertTrue(app.pt is not None)
self.assertEquals(app.uuid, your_uuid)
def test_storageAcceptNodeIdentification(self): def test_storageAcceptNodeIdentification(self):
node = Mock({'setUUID': None}) node = Mock({'setUUID': None})
...@@ -278,9 +283,10 @@ class ClientEventHandlerTest(unittest.TestCase): ...@@ -278,9 +283,10 @@ class ClientEventHandlerTest(unittest.TestCase):
client_handler = ClientAnswerEventHandler(app, dispatcher) client_handler = ClientAnswerEventHandler(app, dispatcher)
conn = self.getConnection() conn = self.getConnection()
uuid = self.getUUID() uuid = self.getUUID()
app.uuid = 'C' * 16
client_handler.handleAcceptNodeIdentification(conn, None, STORAGE_NODE_TYPE, client_handler.handleAcceptNodeIdentification(conn, None, STORAGE_NODE_TYPE,
uuid, '127.0.0.1', 10010, uuid, '127.0.0.1', 10010,
0, 0) 0, 0, INVALID_UUID)
self.assertEquals(len(conn.mockGetNamedCalls('close')), 0) self.assertEquals(len(conn.mockGetNamedCalls('close')), 0)
self.assertEquals(len(conn.mockGetNamedCalls('setUUID')), 1) self.assertEquals(len(conn.mockGetNamedCalls('setUUID')), 1)
setUUID_call_list = node.mockGetNamedCalls('setUUID') setUUID_call_list = node.mockGetNamedCalls('setUUID')
...@@ -288,6 +294,7 @@ class ClientEventHandlerTest(unittest.TestCase): ...@@ -288,6 +294,7 @@ class ClientEventHandlerTest(unittest.TestCase):
self.assertEquals(setUUID_call_list[0].getParam(0), uuid) self.assertEquals(setUUID_call_list[0].getParam(0), uuid)
self.assertTrue(app.storage_node is node) self.assertTrue(app.storage_node is node)
self.assertEquals(app.pt, None) self.assertEquals(app.pt, None)
self.assertEquals(app.uuid, 'C' * 16)
def _testHandleUnexpectedPacketCalledWithMedhod(self, client_handler, method, args=(), kw=()): def _testHandleUnexpectedPacketCalledWithMedhod(self, client_handler, method, args=(), kw=()):
# Monkey-patch handleUnexpectedPacket to check if it is called # Monkey-patch handleUnexpectedPacket to check if it is called
......
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