source: geniuswebcore/test/geniusweb/protocol/session/ConnectionWithPartiesTest.py@ 76

from typing import List
import unittest
from unittest.mock import Mock

from unitpy.GeneralTests import GeneralTests

from geniusweb.actions.PartyId import PartyId
from geniusweb.protocol.partyconnection.ProtocolToPartyConn import ProtocolToPartyConn
from geniusweb.protocol.partyconnection.ProtocolToPartyConnections import ProtocolToPartyConnections


class ConnectionWithPartiesTest (unittest.TestCase, GeneralTests[ProtocolToPartyConnections]):
        party1 = PartyId("party1")
        party2 = PartyId("party2")
        party3 = PartyId("party3")

        con1 = Mock(ProtocolToPartyConn)
        con2 = Mock(ProtocolToPartyConn)
        con3 = Mock(ProtocolToPartyConn)

        conns1 = ProtocolToPartyConnections([con1, con2])
        conns1a = ProtocolToPartyConnections([con1, con2])
        conns2 = ProtocolToPartyConnections([con1, con3])
        conns3 = ProtocolToPartyConnections([con2, con1])
        conns4 = ProtocolToPartyConnections([con2, con1, con3])
        conns5 = ProtocolToPartyConnections([con2, con1, con3, con1])

        con1.__repr__=Mock(return_value="con1") #type:ignore
        con2.__repr__=Mock(return_value="con2") #type:ignore
        con3.__repr__=Mock(return_value="con3")  #type:ignore
        
        con1.getParty=Mock(return_value=party1)
        con2.getParty=Mock(return_value=party2)
        con3.getParty=Mock(return_value=party3)


        def getGeneralTestData(self)->List[List[ProtocolToPartyConnections]] :
                return [[self.conns1, self.conns1a],[self.conns2], [self.conns3],
                                [self.conns4]]

        def getGeneralTestStrings(self) -> List[str] : 
                return ["ConnectionWithParties\\[con1, con2\\]",
                                "ConnectionWithParties\\[con1, con3\\]",
                                "ConnectionWithParties\\[con2, con1\\]",
                                "ConnectionWithParties\\[con2, con1, con3\\]"]

        def testGet(self):
                self.assertEqual(self.con1, self.conns1.getConn(0))
                self.assertEqual(self.con3, self.conns4.getConn(2))

        def testGetParty(self):
                self.assertEquals(self.con1, self.conns1.get(self.party1))
                self.assertEquals(self.con1, self.conns4.get(self.party1))
                self.assertEquals(self.con2, self.conns1.get(self.party2))
                self.assertEquals(self.con2, self.conns3.get(self.party2))
                self.assertEquals(self.con2, self.conns4.get(self.party2))
        
        def testGetUnknownParty(self) :
                self.assertEqual(None, self.conns2.get(self.party2))

        def testSize(self):
                self.assertEqual(2, self.conns1.size())
                self.assertEquals(2, self.conns2.size())
                self.assertEquals(2, self.conns3.size())
                self.assertEquals(3, self.conns4.size())

        def testWithParty(self) :
                conns = self.conns3.With(self.con3)
                self.assertEqual(2, self.conns3.size())
                self.assertEqual(3, conns.size())
                self.assertEqual(self.conns4, conns)

        def testWithExistingParty(self):
                self.conns4.With(self.con1);
                # no throw, default this is ok, it should be the protocol testing this.
        
        def testAllUnique(self):
                self.assertTrue(self.conns4.allunique());
                self.assertFalse(self.conns5.allunique());