def setUp(self):
self.mock_display = MockDisplay(SIZE)
self.server = server.DrawServer(self.mock_display)
class ServerUnitTest(unittest.TestCase):
"""Class for testing individual units of the server."""
def setUp(self):
self.mock_display = MockDisplay(SIZE)
self.server = server.DrawServer(self.mock_display)
def tearDown(self):
del self.server
def testRaisesErrorForSmallPacket(self):
"""Server should raise a parse error for packets that are too short."""
self.assertRaises(
server.ParseError,
self.server._parse_packet,
"foo")
def testDrawPoint(self):
"""Server should send commands to the display on receiving packet."""
packet = "point 0 0 50 50 0 0 255"
expected_args = ((0,0), (0, 0, 255))
self.server._parse_packet(packet)
generated_args = self.mock_display.wasCalled(self.mock_display.drawPoint)
assert generated_args == expected_args
def testDrawLine(self):
"""Server should send commands to the display on receiving packet."""
packet = "line 0 0 50 50 0 0 255"
expected_args = ((0,0), (50, 50), (0, 0, 255))
self.server._parse_packet(packet)
generated_args = self.mock_display.wasCalled(self.mock_display.drawLine)
assert generated_args == expected_args
def testDrawRect(self):
"""Server should send commands to the display on receiving packet."""
packet = "rect 0 0 50 50 0 0 255"
expected_args = ((0,0), (50, 50), (0, 0, 255))
self.server._parse_packet(packet)
generated_args = self.mock_display.wasCalled(self.mock_display.drawRect)
assert generated_args == expected_args
def testDrawEllipse(self):
"""Server should send commands to the display on receiving packet."""
packet = "ellipse 0 0 50 50 0 0 255"
expected_args = ((0,0), (50, 50), (0, 0, 255))
self.server._parse_packet(packet)
generated_args = self.mock_display.wasCalled(
self.mock_display.drawEllipse)
assert generated_args == expected_args
def testDrawRectFilled(self):
"""Server should send commands to the display on receiving packet."""
packet = "rectf 0 0 50 50 0 0 255"
expected_args = ((0,0), (50, 50), (0, 0, 255))
self.server._parse_packet(packet)
generated_args = self.mock_display.wasCalled(self.mock_display.fillRect)
assert generated_args == expected_args
def testDrawEllipseFilled(self):
"""Server should send commands to the display on receiving packet."""
packet = "ellipsef 0 0 50 50 0 0 255"
expected_args = ((0,0), (50, 50), (0, 0, 255))
self.server._parse_packet(packet)
generated_args = self.mock_display.wasCalled(
self.mock_display.fillEllipse)
assert generated_args == expected_args
