def test_save_image_exists_with_delete_not_none(self): raw = self.mox.CreateMockAnything() delete = self.mox.CreateMockAnything() audit_period_beginning = "2013-05-20 17:31:57.939614" audit_period_ending = "2013-06-20 17:31:57.939614" size = 123 uuid = "2df2ccf6-bc1b-4853-aab0-25fda346b3bb" deleted_at = "2013-06-20 14:31:57.939614" body = { "event_type": "image.upload", "timestamp": "2013-06-20 18:31:57.939614", "publisher_id": "glance-api01-r2961.global.preprod-ord.ohthree.com", "payload": { "created_at": str(DUMMY_TIME), "status": "saving", "audit_period_beginning": audit_period_beginning, "audit_period_ending": audit_period_ending, "properties": { "image_type": "snapshot", "instance_uuid": INSTANCE_ID_1, }, "deleted_at": deleted_at, "size": size, "owner": TENANT_ID_1, "id": "2df2ccf6-bc1b-4853-aab0-25fda346b3bb", } } deployment = "1" routing_key = "glance_monitor.info" json = '{["routing_key", {%s}]}' % body self.mox.StubOutWithMock(db, 'create_image_exists') self.mox.StubOutWithMock(db, 'get_image_usage') self.mox.StubOutWithMock(db, 'get_image_delete') created_at_range = (DECIMAL_DUMMY_TIME, DECIMAL_DUMMY_TIME+1) db.get_image_usage(created_at__range=created_at_range, uuid=uuid).AndReturn(None) db.get_image_delete(created_at__range=created_at_range, uuid=uuid).AndReturn(delete) db.create_image_exists( created_at=utils.str_time_to_unix(str(DUMMY_TIME)), owner=TENANT_ID_1, raw=raw, audit_period_beginning=utils.str_time_to_unix(audit_period_beginning), audit_period_ending=utils.str_time_to_unix(audit_period_ending), size=size, uuid=uuid, usage=None, delete=delete, deleted_at=utils.str_time_to_unix(str(deleted_at))).AndReturn(raw) self.mox.ReplayAll() notification = GlanceNotification(body, deployment, routing_key, json) notification.save_exists(raw) self.mox.VerifyAll()
def save_exists(self, raw): if self.created_at: values = { 'uuid': self.uuid, 'audit_period_beginning': self.audit_period_beginning, 'audit_period_ending': self.audit_period_ending, 'owner': self.owner, 'size': self.size, 'raw': raw } created_at_range = (self.created_at, self.created_at+1) usage = db.get_image_usage( uuid=self.uuid, created_at__range=created_at_range) values['usage'] = usage values['created_at'] = self.created_at if self.deleted_at: delete = db.get_image_delete( uuid=self.uuid, created_at__range=created_at_range) values['delete'] = delete values['deleted_at'] = self.deleted_at db.create_image_exists(**values) else: stacklog.warn("Ignoring exists without created_at. GlanceRawData(%s)" % raw.id)
def save_exists(self, raw): if isinstance(self.payload, dict): audit_period_beginning = self.payload.get( 'audit_period_beginning', None) audit_period_beginning = audit_period_beginning and\ utils.str_time_to_unix(audit_period_beginning) audit_period_ending = self.payload.get( 'audit_period_ending', None) audit_period_ending = audit_period_ending and \ utils.str_time_to_unix(audit_period_ending) message_id = self.message_id images = self.payload.get('images', []) else: stacklog.warn("Received exists with invalid payload " "GlanceRawData(%s)" % raw.id) audit_period_beginning = None audit_period_ending = None images = [] for image in images: created_at = image['created_at'] created_at = created_at and utils.str_time_to_unix(created_at) uuid = image['id'] deleted_at = image['deleted_at'] deleted_at = deleted_at and utils.str_time_to_unix(deleted_at) if created_at: values = { 'uuid': uuid, 'audit_period_beginning': audit_period_beginning, 'audit_period_ending': audit_period_ending, 'owner': self.owner, 'size': image['size'], 'raw': raw, 'message_id': message_id } usage = db.get_image_usage(uuid=uuid) values['usage'] = usage values['created_at'] = created_at if deleted_at: delete = db.get_image_delete(uuid=uuid) values['delete'] = delete values['deleted_at'] = deleted_at db.create_image_exists(**values) else: stacklog.warn("Ignoring exists without created_at. GlanceRawData(%s)" % raw.id)
def save_exists(self, raw): if isinstance(self.payload, dict): audit_period_beginning = self.payload.get('audit_period_beginning', None) audit_period_beginning = audit_period_beginning and\ utils.str_time_to_unix(audit_period_beginning) audit_period_ending = self.payload.get('audit_period_ending', None) audit_period_ending = audit_period_ending and \ utils.str_time_to_unix(audit_period_ending) message_id = self.message_id images = self.payload.get('images', []) else: stacklog.warn("Received exists with invalid payload " "GlanceRawData(%s)" % raw.id) audit_period_beginning = None audit_period_ending = None images = [] for image in images: created_at = image['created_at'] created_at = created_at and utils.str_time_to_unix(created_at) uuid = image['id'] deleted_at = image['deleted_at'] deleted_at = deleted_at and utils.str_time_to_unix(deleted_at) if created_at: values = { 'uuid': uuid, 'audit_period_beginning': audit_period_beginning, 'audit_period_ending': audit_period_ending, 'owner': self.owner, 'size': image['size'], 'raw': raw, 'message_id': message_id } usage = db.get_image_usage(uuid=uuid) values['usage'] = usage values['created_at'] = created_at if deleted_at: delete = db.get_image_delete(uuid=uuid) values['delete'] = delete values['deleted_at'] = deleted_at db.create_image_exists(**values) else: stacklog.warn( "Ignoring exists without created_at. GlanceRawData(%s)" % raw.id)
def test_save_image_exists_with_created_at_and_deleted_at(self): raw = self.mox.CreateMockAnything() delete = self.mox.CreateMockAnything() audit_period_beginning = "2013-05-20 17:31:57.939614" audit_period_ending = "2013-06-20 17:31:57.939614" created_at = "2013-05-20 19:31:57.939614" deleted_at = "2013-05-20 21:31:57.939614" size = 123 uuid = "2df2ccf6-bc1b-4853-aab0-25fda346b3bb" body = { "event_type": "image.exists", "timestamp": "2013-06-20 18:31:57.939614", "publisher_id": "glance-api01-r2961.global.preprod-ord.ohthree.com", "message_id": "d14cfa51-6a0e-4cf8-9130-804738be96d2", "payload": { "audit_period_beginning": audit_period_beginning, "audit_period_ending": audit_period_ending, "owner": TENANT_ID_1, "images": [{ "created_at": created_at, "id": uuid, "size": size, "status": "saving", "properties": { "instance_uuid": INSTANCE_ID_1 }, "deleted_at": deleted_at, }, { "created_at": created_at, "id": uuid, "size": size, "status": "saving", "properties": { "instance_uuid": INSTANCE_ID_1 }, "deleted_at": deleted_at, }] } } deployment = "1" routing_key = "glance_monitor.info" json_body = json.dumps([routing_key, body]) self.mox.StubOutWithMock(db, 'create_image_exists') self.mox.StubOutWithMock(db, 'get_image_usage') self.mox.StubOutWithMock(db, 'get_image_delete') for i in range(0, 2): db.get_image_usage(uuid=uuid).AndReturn(None) db.get_image_delete(uuid=uuid).AndReturn(delete) db.create_image_exists( created_at=utils.str_time_to_unix(created_at), owner=TENANT_ID_1, raw=raw, audit_period_beginning=utils.str_time_to_unix( audit_period_beginning), audit_period_ending=utils.str_time_to_unix( audit_period_ending), size=size, uuid=uuid, usage=None, delete=delete, deleted_at=utils.str_time_to_unix(deleted_at), message_id="d14cfa51-6a0e-4cf8-9130-804738be96d2").AndReturn( raw) self.mox.ReplayAll() notification = GlanceNotification(body, deployment, routing_key, json_body) notification.save_exists(raw) self.mox.VerifyAll()
def test_save_image_exists_with_created_at_and_deleted_at(self): raw = self.mox.CreateMockAnything() delete = self.mox.CreateMockAnything() audit_period_beginning = "2013-05-20 17:31:57.939614" audit_period_ending = "2013-06-20 17:31:57.939614" created_at = "2013-05-20 19:31:57.939614" deleted_at = "2013-05-20 21:31:57.939614" size = 123 uuid = "2df2ccf6-bc1b-4853-aab0-25fda346b3bb" body = { "event_type": "image.exists", "timestamp": "2013-06-20 18:31:57.939614", "publisher_id": "glance-api01-r2961.global.preprod-ord.ohthree.com", "message_id": "d14cfa51-6a0e-4cf8-9130-804738be96d2", "payload": { "audit_period_beginning": audit_period_beginning, "audit_period_ending": audit_period_ending, "owner": TENANT_ID_1, "images": [ { "created_at": created_at, "id": uuid, "size": size, "status": "saving", "properties": {"instance_uuid": INSTANCE_ID_1}, "deleted_at": deleted_at, }, { "created_at": created_at, "id": uuid, "size": size, "status": "saving", "properties": {"instance_uuid": INSTANCE_ID_1}, "deleted_at": deleted_at, } ] } } deployment = "1" routing_key = "glance_monitor.info" json_body = json.dumps([routing_key, body]) self.mox.StubOutWithMock(db, 'create_image_exists') self.mox.StubOutWithMock(db, 'get_image_usage') self.mox.StubOutWithMock(db, 'get_image_delete') for i in range(0, 2): db.get_image_usage(uuid=uuid).AndReturn(None) db.get_image_delete(uuid=uuid).AndReturn(delete) db.create_image_exists( created_at=utils.str_time_to_unix(created_at), owner=TENANT_ID_1, raw=raw, audit_period_beginning=utils.str_time_to_unix(audit_period_beginning), audit_period_ending=utils.str_time_to_unix(audit_period_ending), size=size, uuid=uuid, usage=None, delete=delete, deleted_at=utils.str_time_to_unix(deleted_at), message_id="d14cfa51-6a0e-4cf8-9130-804738be96d2").AndReturn(raw) self.mox.ReplayAll() notification = GlanceNotification(body, deployment, routing_key, json_body) notification.save_exists(raw) self.mox.VerifyAll()