populate /var/lib/waagent with ovf-env.xml

this will copy the ovf-env.xml file that was found to
the configured directory (default /var/lib/waagent)

1 files changed, 8 insertions, 0 deletions

diff --git a/tests/unittests/test_datasource/test_azure.py b/tests/unittests/test_datasource/test_azure.py
index a2347f1b..68f4bcca 100644
--- a/tests/unittests/test_datasource/test_azure.py
+++ b/tests/unittests/test_datasource/test_azure.py
@@ -83,6 +83,12 @@ class TestAzureDataSource(MockerTestCase):
         def invoker(cmd):
             data['agent_invoked'] = cmd
 
+        def file_writer(datadir, files):
+            data['files'] = {}
+            data['datadir'] = datadir
+            for (fname, content) in files.items():
+                data['files'][fname] = content
+
         if data.get('ovfcontent') is not None:
             populate_dir(os.path.join(self.paths.seed_dir, "azure"),
                          {'ovf-env.xml': data['ovfcontent']})
@@ -93,6 +99,7 @@ class TestAzureDataSource(MockerTestCase):
             self.apply_patches([(mod, 'list_possible_azure_ds_devs', dsdevs)])
 
         self.apply_patches([(mod, 'invoke_agent', invoker)])
+        self.apply_patches([(mod, 'write_files', file_writer)])
 
         dsrc = mod.DataSourceAzureNet(
             data.get('sys_cfg', {}), distro=None, paths=self.paths)
@@ -109,6 +116,7 @@ class TestAzureDataSource(MockerTestCase):
         self.assertTrue(ret)
         self.assertEqual(dsrc.userdata_raw, "")
         self.assertEqual(dsrc.metadata['local-hostname'], odata['HostName'])
+        self.assertTrue('ovf-env.xml' in data['files'])
 
     def test_user_cfg_set_agent_command(self):
         cfg = {'agent_command': "my_command"}