diff options
author | Ćukasz 'sil2100' Zemczak <lukasz.zemczak@ubuntu.com> | 2018-02-15 14:28:59 +0100 |
---|---|---|
committer | usd-importer <ubuntu-server@lists.ubuntu.com> | 2018-02-15 19:08:29 +0000 |
commit | 3afbcfff2f6e3faf50c0dffee55e7c48b50755b5 (patch) | |
tree | 44d4a5d80868fe6a8defe57f6cc9c21feae59a03 /tests/pa/test_provision.py | |
parent | 6c9cd7e1ac55aae259d8e2f06569375e27a12f20 (diff) | |
parent | 7e7d885433a2bac56ce2361126bf7ec3d565fd66 (diff) | |
download | vyos-walinuxagent-3afbcfff2f6e3faf50c0dffee55e7c48b50755b5.tar.gz vyos-walinuxagent-3afbcfff2f6e3faf50c0dffee55e7c48b50755b5.zip |
Import patches-applied version 2.2.21+really2.2.20-0ubuntu1~16.04.1 to applied/ubuntu/xenial-proposed
Imported using git-ubuntu import.
Changelog parent: 6c9cd7e1ac55aae259d8e2f06569375e27a12f20
Unapplied parent: 7e7d885433a2bac56ce2361126bf7ec3d565fd66
New changelog entries:
* Backport bionic version to xenial.
* Revert to an older upstream release: 2.2.20 (LP: #1749589).
- Rename upstream tarball to 2.2.21+really2.2.20 to end up with a
temporarily higher version number than what's in the archive.
* debian/patches/disable_import_test.patch: refreshed patch.
Diffstat (limited to 'tests/pa/test_provision.py')
-rw-r--r-- | tests/pa/test_provision.py | 85 |
1 files changed, 1 insertions, 84 deletions
diff --git a/tests/pa/test_provision.py b/tests/pa/test_provision.py index 2c2d2c9..7045fcc 100644 --- a/tests/pa/test_provision.py +++ b/tests/pa/test_provision.py @@ -15,13 +15,9 @@ # Requires Python 2.4+ and Openssl 1.0+ # -import json -import socket - import azurelinuxagent.common.utils.fileutil as fileutil -from azurelinuxagent.common.event import WALAEventOperation -from azurelinuxagent.common.exception import ProvisionError +from azurelinuxagent.common.exception import ProtocolError from azurelinuxagent.common.osutil.default import DefaultOSUtil from azurelinuxagent.common.protocol import OVF_FILE_NAME from azurelinuxagent.pa.provision import get_provision_handler @@ -119,85 +115,6 @@ class TestProvision(AgentTestCase): ph.osutil.is_current_instance_id.assert_called_once() deprovision_handler.run_changed_unique_id.assert_called_once() - @distros() - @patch('azurelinuxagent.common.osutil.default.DefaultOSUtil.get_instance_id', - return_value='B9F3C233-9913-9F42-8EB3-BA656DF32502') - def test_provision_telemetry_success(self, mock_util, distro_name, distro_version, - distro_full_name): - """ - Assert that the agent issues two telemetry messages as part of a - successful provisioning. - - 1. Provision - 2. GuestState - """ - ph = get_provision_handler(distro_name, distro_version, - distro_full_name) - ph.report_event = MagicMock() - ph.reg_ssh_host_key = MagicMock(return_value='--thumprint--') - - mock_osutil = MagicMock() - mock_osutil.decode_customdata = Mock(return_value="") - - ph.osutil = mock_osutil - ph.protocol_util.osutil = mock_osutil - ph.protocol_util.get_protocol_by_file = MagicMock() - ph.protocol_util.get_protocol = MagicMock() - - conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir) - ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME) - ovfenv_data = load_data("ovf-env.xml") - fileutil.write_file(ovfenv_file, ovfenv_data) - - ph.run() - - call1 = call("Provisioning succeeded", duration=ANY, is_success=True) - call2 = call(ANY, is_success=True, operation=WALAEventOperation.GuestState) - ph.report_event.assert_has_calls([call1, call2]) - - args, kwargs = ph.report_event.call_args_list[1] - guest_state_json = json.loads(args[0]) - self.assertTrue(1 <= guest_state_json['cpu']) - self.assertTrue(1 <= guest_state_json['mem']) - self.assertEqual(socket.gethostname(), guest_state_json['hostname']) - - @distros() - @patch( - 'azurelinuxagent.common.osutil.default.DefaultOSUtil.get_instance_id', - return_value='B9F3C233-9913-9F42-8EB3-BA656DF32502') - def test_provision_telemetry_fail(self, mock_util, distro_name, - distro_version, - distro_full_name): - """ - Assert that the agent issues one telemetry message as part of a - failed provisioning. - - 1. Provision - """ - ph = get_provision_handler(distro_name, distro_version, - distro_full_name) - ph.report_event = MagicMock() - ph.reg_ssh_host_key = MagicMock(side_effect=ProvisionError( - "--unit-test--")) - - mock_osutil = MagicMock() - mock_osutil.decode_customdata = Mock(return_value="") - - ph.osutil = mock_osutil - ph.protocol_util.osutil = mock_osutil - ph.protocol_util.get_protocol_by_file = MagicMock() - ph.protocol_util.get_protocol = MagicMock() - - conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir) - ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME) - ovfenv_data = load_data("ovf-env.xml") - fileutil.write_file(ovfenv_file, ovfenv_data) - - ph.run() - ph.report_event.assert_called_once_with( - "[ProvisionError] --unit-test--") - - if __name__ == '__main__': unittest.main() |