def testSuccess(self, factory_mock, _):
inventory_service_mock = mock.Mock(spec=services.InventoryService)
factory_mock.return_value = inventory_service_mock
inventory_service_mock.IsRetiredMac.side_effect = lambda s: s == '1234'
test_util.MakeAppleFirmware(serial='1234')
test_util.MakeAppleFirmware(serial='231')
response = self.testapp.get('/api/internal/retired-assets/1234,231')
res = util.FromSafeJson(response.body)
self.assertEqual(['231'], res['active'])
self.assertEqual([{
'password': '******',
'serial': '1234'
}], res['retired'])
def testUpdateOwner(self, factory_mock):
new_owner = 'plague'
factory_mock.return_value.GetMetadataUpdates.return_value = {
'owners': [new_owner],
}
test_util.MakeAppleFirmware(save=True, owner='acidburn')
self.testapp.get('/cron/inventory_sync')
test_util.RunAllDeferredTasks(self.testbed)
entities = firmware.AppleFirmwarePassword.all().fetch(10)
self.assertEqual(1, len(entities))
self.assertEqual([new_owner + '@example.com'], entities[0].owners)
self.assertTrue(entities[0].force_rekeying)
def testUpdateHostnameAppleFirmaware(self, factory_mock):
new_hostname = 'somethingelse'
factory_mock.return_value.GetMetadataUpdates.return_value = {
'hostname': new_hostname,
'non_existing_property': 112,
}
test_util.MakeAppleFirmware(save=True, hostname='hostname1')
self.testapp.get('/cron/inventory_sync')
test_util.RunAllDeferredTasks(self.testbed)
entities = firmware.AppleFirmwarePassword.all().fetch(10)
self.assertEqual(1, len(entities))
self.assertEqual(new_hostname, entities[0].hostname)
def testAccessDenied(self):
test_util.MakeAppleFirmware(serial='1234')
test_util.MakeAppleFirmware(serial='231')
self.testapp.get('/api/internal/retired-assets/1234,231',
status=httplib.FORBIDDEN)