Esempio n. 1
0
 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)