diff options
author | Ben Howard <ben.howard@ubuntu.com> | 2016-02-08 16:33:07 -0700 |
---|---|---|
committer | usd-importer <ubuntu-server@lists.ubuntu.com> | 2016-02-09 00:59:05 +0000 |
commit | a00729ff7421b3661e8b1a1e0fa46393379f2e96 (patch) | |
tree | 4563b927e3a57446a4a928a72a92d72c9ad4f6e6 /tests/test_v2.py | |
parent | 53f54030cae2de3d5fa474a61fe51f16c7a07c79 (diff) | |
download | vyos-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.py | 120 |
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() |