Skip to content

Commit aa977aa

Browse files
author
Matt George
committed
added more tests for better coverage. Total
package coverage is around 89% right now.
1 parent d18df5e commit aa977aa

File tree

3 files changed

+29
-2
lines changed

3 files changed

+29
-2
lines changed

tests/__init__.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,18 @@ def perform(sleep_time):
4343
import time
4444
time.sleep(sleep_time)
4545
print 'Done Sleeping'
46+
4647
def test_str_to_class():
4748
ret = str_to_class('tests.Basic')
4849
assert ret
50+
assert ret == Basic
51+
assert str_to_class('hello.World') == None
52+
53+
def test_safe_str_to_class():
54+
from pyres import safe_str_to_class
55+
assert safe_str_to_class('tests.Basic') == Basic
56+
assert safe_str_to_class('test.Mine') == None
57+
assert safe_str_to_class('hello.World') == None
4958

5059
class PyResTests(unittest.TestCase):
5160
def setUp(self):

tests/test_resq.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def test_redis_property(self):
4242
from redis import Redis
4343
rq = ResQ(server="localhost:6379")
4444
red = Redis()
45-
rq2 = ResQ(server=red)
45+
#rq2 = ResQ(server=red)
4646
self.assertRaises(Exception, rq.redis,[Basic])
4747

4848
def test_info(self):
@@ -89,3 +89,4 @@ def test_keys(self):
8989
self.resq.enqueue_from_string('tests.Basic','basic','test1')
9090
assert 'queue:basic' in self.resq.keys()
9191
assert 'queues' in self.resq.keys()
92+

tests/test_worker.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,11 @@ def test_processed(self):
5050
assert self.redis.exists("resque:stat:processed:%s" % name)
5151
assert self.redis.get("resque:stat:processed") == str(1)
5252
assert self.redis.get("resque:stat:processed:%s" % name) == str(1)
53+
assert worker.get_processed() == 1
5354
worker.processed()
5455
assert self.redis.get("resque:stat:processed") == str(2)
5556
assert self.redis.get("resque:stat:processed:%s" % name) == str(2)
57+
assert worker.get_processed() == 2
5658

5759
def test_failed(self):
5860
name = "%s:%s:%s" % (os.uname()[1],os.getpid(),'basic')
@@ -62,9 +64,11 @@ def test_failed(self):
6264
assert self.redis.exists("resque:stat:failed:%s" % name)
6365
assert self.redis.get("resque:stat:failed") == str(1)
6466
assert self.redis.get("resque:stat:failed:%s" % name) == str(1)
67+
assert worker.get_failed() == 1
6568
worker.failed()
6669
assert self.redis.get("resque:stat:failed") == str(2)
6770
assert self.redis.get("resque:stat:failed:%s" % name) == str(2)
71+
assert worker.get_failed() == 2
6872

6973
def test_process(self):
7074
name = "%s:%s:%s" % (os.uname()[1],os.getpid(),'basic')
@@ -112,6 +116,7 @@ def test_get_job(self):
112116
worker.working_on(job)
113117
name = "%s:%s:%s" % (os.uname()[1],os.getpid(),'basic')
114118
assert worker.job() == ResQ.decode(self.redis.get('resque:worker:%s' % name))
119+
assert worker.processing() == ResQ.decode(self.redis.get('resque:worker:%s' % name))
115120
worker.done_working()
116121
w2 = Worker(['basic'])
117122
print w2.job()
@@ -138,4 +143,16 @@ def test_started(self):
138143
assert self.redis.get('resque:worker:%s:started' % name) == dt.strftime('%Y-%m-%d %H:%M:%S')
139144
assert worker.started == datetime.datetime.strptime(dt.strftime('%Y-%m-%d %H:%M:%S'),'%Y-%m-%d %H:%M:%S')
140145
worker.started = None
141-
assert not self.redis.exists('resque:worker:%s:started' % name)
146+
assert not self.redis.exists('resque:worker:%s:started' % name)
147+
148+
def test_state(self):
149+
worker = Worker(['basic'])
150+
assert worker.state() == 'idle'
151+
self.resq.enqueue_from_string('tests.Basic','basic','test1')
152+
worker.register_worker()
153+
job = Job.reserve('basic', self.resq)
154+
worker.working_on(job)
155+
assert worker.state() == 'working'
156+
worker.done_working()
157+
assert worker.state() == 'idle'
158+

0 commit comments

Comments
 (0)