def test_disable_boot_diagnostics_on_vm(self, mock_vm_set, mock_vm_get): vm_fake = mock.MagicMock() mock_vm_get.return_value = vm_fake vm_fake.diagnostics_profile.boot_diagnostics.enabled = True vm_fake.diagnostics_profile.boot_diagnostics.storage_uri = 'storage_uri1' disable_boot_diagnostics('g1', 'vm1') self.assertFalse(vm_fake.diagnostics_profile.boot_diagnostics.enabled) self.assertIsNone(vm_fake.diagnostics_profile.boot_diagnostics.storage_uri) self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm_fake, mock.ANY)
def test_disable_boot_diagnostics_on_vm(self, mock_vm_set, mock_vm_get): vm_fake = mock.MagicMock() mock_vm_get.return_value = vm_fake vm_fake.diagnostics_profile.boot_diagnostics.enabled = True vm_fake.diagnostics_profile.boot_diagnostics.storage_uri = 'storage_uri1' disable_boot_diagnostics('g1', 'vm1') self.assertFalse(vm_fake.diagnostics_profile.boot_diagnostics.enabled) self.assertIsNone(vm_fake.diagnostics_profile.boot_diagnostics.storage_uri) self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm_fake, mock.ANY)