resources/tools/odltools/odltools/netvirt/tests/test_request.py

   1 import logging
   2 import unittest
   3
   4 import os
   5
   6 from odltools import logg
   7 from odltools.netvirt import request
   8 from odltools.netvirt import tests
   9
  10
  11 class TestRequest(unittest.TestCase):
  12     def setUp(self):
  13         logg.Logger(logging.DEBUG, logging.DEBUG)
  14         self.filename = "{}/flow_dumps.1.txt".format(tests.get_resources_path())
  15         self.outpath = "/tmp/flow_dumps.1.out.txt"
  16
  17     def test_read_file(self):
  18         data = request.read_file(self.filename)
  19         self.assertEquals(len(data), 76)
  20
  21     def test_write_file(self):
  22         data = request.read_file(self.filename)
  23         self.assertEquals(len(data), 76)
  24         request.write_file(self.outpath, data)
  25         self.assertTrue(os.path.exists(self.outpath))
  26
  27 if __name__ == '__main__':
  28     unittest.main()