import socket
from unittest import main
import util
 
 
class Test(util.TestServer):
    server = 'udp_server.py'
 
    def _run_all_tests(self):
        sock = socket.socket(type=socket.SOCK_DGRAM)
        sock.connect(('127.0.0.1', 9000))
        sock.send('Test udp_server')
        data, address = sock.recvfrom(8192)
        self.assertEqual(data, 'Received 15 bytes')
 
 
if __name__ == '__main__':
    main()