def test_basic(self): sot = agent.Agent(EXAMPLE) self.assertEqual('MachineAgentId', sot.id_attribute) self.assertEqual('/user/agents', sot.base_path) self.assertEqual('cloudBackup', sot.service.service_name) self.assertFalse(sot.allow_create) self.assertTrue(sot.allow_retrieve) self.assertFalse(sot.allow_update) self.assertTrue(sot.allow_delete) self.assertTrue(sot.allow_list) self.assertEqual(EXAMPLE['CleanupAllowed'], sot.can_cleanup) self.assertEqual(EXAMPLE['BackupContainer'], sot.container) self.assertEqual(EXAMPLE['Datacenter'], sot.datacenter) self.assertEqual(EXAMPLE['Flavor'], sot.flavor) self.assertEqual(EXAMPLE['IsDisabled'], sot.is_disabled) self.assertEqual(EXAMPLE['IsEncrypted'], sot.is_encrypted) self.assertEqual(EXAMPLE['TimeOfLastSuccessfulBackup'], sot.last_successful_backup_at) self.assertEqual(EXAMPLE['PublicKey'], sot.public_key) self.assertEqual(EXAMPLE['Architecture'], sot.server_arch) self.assertEqual(EXAMPLE['MachineName'], sot.server_name) self.assertEqual(EXAMPLE['OperatingSystem'], sot.server_os) self.assertEqual(EXAMPLE['OperatingSystemVersion'], sot.server_os_version) self.assertEqual(EXAMPLE['IPAddress'], sot.server_ipaddress) self.assertEqual(EXAMPLE['HostServerId'], sot.server_uuid) self.assertEqual(EXAMPLE['UseServiceNet'], sot.servicenet) self.assertEqual(EXAMPLE['Status'], sot.status) self.assertEqual(EXAMPLE['BackupVaultSize'], sot.vault_size) self.assertEqual(EXAMPLE['AgentVersion'], sot.version)
def test_disable(self): response = mock.Mock() response.body = '' sess = mock.Mock() sess.post = mock.Mock() sess.post.return_value = response sot = agent.Agent(EXAMPLE) self.assertIsNone(sot.disable(sess)) body = {'MachineAgentId': sot.id, 'Enable': 'false'} url = 'agent/enable' sess.post.assert_called_with(url, service=sot.service, json=body)
def test_update_behavior(self): response = mock.Mock() response.body = '' sess = mock.Mock() sess.post = mock.Mock() sess.post.return_value = response sot = agent.Agent(EXAMPLE) self.assertIsNone(sot.update_behavior(sess, DATACENTER, SERVICENET)) body = {'BackupDataCenter': DATACENTER, 'UseServiceNet': SERVICENET} url = 'agent/%s' % sot.id sess.post.assert_called_with(url, service=sot.service, json=body)
def test_migrate_vault(self): response = mock.Mock() response.body = '' sess = mock.Mock() sess.post = mock.Mock() sess.post.return_value = response sot = agent.Agent(EXAMPLE) self.assertIsNone(sot.migrate_vault(sess, DESTINATION)) body = { 'SourceMachineAgentId': sot.id, 'DestinationMachineAgentId': DESTINATION } url = 'agent/migratevault' sess.post.assert_called_with(url, service=sot.service, json=body)
def test_encrypt_volume(self): response = mock.Mock() response.body = '' sess = mock.Mock() sess.post = mock.Mock() sess.post.return_value = response sot = agent.Agent(EXAMPLE) self.assertEqual('', sot.encrypt_volume(sess, ENCRYPTIONKEY)) body = { 'MachineAgentId': sot.id, 'EncryptedPasswordHex': ENCRYPTIONKEY } url = 'agent/encrypt' sess.post.assert_called_with(url, service=sot.service, json=body)
def test_update_encryption_password(self): response = mock.Mock() response.body = '' sess = mock.Mock() sess.post = mock.Mock() sess.post.return_value = response sot = agent.Agent(EXAMPLE) self.assertEqual( '', sot.update_encryption_password(sess, OLD_KEY, NEW_KEY)) body = { 'MachineAgentId': sot.id, 'OldEncryptedPasswordHex': OLD_KEY, 'NewEncryptedPasswordHex': NEW_KEY } url = 'agent/changeencryption' sess.post.assert_called_with(url, service=sot.service, json=body)