def test_PresentYourselfResponse(self):
senderData = {'type': Charstring().assignValueType(CharstringValue("HelloCritty")),
'nick': Charstring().assignValueType(CharstringValue("Sender"))}
receiverData = {'type': Charstring().assignValueType(CharstringValue("HelloCritty")),
'nick': Charstring().assignValueType(CharstringValue("Receiver"))}
presentYourselfResponse = PiewikPresentYourselfResponse()
presentYourselfResponse.assignValueType({
'messageName': Charstring().assignValueType(CharstringValue("PresentYourselfResponse")),
'sender': senderData,
'receiver': receiverData
})
encoder = ProtobufEncoder()
payload = PresentYourselfResponse()
encoder.encodePayload(aPayloadContent=payload,
aPayloadData=presentYourselfResponse)
self.assertEqual(payload.messageName, "PresentYourselfResponse")
self.assertEqual(payload.sender.type, "HelloCritty")
self.assertEqual(payload.sender.nick, "Sender")
self.assertEqual(payload.receiver.type, "HelloCritty")
self.assertEqual(payload.receiver.nick, "Receiver")
def test_LoadGraphAndWorkResponse(self):
senderData = {'type': Charstring().assignValueType(CharstringValue("HelloCritty")),
'nick': Charstring().assignValueType(CharstringValue("Sender"))}
receiverData = {'type': Charstring().assignValueType(CharstringValue("HelloCritty")),
'nick': Charstring().assignValueType(CharstringValue("Receiver"))}
graphsData = [
{'graphName': Charstring().assignValueType(CharstringValue("Graph1"))},
{'graphName': Charstring().assignValueType(CharstringValue("Graph2"))}
]
worksData =[
{'graphName': Charstring().assignValueType(CharstringValue("Graph1")),
'workName': Charstring().assignValueType(CharstringValue("Work1"))},
{'graphName': Charstring().assignValueType(CharstringValue("Graph1")),
'workName': Charstring().assignValueType(CharstringValue("Work2"))}
]
workPredecessorsData = [
{'workName': Charstring().assignValueType(CharstringValue("Work2")),
'predecessorWorkName': Charstring().assignValueType(CharstringValue("Work1"))}
]
loadGraphAndWorkResponse = PiewikLoadGraphAndWorkResponse()
loadGraphAndWorkResponse.assignValueType({
'messageName': Charstring().assignValueType(CharstringValue("LoadGraphAndWorkResponse")),
'sender': senderData,
'receiver': receiverData,
'graphs': graphsData,
'works': worksData,
'workPredecessors': workPredecessorsData
})
encoder = ProtobufEncoder()
payload = LoadGraphAndWorkResponse()
encoder.encodePayload(aPayloadContent=payload,
aPayloadData=loadGraphAndWorkResponse)
self.assertEqual(payload.messageName, "LoadGraphAndWorkResponse")
self.assertEqual(payload.sender.type, "HelloCritty")
self.assertEqual(payload.sender.nick, "Sender")
self.assertEqual(payload.receiver.type, "HelloCritty")
self.assertEqual(payload.receiver.nick, "Receiver")
self.assertEqual(payload.graphs[0].graphName, "Graph1")
self.assertEqual(payload.graphs[1].graphName, "Graph2")
self.assertEqual(payload.works[0].graphName, "Graph1")
self.assertEqual(payload.works[0].workName, "Work1")
self.assertEqual(payload.works[1].graphName, "Graph1")
self.assertEqual(payload.works[1].workName, "Work2")
self.assertEqual(payload.workPredecessors[0].workName, "Work2")
self.assertEqual(payload.workPredecessors[0].predecessorWorkName, "Work1")
def test_LoadGraphAndWorkReqeust(self):
senderData = {'type': Charstring().assignValueType(CharstringValue("HelloCritty")),
'nick': Charstring().assignValueType(CharstringValue("Sender"))}
loadGraphAndWorkRequest = PiewikLoadGraphAndWorkRequest()
loadGraphAndWorkRequest.assignValueType({
'messageName': Charstring().assignValueType(CharstringValue("LoadGraphAndWorkRequest")),
'sender': senderData
})
encoder = ProtobufEncoder()
payload = LoadGraphAndWorkRequest()
encoder.encodePayload(aPayloadContent=payload,
aPayloadData=loadGraphAndWorkRequest)
self.assertEqual(payload.messageName, "LoadGraphAndWorkRequest")
self.assertEqual(payload.sender.type, "HelloCritty")
self.assertEqual(payload.sender.nick, "Sender")
def test_HeartbeatAnnouncement(self):
critterData = {'type': Charstring().assignValueType(CharstringValue("TYPE")),
'nick': Charstring().assignValueType(CharstringValue("NICK"))}
heartbeatAnnouncement = PiewikHeartbeatAnnouncement()
heartbeatAnnouncement.assignValueType({
'messageName': Charstring().assignValueType(CharstringValue("HeartbeatAnnouncement")),
'sender': critterData,
'timestamp': Float().assignValueType(FloatValue(1234.5678))
})
encoder = ProtobufEncoder()
payload = HeartbeatAnnouncement()
encoder.encodePayload(aPayloadContent=payload,
aPayloadData=heartbeatAnnouncement)
self.assertEqual(payload.messageName, "HeartbeatAnnouncement")
self.assertEqual(payload.sender.type, "TYPE")
self.assertEqual(payload.sender.nick, "NICK")
self.assertEqual(payload.timestamp, 1234.5678)
