test/util/server.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 from nepi.util import server
   5 import os
   6 import shutil
   7 import sys
   8 import unittest
   9 import uuid
  10
  11 class ServerTestCase(unittest.TestCase):
  12     def setUp(self):
  13         self._root_dir = os.path.join(os.getenv("HOME"), ".nepi",
  14                 str(uuid.uuid1()))
  15         os.makedirs(self._root_dir)
  16
  17     def test_server(self):
  18         s = server.Server(self._root_dir)
  19         s.run()
  20         c = server.Client(self._root_dir)
  21         c.send_msg("Hola")
  22         reply = c.read_reply()
  23         self.assertTrue(reply == "Reply to: Hola")
  24         c.send_stop()
  25         reply = c.read_reply()
  26         self.assertTrue(reply == "Stopping server")
  27
  28     def test_server_long_message(self):
  29         s = server.Server(self._root_dir)
  30         s.run()
  31         c = server.Client(self._root_dir)
  32         c.send_msg("1111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111")
  33         reply = c.read_reply()
  34         self.assertTrue(reply == "Reply to: 1111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111")
  35         c.send_stop()
  36         reply = c.read_reply()
  37         self.assertTrue(reply == "Stopping server")
  38
  39     def tearDown(self):
  40         shutil.rmtree(self._root_dir)
  41
  42 if __name__ == '__main__':
  43     unittest.main()
  44