tests.py

from echo_client import client
import socket
import unittest





class EchoTestCase(unittest.TestCase):

    """tests for the echo server and client"""



    def send_message(self, message):

        """Attempt to send a message using the client



        In case of a socket error, fail and report the problem

        """

        try:

            reply = client(message)

        except socket.error as e:

            if e.errno == 61:

                msg = "Error: {0}, is the server running?"

                self.fail(msg.format(e.strerror))

            else:

                self.fail("Unexpected Error: {0}".format(str(e)))

        return reply


    def test_short_message_echo(self):

        """test that a message short than 16 bytes echoes cleanly"""

        expected = "short message"

        actual = self.send_message(expected)

        self.assertEqual(

            expected,

            actual,

            "expected {0}, got {1}".format(expected, actual)

        )



    def test_long_message_echo(self):

        """test that a message longer than 16 bytes echoes in 16-byte chunks"""

        expected = "Four score and seven years ago our fathers did stuff"

        actual = self.send_message(expected)

        self.assertEqual(

            expected,

            actual,

            "expected {0}, got {1}".format(expected, actual)

        )
if __name__ == '__main__':

    unittest.main()