source: geniuswebcore/test/geniusweb/protocol/session/SessionResultTest.py@ 75

from collections import OrderedDict
import json
from typing import List, Dict
import unittest

from pyson.ObjectMapper import ObjectMapper
from unitpy.GeneralTests import GeneralTests
from uri.uri import URI

from geniusweb.actions.PartyId import PartyId
from geniusweb.inform.Agreements import Agreements
from geniusweb.issuevalue.Bid import Bid
from geniusweb.issuevalue.DiscreteValue import DiscreteValue
from geniusweb.issuevalue.Value import Value
from geniusweb.protocol.session.SessionResult import SessionResult
from geniusweb.references.Parameters import Parameters
from geniusweb.references.PartyRef import PartyRef
from geniusweb.references.PartyWithParameters import PartyWithParameters
from geniusweb.references.PartyWithProfile import PartyWithProfile
from geniusweb.references.ProfileRef import ProfileRef


class SessionResultTest(unittest.TestCase,  GeneralTests[SessionResult]):
        jackson = ObjectMapper()
        error = ValueError("test");

        ISSUE1 = "issue1";
        jsonstring = "{\"participants\":{\"party2\":{\"party\":{\"partyref\":\"party2\",\"parameters\":{}},\"profile\":\"profile2\"},\"party1\":{\"party\":{\"partyref\":\"party1\",\"parameters\":{}},\"profile\":\"profile1\"}},\"agreements\":{\"party2\":{\"issuevalues\":{\"issue1\":\"a\"}},\"party1\":{\"issuevalues\":{\"issue1\":\"a\"}}},\"penalties\":{\"party2\":0.0,\"party1\":0.0},\"error\":null}";
        nopenalties:Dict[PartyId, float] = {}
        penalties:Dict[PartyId, float]  = {}

        PARTY1 = PartyId("party1")
        PARTY2 = PartyId("party2")
        PARTY3 = PartyId("party3")

        penalties[PARTY1]= 0.1
        penalties[PARTY2]= 0.2
        nopenalties[PARTY1]= 0.
        nopenalties[PARTY2]= 0.

        errorstring = "\"error\":{\"java.lang.RuntimeException\":"\
                        + json.dumps(jackson.toJson(error)) + "}"
        print(errorstring)

        party1 = PartyWithParameters(PartyRef(URI("party1")), Parameters())
        party2 = PartyWithParameters(PartyRef(URI("party2")), Parameters())

        partyprofile1 = PartyWithProfile(party1, ProfileRef(URI("profile1")))
        partyprofile2 = PartyWithProfile(party2, ProfileRef(URI("profile2")))

        issuevalues1:Dict[str, Value] = {}
        issuevalues1[ISSUE1]=DiscreteValue("a")
        bid1 = Bid(issuevalues1)
        agreement1 = Agreements().With(Agreements({PARTY1:bid1, PARTY2:bid1} ))

        # different order but that shouldn't matter
        issuevalues2:Dict[str, Value]  =OrderedDict()
        issuevalues2[ISSUE1]= DiscreteValue("b")
        bid2 = Bid(issuevalues2)
        agreement2 = Agreements().With(Agreements({ PARTY1:bid2, PARTY3:bid2} ))

        partiesmap:Dict[PartyId, PartyWithProfile] = OrderedDict()
        partiesmap[PARTY2]= partyprofile2
        partiesmap[PARTY1]= partyprofile1

        partiesmap2:Dict[PartyId, PartyWithProfile] = OrderedDict()
        partiesmap2[PARTY1]= partyprofile2
        partiesmap2[PARTY3]= partyprofile1

        result1 = SessionResult(partiesmap, agreement1, nopenalties, None) 
        result1a = SessionResult(partiesmap, agreement1, nopenalties, None)
        result2 = SessionResult(partiesmap, agreement2, nopenalties, None)
        result3 = SessionResult(partiesmap2, agreement1, nopenalties, None)
        result4 = SessionResult(partiesmap2, agreement1, penalties, None)

        # IGNORE ERROR for now, it fails somewhere deep in maven suddenly.
        # result4 = new SessionResult(Arrays.asList(partyprofile1,
        # partyprofile2), bid1, error);

        def getGeneralTestData(self) ->List[List[SessionResult]]: 
                return [[self.result1, self.result1a],
                                [self.result2], [self.result3],
                                [self.result4]]

        def  getGeneralTestStrings(self) -> List[str]:
                return [
                                "SessionResult.*party2.*profile2.*,.*party1.*profile1.*Agreements.*Bid.*issue1=\"a\".*0\\.0.*0\\.0.*None.*",
                                "SessionResult.*party2.*profile2.*,.*party1.*profile1.*Agreements.*Bid.*issue1=\"b\".*0\\.0.*0\\.0.*None.*",
                                "SessionResult.*party1.*profile2.*,.*party3.*profile1.*Agreements.*Bid.*issue1=\"a\".*0\\.0.*0\\.0.*None.*",
                                "SessionResult.*party1.*profile2.*,.*party3.*profile1.*Agreements.*Bid.*issue1=\"a\".*party1.*0\\.1.*party2.*0\\.2.*None.*"
                                ]

        def serializeTest(self) :
                print(self.jackson.toJson(self.result1))
                self.assertEqual(json.loads(self.jsonstring), self.jackson.toJson(self.result1))

        def deserializeTest(self):
                act = self.jackson.parse(json.loads(self.jsonstring), SessionResult)
                self.assertEqual(self.result1, act)