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)