vstf/vstf/controller/unittest/test_collect.py

   1 """
   2 Created on 2015-9-28
   3
   4 @author: y00228926
   5 """
   6 import unittest
   7 import json
   8
   9 from vstf.controller.env_build import env_collect
  10 from vstf.controller.unittest import model
  11
  12
  13 class TestCollect(model.Test):
  14
  15     def setUp(self):
  16         super(TestCollect, self).setUp()
  17         self.obj = env_collect.EnvCollectApi(self.conn)
  18
  19     def test_collect_host_info(self):
  20         ret_str = json.dumps(self.obj.collect_host_info(self.tester_host), indent = 4)
  21         for key in ("CPU INFO","MEMORY INFO","HW_INFO","OS INFO"):
  22             self.assertTrue(key in ret_str, "collect_host_info failed, ret_str = %s" % ret_str)
  23
  24     def test_list_nic_devices(self):
  25         ret_str = json.dumps(self.obj.list_nic_devices(self.tester_host), indent = 4)
  26         for key in ("device","mac","bdf","desc"):
  27             self.assertTrue(key in ret_str, "list_nic_devices failed, ret_str = %s" % ret_str)
  28         print ret_str
  29
  30     def test_get_device_detail(self):
  31         identity = "01:00.0"
  32         ret = self.obj.get_device_detail(self.tester_host, "01:00.0")
  33         for key in ("device","mac","bdf","desc"):
  34             self.assertTrue(key in ret)
  35         self.assertTrue(ret['bdf'] == identity)
  36
  37
  38 if __name__ == "__main__":
  39     import logging
  40     logging.basicConfig(level = logging.INFO)
  41     unittest.main()