You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
3.3 KiB
87 lines
3.3 KiB
import unittest
|
|
import mock
|
|
|
|
from os.path import dirname, realpath
|
|
from sys import argv, path
|
|
path.append(dirname(dirname(realpath(__file__))) + '/lib')
|
|
|
|
from Communication.EndPoint import CommunicationEndPoint
|
|
|
|
class TestCommunicationEndPoint(unittest.TestCase):
|
|
def setUp(self):
|
|
self._transport = mock.Mock()
|
|
self._protocol = mock.Mock()
|
|
self._bufsize = 11773
|
|
self._endpoint = CommunicationEndPoint(
|
|
self._transport, self._protocol, self._bufsize)
|
|
|
|
def testSetClose(self):
|
|
self.assertFalse(self._endpoint.shouldClose())
|
|
self._endpoint.setClose()
|
|
self.assertTrue(self._endpoint.shouldClose())
|
|
|
|
def testHasProtocol(self):
|
|
self.assertTrue(self._endpoint.hasProtocol(mock.Mock))
|
|
|
|
def testHasPendingData(self):
|
|
self.assertFalse(self._endpoint.hasPendingData())
|
|
|
|
def testGetTransport(self):
|
|
self.assertEqual(self._endpoint.getTransport(), self._transport)
|
|
|
|
def testGetProtocol(self):
|
|
self.assertEqual(self._endpoint.getProtocol(), self._protocol)
|
|
|
|
def testGetHandle(self):
|
|
self._transport.getHandle.return_value = 10
|
|
self.assertEqual(self._endpoint.getHandle(), 10)
|
|
self._transport.getHandle.assert_call_once()
|
|
|
|
def testBufferRead(self):
|
|
self._transport.recv.return_value = False
|
|
self.assertFalse(self._endpoint.bufferRead())
|
|
self._transport.recv.assert_call_once_with(11773)
|
|
|
|
self._transport.reset_mock()
|
|
self._endpoint.appendReadData = mock.Mock()
|
|
|
|
self._transport.recv.side_effect = iter(['111', '2222', '33333', False])
|
|
self.assertTrue(self._endpoint.bufferRead())
|
|
self._transport.recv.assert_call_with(11773)
|
|
self.assertEqual(self._transport.recv.call_count, 4)
|
|
self.assertEqual(self._endpoint.appendReadData.call_count, 3)
|
|
|
|
def testWriteBuffered(self):
|
|
self._endpoint.nextWriteData = mock.Mock()
|
|
self._endpoint.nextWriteData.return_value = ('', 1212)
|
|
self.assertFalse(self._endpoint.writeBuffered())
|
|
self._endpoint.nextWriteData.assert_called_once_with()
|
|
|
|
self._endpoint.nextWriteData.reset_mock()
|
|
self._endpoint.nextWriteData.return_value = ('12345', 1212)
|
|
self._endpoint.appendWriteData = mock.Mock()
|
|
self._transport.send.return_value = 0
|
|
self.assertFalse(self._endpoint.writeBuffered())
|
|
self._endpoint.nextWriteData.assert_called_once_with()
|
|
self._transport.send.assert_called_once_with('12345', 1212)
|
|
|
|
self._transport.reset_mock()
|
|
self._endpoint.nextWriteData.reset_mock()
|
|
self._endpoint.nextWriteData.side_effect = iter(
|
|
[('111222', 1212), ('333', 1313), ('', 1212)])
|
|
self._endpoint.appendWriteData = mock.Mock()
|
|
self._transport.send.return_value = 3
|
|
self.assertTrue(self._endpoint.writeBuffered())
|
|
self._endpoint.nextWriteData.assert_called_with()
|
|
self._transport.send.assert_any_call('111222', 1212)
|
|
self._transport.send.assert_any_call('222', 1212)
|
|
self._transport.send.assert_called_with('333', 1313)
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromTestCase(TestCommunicationEndPoint)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.TextTestRunner(verbosity=2).run(suite())
|
|
|
|
# vim: set ft=python et ts=8 sw=4 sts=4:
|