Merge "ensure failure to inject user files results in startup error"

1 files changed, 21 insertions, 0 deletions

diff --git a/nova/tests/test_virt_disk.py b/nova/tests/test_virt_disk.py
index 902d49704..e6a57e085 100644
--- a/nova/tests/test_virt_disk.py
+++ b/nova/tests/test_virt_disk.py
@@ -14,8 +14,10 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
+import os
 import sys
 
+from nova import exception
 from nova import test
 from nova.tests import fakeguestfs
 from nova.virt.disk import api as diskapi
@@ -29,6 +31,25 @@ class VirtDiskTest(test.TestCase):
         sys.modules['guestfs'] = fakeguestfs
         vfsguestfs.guestfs = fakeguestfs
 
+    def test_inject_data(self):
+
+        self.assertTrue(diskapi.inject_data("/some/file", use_cow=True))
+
+        self.assertTrue(diskapi.inject_data("/some/file",
+                                            mandatory=('files',)))
+
+        self.assertTrue(diskapi.inject_data("/some/file", key="mysshkey",
+                                            mandatory=('key',)))
+
+        os_name = os.name
+        os.name = 'nt'  # Cause password injection to fail
+        self.assertRaises(exception.NovaException,
+                          diskapi.inject_data,
+                          "/some/file", admin_password="p",
+                          mandatory=('admin_password',))
+        self.assertFalse(diskapi.inject_data("/some/file", admin_password="p"))
+        os.name = os_name
+
     def test_inject_data_key(self):
 
         vfs = vfsguestfs.VFSGuestFS("/some/file", "qcow2")