Upload the contribution of vstf as bottleneck network framework.

[bottlenecks.git] / vstf / vstf / agent / unittest / env / test_vm9pfs.py

diff --git a/vstf/vstf/agent/unittest/env/test_vm9pfs.py b/vstf/vstf/agent/unittest/env/test_vm9pfs.py
new file mode 100755 (executable)
index 0000000..4f52013
--- /dev/null
+++ b/vstf/vstf/agent/unittest/env/test_vm9pfs.py
@@ -0,0 +1,71 @@
+"""
+Created on 2015-9-24
+
+@author: y00228926
+"""
+import unittest
+
+from vstf.agent.unittest.env import model
+from vstf.agent.env.basic.vm9pfs import VMConfigBy9pfs
+from vstf.agent.env.basic.vm_manager import VMControlOperation
+
+
+class TestVM9pfs(model.Test):
+    def setUp(self):
+        super(TestVM9pfs, self).setUp()
+        self.vm_config = {
+            'vm_name': 'vm1',
+            'vm_cpu': 5,
+            'image_path': "/mnt/sdb/ubuntu_salt_master.img",
+            'child_dir': '/mnt/sdb/',
+            'image_type': 'qcow2',
+            'ctrl_br': 'br0',
+            'ctrl_mac': '56:6f:44:a5:3f:a4',
+            "taps": [
+                {
+                    "tap_name": "tap_in",
+                    "br_type": "bridge",
+                    "br_name": "br0",
+                    "tap_mac": "56:6f:44:a5:3f:a2",
+                },
+                {
+                    "tap_name": "tap_out",
+                    "br_type": "bridge",
+                    "br_name": "br0",
+                    "tap_mac": "56:6f:44:a5:3f:a3",
+                }
+            ],
+            'init_config': {}
+        }
+        self.init_config = {
+            'amqp_server': '192.168.188.10',
+            'ctrl_ip_setting': '192.168.188.200/23',
+            'tap_pktloop_config': 'dpdk',
+            'ctrl_gw': '192.168.188.1'
+        }
+        self.mgr = VMControlOperation()
+        self.mgr.clean_all_vms()
+        self.mgr.create_vm(self.vm_config)
+        self.mgr.wait_vm(self.vm_config["vm_name"])
+        self.vm9pctrl = self.mgr.vm_9p_controllers[self.vm_config["vm_name"]]
+
+    def tearDown(self):
+        self.mgr.clean_all_vms()
+        super(TestVM9pfs, self).tearDown()
+
+    def test_init_config(self):
+        ret = self.vm9pctrl.config_ip(self.vm_config['ctrl_mac'], self.init_config['ctrl_ip_setting'])
+        self.assertTrue(ret)
+        ret = self.vm9pctrl.config_gw(self.init_config['ctrl_gw'])
+        self.assertTrue(ret)
+        ret = self.vm9pctrl.set_pktloop_dpdk([self.vm_config['taps'][0]['tap_mac'], self.vm_config['taps'][1]['tap_mac']])
+        self.assertTrue(ret)
+        ret = self.vm9pctrl.config_amqp(self.init_config['ctrl_ip_setting'].split('/')[0], self.init_config['amqp_server'])
+        self.assertTrue(ret)
+
+
+if __name__ == "__main__":
+    import logging
+    logging.basicConfig(level=logging.INFO)
+    LOG = logging.getLogger(__name__)
+    unittest.main()