103 lines
4.0 KiB
Python
103 lines
4.0 KiB
Python
|
|
"""
|
|
Contributors can be viewed at:
|
|
http://svn.secondlife.com/svn/linden/projects/2008/pyogp/lib/base/trunk/CONTRIBUTORS.txt
|
|
|
|
$LicenseInfo:firstyear=2008&license=apachev2$
|
|
|
|
Copyright 2009, Linden Research, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0.
|
|
You may obtain a copy of the License at:
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
or in
|
|
http://svn.secondlife.com/svn/linden/projects/2008/pyogp/lib/base/LICENSE.txt
|
|
|
|
$/LicenseInfo$
|
|
"""
|
|
|
|
# standard python libs
|
|
import unittest
|
|
|
|
# pyogp
|
|
from pyogp.lib.base.message_manager import MessageManager
|
|
from pyogp.lib.base.caps import Capability
|
|
from pyogp.lib.base.message.message import Message, Block
|
|
from pyogp.lib.base.tests.mockup_net import MockupUDPServer, MockupUDPClient
|
|
from pyogp.lib.base.message.circuit import Host
|
|
from pyogp.lib.base.message.udpdispatcher import UDPDispatcher
|
|
|
|
# pyogp tests
|
|
import pyogp.lib.base.tests.config
|
|
|
|
from eventlet import api
|
|
|
|
class TestMessageManager(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.host = Host((MockupUDPServer(), 80))
|
|
self.message_manager = MessageManager(self.host,
|
|
capabilities={'EventQueueGet' : Capability('EventQueueGet', 'http://127.0.0.1')})
|
|
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def test_start_stop_monitors(self):
|
|
self.message_manager.start_monitors()
|
|
api.sleep(0)
|
|
self.assertTrue(self.message_manager._is_running)
|
|
self.assertTrue(self.message_manager.event_queue._running)
|
|
self.message_manager.stop_monitors()
|
|
api.sleep(2)
|
|
self.assertFalse(self.message_manager._is_running)
|
|
self.assertTrue(self.message_manager.event_queue.stopped)
|
|
self.assertFalse(self.message_manager.event_queue._running)
|
|
|
|
def test_enqueue_message(self):
|
|
message = Message('TestMessage1',
|
|
Block('TestBlock1',
|
|
Test1 = 0),
|
|
Block('NeighborBlock',
|
|
Test0 = 0,
|
|
Test1 = 1,
|
|
Test2 = 2))
|
|
self.message_manager.enqueue_message(message,
|
|
reliable = True)
|
|
self.assertEqual(self.message_manager.outgoing_queue[0][0].name,
|
|
'TestMessage1')
|
|
self.assertTrue(self.message_manager.outgoing_queue[0][1])
|
|
message2 = Message('TestMessage2',
|
|
Block('TestBlock1',
|
|
Test1 = 0),
|
|
Block('NeighborBlock',
|
|
Test0 = 0,
|
|
Test1 = 1,
|
|
Test2 = 2))
|
|
self.message_manager.enqueue_message(message2,
|
|
reliable = False,
|
|
now = True)
|
|
self.assertEqual(self.message_manager.outgoing_queue[0][0].name,
|
|
'TestMessage2')
|
|
self.assertFalse(self.message_manager.outgoing_queue[0][1])
|
|
|
|
|
|
def test_send_udp_message(self):
|
|
self.message_manager.udp_dispatcher = UDPDispatcher(MockupUDPClient(),
|
|
self.message_manager.settings,
|
|
self.message_manager.message_handler)
|
|
message = Message('PacketAck',
|
|
Block('Packets', ID=0x00000003))
|
|
buf = self.message_manager.send_udp_message(message)
|
|
assert buf == \
|
|
'\x00' + '\x00\x00\x00\x01' + '\x00' + '\xff\xff\xff\xfb' + \
|
|
'\x01' + '\x03\x00\x00\x00', \
|
|
'Received: ' + repr(buf) + ' ' + \
|
|
'Expected: ' + repr('\x00' + '\x00\x00\x00\x01' + '\x00' + \
|
|
'\xff\xff\xff\xfb' + '\x01' + '\x03\x00\x00\x00')
|
|
def test_suite():
|
|
from unittest import TestSuite, makeSuite
|
|
suite = TestSuite()
|
|
suite.addTest(makeSuite(TestMessageManager))
|
|
return suite
|