Add flake8 support and fix issues
[netvirt.git] / resources / tools / odltools / odltools / netvirt / tests / test_ovs_flows.py
index 1787539c2384e55ff18306a6ec89390c910f4a00..b5cd38f9711ac6f6b45e09ca8c46e2dbaadb4208 100644 (file)
@@ -1,28 +1,34 @@
+import logging
 import unittest
 
 import os
 
 from odltools import logg
-from odltools.netvirt import request, ovs_flows
+from odltools.netvirt import ovs_flows
+from odltools.netvirt import request
+from odltools.netvirt import tests
 
 
 class TestFlows(unittest.TestCase):
     def setUp(self):
-        logg.Logger()
-        self.filename = "../../tests/resources/flow_dumps.3.txt"
+        logg.Logger(logging.INFO, logging.INFO)
+        self.filename = "{}/flow_dumps.1.txt".format(tests.get_resources_path())
         self.data = request.read_file(self.filename)
         self.flows = ovs_flows.Flows(self.data)
 
     def test_process_data(self):
-        print "pretty_print:\n{}".format(self.flows.pretty_print(self.flows.pdata))
+        # print "pretty_print:\n{}".format(self.flows.pretty_print(self.flows.pdata))
+        self.assertIsNotNone(self.flows.data)
 
     def test_format_data(self):
-        print "pretty_print:\n{}".format(self.flows.pretty_print(self.flows.fdata))
+        # print "pretty_print:\n{}".format(self.flows.pretty_print(self.flows.fdata))
+        self.assertIsNotNone(self.flows.fdata)
 
     def test_write_file(self):
         filename = "/tmp/flow_dumps.3.out.txt"
         self.flows.write_fdata(filename)
         self.assertTrue(os.path.exists(filename))
 
+
 if __name__ == '__main__':
     unittest.main()