Hippolyzer/pyogp/lib/base/tests/test_packetdata.py

#standard libraries
import unittest, doctest
import pprint
import binascii

#local libraries
from pyogp.lib.base.data import msg_tmpl
from pyogp.lib.base.message.message_system import MessageSystem
from pyogp.lib.base.message.message_types import MsgType
from pyogp.lib.base.message.circuitdata import Host
from pyogp.lib.base.message.message_template_reader import MessageTemplateReader
from pyogp.lib.base.message.message_template_parser import MessageTemplateParser
from pyogp.lib.base.message.message_template_builder import MessageTemplateBuilder
from pyogp.lib.base.message.message_template_reader import MessageTemplateReader
from pyogp.lib.base.message.message_template_dict import TemplateDictionary

AGENT_DATA_UPDATE="C0 00 00 00 02 00 FF FF 01 83 1C 8A 77 67 E3 7B 42 2E AF B3 85 09 31 97 CA D1 03 4A 42 00 01 06 4B 72 61 66 74 00 01 01 00 1A"
AGENT_DATA_UPDATE =  binascii.unhexlify(''.join(AGENT_DATA_UPDATE.split()))

AGENT_ANIMATION="40 00 00 00 0A 00 05 1C 8A 77 67 E3 7B 42 2E AF B3 85 09 31 97 CA D1 4B 6F FF 1F B5 67 41 FD 85 EF A1 98 3B F2 B5 77 01 EF CF 67 0C 2D 18 81 28 97 3A 03 4E BC 80 6B 67 00 01 00"
AGENT_ANIMATION =  binascii.unhexlify(''.join(AGENT_ANIMATION.split()))



class TestPacketDecode(unittest.TestCase):
    
    def tearDown(self):
        pass

    def setUp(self):
        self.template_file = msg_tmpl
        parser = MessageTemplateParser(self.template_file)
        self.template_list = parser.message_templates        
        self.template_dict = TemplateDictionary(self.template_list)
        self.reader = MessageTemplateReader(self.template_dict)
        
    def test_agent_data_update(self):
        """test if the agent data update packet can be decoded"""
        message = AGENT_DATA_UPDATE
        self.reader.clear_message()
        size = len(message)
        message = MessageSystem(80).zero_code_expand(message, size)
        try:
            assert self.reader.validate_message(message, size), "Validation failed for test_read"
        except:
            assert False, "Validation got error"
        try:
            assert self.reader.read_message(message), "Read failed"
        except:
            assert False, "Read got error"
        
    def test_agent_animation(self):
        """test if the agent data update packet can be decoded"""
        message = AGENT_ANIMATION
        self.reader.clear_message()
        size = len(message)
        assert self.reader.validate_message(message, size), "Validation failed for test_read"
        assert self.reader.read_message(message), "Read failed"    
    
        
def test_suite():
    from unittest import TestSuite, makeSuite
    suite = TestSuite()
    suite.addTest(makeSuite(TestPacketDecode))
    return suite