[PATCH p2v-transfer.git] Fix unit tests to account for last two patches

[Ben Lipton]

Signed-off-by: Ben Lipton <benlip...@google.com>
---
 p2v-transfer/test/p2v_transfer_test.py |    7 ++++---
 1 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/p2v-transfer/test/p2v_transfer_test.py 
b/p2v-transfer/test/p2v_transfer_test.py
index 13efd74..369afca 100755
--- a/p2v-transfer/test/p2v_transfer_test.py
+++ b/p2v-transfer/test/p2v_transfer_test.py
@@ -354,7 +354,7 @@ EOF
     self.module.RunFixScripts(self.client)
     self.module.ShutDownTarget(self.client)
     self.module.UnmountSourceFilesystems(self.fs_devs)
-    self.module.CleanUpTarget(self.client)
+    # Don't call CleanUpTarget, the target is shut down
 
     self.mox.ReplayAll()
     self.module.main(self.test_argv)
@@ -447,10 +447,11 @@ EOF
                              use_mock_anything=True)
     self.mox.StubOutWithMock(self.module.os.path, 'expanduser')
 
-    known_hosts = "/home/%s/.ssh/known_hosts" % self.user
+    known_hosts = "/root/.ssh/known_hosts"
 
     self.module.paramiko.SSHClient().AndReturn(self.client)
-    self.module.os.path.expanduser("~/.ssh/known_hosts").AndReturn(known_hosts)
+    call = self.module.os.path.expanduser("~root/.ssh/known_hosts")
+    call.AndReturn(known_hosts)
     self.client.set_missing_host_key_policy(mox.IsA(self.module.AskAddPolicy))
     self.client.load_host_keys(known_hosts)
     self.client.connect(self.host, username=self.user, pkey=self.pkey,
-- 
1.7.3.1