6a6dd5fdb1
- renamed all files starting with message_ to a name without message_ - exception: msgdict as dict is a reserved word - moved all tests into the message/ directory - changed all tests to use the new names - changed all modules to use the new names - shortened imports to not use the full path (pyogp.lib.base. but only the short path) - removed makepacketdict.py as it's not needed anymore - moved the data/ directory into message/ as it's local to the msg system and some small cleanups on the way. please run the tests! I haven't adjusted pyogp.interop yet, Enus wanted to look after it.
75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
#standard libraries
|
|
import unittest, doctest
|
|
from uuid import UUID
|
|
|
|
#local libraries
|
|
from pyogp.lib.base.message.data import msg_tmpl
|
|
from pyogp.lib.base.message.types import MsgType
|
|
from pyogp.lib.base.message.message import Message, Block
|
|
from pyogp.lib.base.interfaces import IDeserialization
|
|
from pyogp.lib.base.registration import init
|
|
|
|
class TestMessage(unittest.TestCase):
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def setUp(self):
|
|
init()
|
|
|
|
def test_block(self):
|
|
block = Block('CircuitCode', ID=1234,Code=531)
|
|
|
|
def test_build(self):
|
|
msg = Message('TestPacket',
|
|
Block('CircuitCode', ID=1234,Code=531)
|
|
)
|
|
|
|
assert msg.blocks['CircuitCode'][0].vars['ID'].data == 1234, \
|
|
"Incorrect data in block ID"
|
|
assert msg.blocks['CircuitCode'][0].vars['Code'].data == 531, \
|
|
"Incorrect data in block Code"
|
|
|
|
def test_build_multiple(self):
|
|
msg = Message('TestPacket',
|
|
Block('CircuitCode', ID=1234,Code=789),
|
|
Block('CircuitCode', ID=5678,Code=456),
|
|
Block('Test', ID=9101,Code=123)
|
|
)
|
|
|
|
assert msg.blocks['CircuitCode'][0].vars['ID'].data == 1234, \
|
|
"Incorrect data in block ID"
|
|
assert msg.blocks['CircuitCode'][1].vars['ID'].data == 5678, \
|
|
"Incorrect data in block 2 ID"
|
|
|
|
assert msg.blocks['CircuitCode'][0].vars['Code'].data == 789, \
|
|
"Incorrect data in block Code"
|
|
assert msg.blocks['CircuitCode'][1].vars['Code'].data == 456, \
|
|
"Incorrect data in block 2 Code"
|
|
|
|
assert msg.blocks['Test'][0].vars['ID'].data == 9101, \
|
|
"Incorrect data in block Test ID"
|
|
assert msg.blocks['Test'][0].vars['Code'].data == 123, \
|
|
"Incorrect data in block Test ID"
|
|
|
|
def test_build_chat(self):
|
|
import uuid
|
|
msg = Message('ChatFromViewer',
|
|
Block('AgentData', AgentID=uuid.UUID('550e8400-e29b-41d4-a716-446655440000'),
|
|
SessionID=uuid.UUID('550e8400-e29b-41d4-a716-446655440000')),
|
|
Block('ChatData', Message="Chatting\n", Type=1, Channel=0))
|
|
assert msg.blocks['ChatData'][0].vars['Type'].data == 1, "Bad type sent"
|
|
assert msg.blocks['ChatData'][0].vars['Channel'].data == 0, "Bad Channel sent"
|
|
|
|
from pyogp.lib.base.interfaces import ISerialization
|
|
from pyogp.lib.base.message.interfaces import IPacket
|
|
packet = IPacket(msg)
|
|
serial = ISerialization(packet)
|
|
msg = serial.serialize()
|
|
|
|
def test_suite():
|
|
from unittest import TestSuite, makeSuite
|
|
suite = TestSuite()
|
|
suite.addTest(makeSuite(TestMessage))
|
|
return suite
|