1 files changed, 81 insertions, 0 deletions
diff --git a/tests/test_ovfxml.py b/tests/test_ovfxml.py
new file mode 100644
index 0000000..7b6990b
--- /dev/null
+++ b/tests/test_ovfxml.py
@@ -0,0 +1,81 @@
+# Copyright 2014 Microsoft Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# Requires Python 2.4+ and Openssl 1.0+
+#
+# Implements parts of RFC 2131, 1541, 1497 and
+# http://msdn.microsoft.com/en-us/library/cc227282%28PROT.10%29.aspx
+# http://msdn.microsoft.com/en-us/library/cc227259%28PROT.13%29.aspx
+
+import tests.env
+import tests.tools as tools
+import uuid
+import unittest
+import os
+import json
+from azurelinuxagent.future import text
+import azurelinuxagent.protocol.ovfenv as ovfenv
+
+ExtensionsConfigSample="""\
+<?xml version="1.0" encoding="utf-8"?>
+ <Environment xmlns="http://schemas.dmtf.org/ovf/environment/1" xmlns:oe="http://schemas.dmtf.org/ovf/environment/1" xmlns:wa="http://schemas.microsoft.com/windowsazure" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
+    <wa:ProvisioningSection>
+      <wa:Version>1.0</wa:Version>
+      <LinuxProvisioningConfigurationSet xmlns="http://schemas.microsoft.com/windowsazure" xmlns:i="http://www.w3.org/2001/XMLSchema-instance">
+        <ConfigurationSetType>LinuxProvisioningConfiguration</ConfigurationSetType>
+        <HostName>HostName</HostName>
+        <UserName>UserName</UserName>
+        <UserPassword>UserPassword</UserPassword>
+        <DisableSshPasswordAuthentication>false</DisableSshPasswordAuthentication>
+        <SSH>
+          <PublicKeys>
+            <PublicKey>
+              <Fingerprint>EB0C0AB4B2D5FC35F2F0658D19F44C8283E2DD62</Fingerprint>
+              <Path>$HOME/UserName/.ssh/authorized_keys</Path>
+              <Value>ssh-rsa AAAANOTAREALKEY== foo@bar.local</Value>
+            </PublicKey>
+          </PublicKeys>
+          <KeyPairs>
+            <KeyPair>
+              <Fingerprint>EB0C0AB4B2D5FC35F2F0658D19F44C8283E2DD62</Fingerprint>
+              <Path>$HOME/UserName/.ssh/id_rsa</Path>
+            </KeyPair>
+          </KeyPairs>
+        </SSH>
+        <CustomData>CustomData</CustomData>
+      </LinuxProvisioningConfigurationSet>
+    </wa:ProvisioningSection>
+ </Environment>
+"""
+
+class TestOvf(unittest.TestCase):
+    def test_ovf(self):
+        config = ovfenv.OvfEnv(ExtensionsConfigSample)
+        self.assertEquals("HostName", config.hostname)
+        self.assertEquals("UserName", config.username)
+        self.assertEquals("UserPassword", config.user_password)
+        self.assertEquals(False, config.disable_ssh_password_auth)
+        self.assertEquals("CustomData", config.customdata)
+        self.assertNotEquals(None, config.ssh_pubkeys)
+        self.assertEquals(1, len(config.ssh_pubkeys))
+        pubkey = config.ssh_pubkeys[0]
+        path, fingerprint, value = pubkey
+        self.assertEquals(path, "$HOME/UserName/.ssh/authorized_keys")
+        self.assertEquals(fingerprint, "EB0C0AB4B2D5FC35F2F0658D19F44C8283E2DD62"),
+        self.assertEquals(value, "ssh-rsa AAAANOTAREALKEY== foo@bar.local")
+        self.assertNotEquals(None, config.ssh_keypairs)
+        self.assertEquals(1, len(config.ssh_keypairs))
+        
+if __name__ == '__main__':
+    unittest.main()