Commit 7856a878 authored by Aurel's avatar Aurel

fix a bad call and missing method


git-svn-id: https://svn.erp5.org/repos/neo/branches/prototype3@283 71dcc9de-d417-0410-9af5-da40c76e7ee4
parent 18ecff5f
...@@ -200,6 +200,13 @@ server: 127.0.0.1:10023 ...@@ -200,6 +200,13 @@ server: 127.0.0.1:10023
self.assertTrue(isinstance(packet, Packet)) self.assertTrue(isinstance(packet, Packet))
self.assertEquals(packet.getType(), ERROR) self.assertEquals(packet.getType(), ERROR)
def checkCalledAnswerPrimaryMaster(self, conn, packet_number=0):
""" Check Answer primaty master message has been send"""
call = conn.mockGetNamedCalls("addPacket")[packet_number]
packet = call.getParam(0)
self.assertTrue(isinstance(packet, Packet))
self.assertEquals(packet.getType(), ANSWER_PRIMARY_MASTER)
# Tests # Tests
def test_01_connectionStarted(self): def test_01_connectionStarted(self):
uuid = self.identifyToMasterNode(node_type=MASTER_NODE_TYPE, port=self.master_port) uuid = self.identifyToMasterNode(node_type=MASTER_NODE_TYPE, port=self.master_port)
...@@ -566,7 +573,7 @@ server: 127.0.0.1:10023 ...@@ -566,7 +573,7 @@ server: 127.0.0.1:10023
election.handleAskPrimaryMaster(conn, packet) election.handleAskPrimaryMaster(conn, packet)
self.assertEquals(len(conn.mockGetNamedCalls("addPacket")), 1) self.assertEquals(len(conn.mockGetNamedCalls("addPacket")), 1)
self.assertEquals(len(conn.mockGetNamedCalls("abort")), 0) self.assertEquals(len(conn.mockGetNamedCalls("abort")), 0)
conn.checkCalledAnswerPrimaryMaster(conn, 1) self.checkCalledAnswerPrimaryMaster(conn, 1)
def test_12_handleAnnouncePrimaryMaster(self): def test_12_handleAnnouncePrimaryMaster(self):
election = self.election election = self.election
......
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