def test_sendBatchMessages(sqs): fake_data0 = {"foo": "bar"} fake_data1 = {"john": "doe"} jsonized0 = json.dumps(fake_data0) jsonized1 = json.dumps(fake_data1) md5_0 = hashlib.md5(jsonized0.encode("utf-8")).hexdigest() md5_1 = hashlib.md5(jsonized1.encode("utf-8")).hexdigest() proj = generate_proj() from ndingest.ndqueue.cleanupqueue import CleanupQueue CleanupQueue.createQueue(proj) cleanup_queue = CleanupQueue(proj) try: response = cleanup_queue.sendBatchMessages([fake_data0, fake_data1], 0) assert "Successful" in response success_ids = [] for msg_result in response["Successful"]: id = msg_result["Id"] success_ids.append(id) if id == "0": assert md5_0 == msg_result["MD5OfMessageBody"] elif id == "1": assert md5_1 == msg_result["MD5OfMessageBody"] assert "0" in success_ids assert "1" in success_ids finally: for message_id, receipt_handle, _ in cleanup_queue.receiveMessage(): cleanup_queue.deleteMessage(message_id, receipt_handle)
class Test_Cleanup_Queue(): def setup_class(self): """Setup class parameters""" if 'SQS_ENDPOINT' in dir(settings): self.endpoint_url = settings.SQS_ENDPOINT else: self.endpoint_url = None CleanupQueue.createQueue(nd_proj, endpoint_url=self.endpoint_url) self.cleanup_queue = CleanupQueue(nd_proj, endpoint_url=self.endpoint_url) def teardown_class(self): """Teardown parameters""" CleanupQueue.deleteQueue(nd_proj, endpoint_url=self.endpoint_url) def test_Message(self): """Testing the upload queue""" supercuboid_key = 'kasthuri11&image&0&0' self.cleanup_queue.sendMessage(supercuboid_key) for message_id, receipt_handle, message_body in self.cleanup_queue.receiveMessage(): assert(supercuboid_key == message_body) response = self.cleanup_queue.deleteMessage(message_id, receipt_handle) assert('Successful' in response) def test_sendBatchMessages(self): fake_data0 = {'foo': 'bar'} fake_data1 = {'john': 'doe'} jsonized0 = json.dumps(fake_data0) jsonized1 = json.dumps(fake_data1) md5_0 = hashlib.md5(jsonized0.encode('utf-8')).hexdigest() md5_1 = hashlib.md5(jsonized1.encode('utf-8')).hexdigest() try: response = self.cleanup_queue.sendBatchMessages([fake_data0, fake_data1], 0) assert('Successful' in response) success_ids = [] for msg_result in response['Successful']: id = msg_result['Id'] success_ids.append(id) if id == '0': assert(md5_0 == msg_result['MD5OfMessageBody']) elif id == '1': assert(md5_1 == msg_result['MD5OfMessageBody']) assert('0' in success_ids) assert('1' in success_ids) finally: for message_id, receipt_handle, _ in self.cleanup_queue.receiveMessage(): self.cleanup_queue.deleteMessage(message_id, receipt_handle)
def test_message(sqs): """Testing the upload queue""" proj = generate_proj() from ndingest.ndqueue.cleanupqueue import CleanupQueue CleanupQueue.createQueue(proj) cleanup_queue = CleanupQueue(proj) supercuboid_key = "kasthuri11&image&0&0" cleanup_queue.sendMessage(supercuboid_key) for message_id, receipt_handle, message_body in cleanup_queue.receiveMessage(): assert supercuboid_key == message_body response = cleanup_queue.deleteMessage(message_id, receipt_handle) assert "Successful" in response