CatalogueRevamping
/
harvester-d4science
14
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changes in Redis conusmer to make tests work

This commit is contained in:
amercader 2013-04-22 18:08:19 +01:00
parent 70dfee1a36
commit 9041f3f3ad
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ckanext/harvest/queue.py
View File

@ -125,6 +125,7 @@ class RedisPublisher(object):
if self.routing_key == 'harvest_job_id': if self.routing_key == 'harvest_job_id':
self.redis.lrem(self.routing_key, 0, value) self.redis.lrem(self.routing_key, 0, value)
self.redis.rpush(self.routing_key, value) self.redis.rpush(self.routing_key, value)
def close(self): def close(self):
return return
@ -162,7 +163,11 @@ class RedisConsumer(object):
return self.routing_key + ':' + message[self.routing_key] return self.routing_key + ':' + message[self.routing_key]
def basic_ack(self, message): def basic_ack(self, message):
self.redis.delete(self.persistance_key(message)) self.redis.delete(self.persistance_key(message))
def queue_purge(self, queue):
self.redis.flushall()
def basic_get(self, queue):
body = self.redis.lpop(self.routing_key)
return (FakeMethod(body), self, body)
def get_consumer(queue_name, routing_key): def get_consumer(queue_name, routing_key):