vstf/vstf/agent/unittest/env/test_vm9pfs.py

   1 """
   2 Created on 2015-9-24
   3
   4 @author: y00228926
   5 """
   6 import unittest
   7
   8 from vstf.agent.unittest.env import model
   9 from vstf.agent.env.basic.vm9pfs import VMConfigBy9pfs
  10 from vstf.agent.env.basic.vm_manager import VMControlOperation
  11
  12
  13 class TestVM9pfs(model.Test):
  14     def setUp(self):
  15         super(TestVM9pfs, self).setUp()
  16         self.vm_config = {
  17             'vm_name': 'vm1',
  18             'vm_cpu': 5,
  19             'image_path': "/mnt/sdb/ubuntu_salt_master.img",
  20             'child_dir': '/mnt/sdb/',
  21             'image_type': 'qcow2',
  22             'ctrl_br': 'br0',
  23             'ctrl_mac': '56:6f:44:a5:3f:a4',
  24             "taps": [
  25                 {
  26                     "tap_name": "tap_in",
  27                     "br_type": "bridge",
  28                     "br_name": "br0",
  29                     "tap_mac": "56:6f:44:a5:3f:a2",
  30                 },
  31                 {
  32                     "tap_name": "tap_out",
  33                     "br_type": "bridge",
  34                     "br_name": "br0",
  35                     "tap_mac": "56:6f:44:a5:3f:a3",
  36                 }
  37             ],
  38             'init_config': {}
  39         }
  40         self.init_config = {
  41             'amqp_server': '192.168.188.10',
  42             'ctrl_ip_setting': '192.168.188.200/23',
  43             'tap_pktloop_config': 'dpdk',
  44             'ctrl_gw': '192.168.188.1'
  45         }
  46         self.mgr = VMControlOperation()
  47         self.mgr.clean_all_vms()
  48         self.mgr.create_vm(self.vm_config)
  49         self.mgr.wait_vm(self.vm_config["vm_name"])
  50         self.vm9pctrl = self.mgr.vm_9p_controllers[self.vm_config["vm_name"]]
  51
  52     def tearDown(self):
  53         self.mgr.clean_all_vms()
  54         super(TestVM9pfs, self).tearDown()
  55
  56     def test_init_config(self):
  57         ret = self.vm9pctrl.config_ip(self.vm_config['ctrl_mac'], self.init_config['ctrl_ip_setting'])
  58         self.assertTrue(ret)
  59         ret = self.vm9pctrl.config_gw(self.init_config['ctrl_gw'])
  60         self.assertTrue(ret)
  61         ret = self.vm9pctrl.set_pktloop_dpdk([self.vm_config['taps'][0]['tap_mac'], self.vm_config['taps'][1]['tap_mac']])
  62         self.assertTrue(ret)
  63         ret = self.vm9pctrl.config_amqp(self.init_config['ctrl_ip_setting'].split('/')[0], self.init_config['amqp_server'])
  64         self.assertTrue(ret)
  65
  66
  67 if __name__ == "__main__":
  68     import logging
  69     logging.basicConfig(level=logging.INFO)
  70     LOG = logging.getLogger(__name__)
  71     unittest.main()