[libvirt] [test-API][PATCH 2/2] Modify migration.py to support testing with tcp and tls

[Guannan Ren]

---
 repos/domain/migrate.py |    7 +++++--
 1 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/repos/domain/migrate.py b/repos/domain/migrate.py
index 3f80bd0..6b86188 100644
--- a/repos/domain/migrate.py
+++ b/repos/domain/migrate.py
@@ -1,6 +1,8 @@
 #!/usr/bin/env python
 """this script is for migration testing
    domain:migrate
+       transport
+           tcp|tls|ssh
        target_machine
            10.66.5.5
        username
@@ -105,7 +107,7 @@ def env_clean(src, dst, srcdom, dstdom, target_machine, guestname, logger):
 def check_params(params):
     """check out the arguments requried for migration"""
     logger = params['logger']
-    keys = ['target_machine', 'username', 'password', 'guestname', 'flags']
+    keys = ['transport', 'target_machine', 'username', 'password', 'guestname', 'flags']
     for key in keys:
         if key not in params:
             logger.error("Argument %s is required" % key)
@@ -194,6 +196,7 @@ def migrate(params):
     if params_check_result:
         return 1
 
+    transport = params['transport']
     target_machine = params['target_machine']
     username = params['username']
     password = params['password']
@@ -243,7 +246,7 @@ def migrate(params):
     commands.getstatusoutput("ssh-add")
 
     srcuri = "qemu:///system"
-    dsturi = "qemu+ssh://%s/system" % target_machine
+    dsturi = "qemu+%s://%s/system" % (transport, target_machine)
 
     # Connect to local hypervisor connection URI
     util = utils.Utils()
-- 
1.7.1