summaryrefslogtreecommitdiff
path: root/tests/test_v2.py
diff options
context:
space:
mode:
authorBen Howard <ben.howard@ubuntu.com>2016-02-08 16:33:07 -0700
committerusd-importer <ubuntu-server@lists.ubuntu.com>2016-02-09 00:59:05 +0000
commita00729ff7421b3661e8b1a1e0fa46393379f2e96 (patch)
tree4563b927e3a57446a4a928a72a92d72c9ad4f6e6 /tests/test_v2.py
parent53f54030cae2de3d5fa474a61fe51f16c7a07c79 (diff)
downloadvyos-walinuxagent-a00729ff7421b3661e8b1a1e0fa46393379f2e96.tar.gz
vyos-walinuxagent-a00729ff7421b3661e8b1a1e0fa46393379f2e96.zip
Import patches-unapplied version 2.1.3-0ubuntu1 to ubuntu/xenial-proposed
Imported using git-ubuntu import. Changelog parent: 53f54030cae2de3d5fa474a61fe51f16c7a07c79 New changelog entries: * New upstream release (LP: #1543359): - Bug fixes for extension handling - Feature enablement for AzureStack.
Diffstat (limited to 'tests/test_v2.py')
-rw-r--r--tests/test_v2.py120
1 files changed, 0 insertions, 120 deletions
diff --git a/tests/test_v2.py b/tests/test_v2.py
deleted file mode 100644
index c4a0b4d..0000000
--- a/tests/test_v2.py
+++ /dev/null
@@ -1,120 +0,0 @@
-# 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
-from tests.tools import *
-import unittest
-import json
-import azurelinuxagent.protocol.v2 as v2
-
-SAMPLE_IDENTITY=u"""{
- "vmName":"foo",
- "subscriptionId":"bar"
-}"""
-
-SAMPLE_CERTS=u"""{
- "certificates":[{
- "name":"foo",
- "thumbprint":"bar",
- "certificateDataUri":"baz"
- }]
-}"""
-
-SAMPLE_EXT_HANDLER=u"""[{
- "name":"foo",
- "properties":{
- "version":"bar",
- "upgradePolicy": "manual",
- "state": "enabled",
- "extensions":[{
- "name":"baz",
- "sequenceNumber":0,
- "publicSettings":{
- "commandToExecute": "echo 123",
- "uris":[]
- }
- }]
- },
- "versionUris":[{
- "uri":"versionUri.foo"
- }]
-}]"""
-
-SAMPLE_EXT_HANDLER_PKGS=u"""{
- "versions": [{
- "version":"foo",
- "uris":[{
- "uri":"bar"
- },{
- "uri":"baz"
- }]
- }]
-}"""
-
-def mock_get_data(self, url, headers=None):
- data = u"{}"
- if url.count(u"identity") > 0:
- data = SAMPLE_IDENTITY
- elif url.count(u"certificates") > 0:
- data = SAMPLE_CERTS
- elif url.count(u"extensionHandlers") > 0:
- data = SAMPLE_EXT_HANDLER
- elif url.count(u"versionUri") > 0:
- data = SAMPLE_EXT_HANDLER_PKGS
- return json.loads(data)
-
-class TestMetadataProtocol(unittest.TestCase):
- @mock(v2.MetadataProtocol, '_get_data', mock_get_data)
- def test_getters(self):
- protocol = v2.MetadataProtocol()
- vminfo = protocol.get_vminfo()
- self.assertNotEquals(None, vminfo)
- self.assertNotEquals(None, vminfo.vmName)
- self.assertNotEquals(None, vminfo.subscriptionId)
-
- protocol.get_certs()
-
- ext_handers = protocol.get_ext_handlers()
- self.assertNotEquals(None, ext_handers)
- self.assertNotEquals(None, ext_handers.extHandlers)
- self.assertNotEquals(0, len(ext_handers.extHandlers))
-
- ext_hander = ext_handers.extHandlers[0]
- self.assertNotEquals(None, ext_hander)
- self.assertNotEquals(0, len(ext_hander.properties.extensions))
-
- ext = ext_hander.properties.extensions[0]
- self.assertNotEquals(None, ext)
- self.assertNotEquals(None, ext.publicSettings)
- self.assertEquals("echo 123", ext.publicSettings.get('commandToExecute'))
-
- packages = protocol.get_ext_handler_pkgs(ext_handers.extHandlers[0])
- self.assertNotEquals(None, packages)
-
- @mock(v2.MetadataProtocol, '_put_data', MockFunc())
- def test_reporters(self):
- protocol = v2.MetadataProtocol()
- protocol.report_provision_status(v2.ProvisionStatus())
- protocol.report_vm_status(v2.VMStatus())
- protocol.report_ext_status("foo", "baz", v2.ExtensionStatus())
- protocol.report_event(v2.TelemetryEventList())
-
-if __name__ == '__main__':
- unittest.main()