3
3
from pyres import ResQ , str_to_class , Stat
4
4
from pyres .job import Job
5
5
from pyres .worker import Worker
6
+
6
7
class Basic (object ):
7
8
queue = 'basic'
8
9
@@ -50,24 +51,24 @@ def test_enqueue(self):
50
51
self .resq .enqueue (Basic ,"test1" )
51
52
self .resq .enqueue (Basic ,"test2" )
52
53
ResQ ._enqueue (Basic , "test3" )
53
- assert self .redis .llen ("queue:basic" ) == 3
54
- assert self .redis .sismember ('queues' ,'basic' )
54
+ assert self .redis .llen ("resque: queue:basic" ) == 3
55
+ assert self .redis .sismember ('resque: queues' ,'basic' )
55
56
56
57
def test_push (self ):
57
58
self .resq .push ('pushq' ,'content-newqueue' )
58
59
self .resq .push ('pushq' ,'content2-newqueue' )
59
- assert self .redis .llen ('queue:pushq' ) == 2
60
- assert self .redis .lindex ('queue:pushq' , 0 ) == ResQ .encode ('content-newqueue' )
61
- assert self .redis .lindex ('queue:pushq' , 1 ) == ResQ .encode ('content2-newqueue' )
60
+ assert self .redis .llen ('resque: queue:pushq' ) == 2
61
+ assert self .redis .lindex ('resque: queue:pushq' , 0 ) == ResQ .encode ('content-newqueue' )
62
+ assert self .redis .lindex ('resque: queue:pushq' , 1 ) == ResQ .encode ('content2-newqueue' )
62
63
63
64
def test_pop (self ):
64
65
self .resq .push ('pushq' ,'content-newqueue' )
65
66
self .resq .push ('pushq' ,'content2-newqueue' )
66
- assert self .redis .llen ('queue:pushq' ) == 2
67
+ assert self .redis .llen ('resque: queue:pushq' ) == 2
67
68
assert self .resq .pop ('pushq' ) == 'content-newqueue'
68
- assert self .redis .llen ('queue:pushq' ) == 1
69
+ assert self .redis .llen ('resque: queue:pushq' ) == 1
69
70
assert self .resq .pop ('pushq' ) == 'content2-newqueue'
70
- assert self .redis .llen ('queue:pushq' ) == 0
71
+ assert self .redis .llen ('resque: queue:pushq' ) == 0
71
72
72
73
def test_peek (self ):
73
74
self .resq .enqueue (Basic ,"test1" )
@@ -109,23 +110,23 @@ def test_workers(self):
109
110
def test_enqueue_from_string (self ):
110
111
self .resq .enqueue_from_string ('tests.Basic' ,'basic' ,'test1' )
111
112
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
112
- assert self .redis .llen ("queue:basic" ) == 1
113
+ assert self .redis .llen ("resque: queue:basic" ) == 1
113
114
job = Job .reserve ('basic' , self .resq )
114
115
worker = Worker (['basic' ])
115
116
worker .process (job )
116
- assert not self .redis .get ('worker:%s' % worker )
117
- assert not self .redis .get ("stat:failed" )
118
- assert not self .redis .get ("stat:failed:%s" % name )
117
+ assert not self .redis .get ('resque: worker:%s' % worker )
118
+ assert not self .redis .get ("resque: stat:failed" )
119
+ assert not self .redis .get ("resque: stat:failed:%s" % name )
119
120
120
121
def test_remove_queue (self ):
121
122
self .resq .enqueue_from_string ('tests.Basic' ,'basic' ,'test1' )
122
123
assert 'basic' in self .resq ._watched_queues
123
- assert self .redis .sismember ('queues' ,'basic' )
124
- assert self .redis .llen ('queue:basic' ) == 1
124
+ assert self .redis .sismember ('resque: queues' ,'basic' )
125
+ assert self .redis .llen ('resque: queue:basic' ) == 1
125
126
self .resq .remove_queue ('basic' )
126
127
assert 'basic' not in self .resq ._watched_queues
127
- assert not self .redis .sismember ('queues' ,'basic' )
128
- assert not self .redis .exists ('queue:basic' )
128
+ assert not self .redis .sismember ('resque: queues' ,'basic' )
129
+ assert not self .redis .exists ('resque: queue:basic' )
129
130
130
131
def test_keys (self ):
131
132
self .resq .enqueue_from_string ('tests.Basic' ,'basic' ,'test1' )
@@ -152,9 +153,9 @@ def test_perform(self):
152
153
def test_fail (self ):
153
154
self .resq .enqueue (Basic ,"test1" )
154
155
job = Job .reserve ('basic' ,self .resq )
155
- assert self .redis .llen ('failed' ) == 0
156
+ assert self .redis .llen ('resque: failed' ) == 0
156
157
job .fail ("problem" )
157
- assert self .redis .llen ('failed' ) == 1
158
+ assert self .redis .llen ('resque: failed' ) == 1
158
159
159
160
160
161
class WorkerTests (PyResTests ):
@@ -167,7 +168,7 @@ def test_startup(self):
167
168
worker = Worker (['basic' ])
168
169
worker .startup ()
169
170
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
170
- assert self .redis .sismember ('workers' ,name )
171
+ assert self .redis .sismember ('resque: workers' ,name )
171
172
import signal
172
173
assert signal .getsignal (signal .SIGTERM ) == worker .shutdown_all
173
174
assert signal .getsignal (signal .SIGINT ) == worker .shutdown_all
@@ -178,62 +179,62 @@ def test_register(self):
178
179
worker = Worker (['basic' ])
179
180
worker .register_worker ()
180
181
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
181
- assert self .redis .sismember ('workers' ,name )
182
+ assert self .redis .sismember ('resque: workers' ,name )
182
183
183
184
def test_unregister (self ):
184
185
worker = Worker (['basic' ])
185
186
worker .register_worker ()
186
187
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
187
- assert self .redis .sismember ('workers' ,name )
188
+ assert self .redis .sismember ('resque: workers' ,name )
188
189
worker .unregister_worker ()
189
- assert name not in self .redis .smembers ('workers' )
190
+ assert name not in self .redis .smembers ('resque: workers' )
190
191
191
192
def test_working_on (self ):
192
193
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
193
194
self .resq .enqueue (Basic ,"test1" )
194
195
job = Job .reserve ('basic' , self .resq )
195
196
worker = Worker (['basic' ])
196
197
worker .working_on (job )
197
- assert self .redis .exists ("worker:%s" % name )
198
+ assert self .redis .exists ("resque: worker:%s" % name )
198
199
199
200
def test_processed (self ):
200
201
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
201
202
worker = Worker (['basic' ])
202
203
worker .processed ()
203
- assert self .redis .exists ("stat:processed" )
204
- assert self .redis .exists ("stat:processed:%s" % name )
205
- assert self .redis .get ("stat:processed" ) == 1
206
- assert self .redis .get ("stat:processed:%s" % name ) == 1
204
+ assert self .redis .exists ("resque: stat:processed" )
205
+ assert self .redis .exists ("resque: stat:processed:%s" % name )
206
+ assert self .redis .get ("resque: stat:processed" ) == 1
207
+ assert self .redis .get ("resque: stat:processed:%s" % name ) == 1
207
208
worker .processed ()
208
- assert self .redis .get ("stat:processed" ) == 2
209
- assert self .redis .get ("stat:processed:%s" % name ) == 2
209
+ assert self .redis .get ("resque: stat:processed" ) == 2
210
+ assert self .redis .get ("resque: stat:processed:%s" % name ) == 2
210
211
211
212
def test_failed (self ):
212
213
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
213
214
worker = Worker (['basic' ])
214
215
worker .failed ()
215
- assert self .redis .exists ("stat:failed" )
216
- assert self .redis .exists ("stat:failed:%s" % name )
217
- assert self .redis .get ("stat:failed" ) == 1
218
- assert self .redis .get ("stat:failed:%s" % name ) == 1
216
+ assert self .redis .exists ("resque: stat:failed" )
217
+ assert self .redis .exists ("resque: stat:failed:%s" % name )
218
+ assert self .redis .get ("resque: stat:failed" ) == 1
219
+ assert self .redis .get ("resque: stat:failed:%s" % name ) == 1
219
220
worker .failed ()
220
- assert self .redis .get ("stat:failed" ) == 2
221
- assert self .redis .get ("stat:failed:%s" % name ) == 2
221
+ assert self .redis .get ("resque: stat:failed" ) == 2
222
+ assert self .redis .get ("resque: stat:failed:%s" % name ) == 2
222
223
223
224
def test_process (self ):
224
225
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
225
226
self .resq .enqueue (Basic ,"test1" )
226
227
job = Job .reserve ('basic' , self .resq )
227
228
worker = Worker (['basic' ])
228
229
worker .process (job )
229
- assert not self .redis .get ('worker:%s' % worker )
230
- assert not self .redis .get ("stat:failed" )
231
- assert not self .redis .get ("stat:failed:%s" % name )
230
+ assert not self .redis .get ('resque: worker:%s' % worker )
231
+ assert not self .redis .get ("resque: stat:failed" )
232
+ assert not self .redis .get ("resque: stat:failed:%s" % name )
232
233
self .resq .enqueue (Basic ,"test1" )
233
234
worker .process ()
234
- assert not self .redis .get ('worker:%s' % worker )
235
- assert not self .redis .get ("stat:failed" )
236
- assert not self .redis .get ("stat:failed:%s" % name )
235
+ assert not self .redis .get ('resque: worker:%s' % worker )
236
+ assert not self .redis .get ("resque: stat:failed" )
237
+ assert not self .redis .get ("resque: stat:failed:%s" % name )
237
238
238
239
239
240
def test_signals (self ):
@@ -255,17 +256,17 @@ def test_job_failure(self):
255
256
worker = Worker (['basic' ])
256
257
worker .process ()
257
258
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
258
- assert not self .redis .get ('worker:%s' % worker )
259
- assert self .redis .get ("stat:failed" ) == 1
260
- assert self .redis .get ("stat:failed:%s" % name ) == 1
259
+ assert not self .redis .get ('resque: worker:%s' % worker )
260
+ assert self .redis .get ("resque: stat:failed" ) == 1
261
+ assert self .redis .get ("resque: stat:failed:%s" % name ) == 1
261
262
262
263
def test_get_job (self ):
263
264
worker = Worker (['basic' ])
264
265
self .resq .enqueue (Basic ,"test1" )
265
266
job = Job .reserve ('basic' , self .resq )
266
267
worker .working_on (job )
267
268
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
268
- assert worker .job () == ResQ .decode (self .redis .get ('worker:%s' % name ))
269
+ assert worker .job () == ResQ .decode (self .redis .get ('resque: worker:%s' % name ))
269
270
worker .done_working ()
270
271
w2 = Worker (['basic' ])
271
272
print w2 .job ()
@@ -280,29 +281,31 @@ def test_working(self):
280
281
name = "%s:%s:%s" % (os .uname ()[1 ],os .getpid (),'basic' )
281
282
workers = Worker .working (self .resq )
282
283
assert len (workers ) == 1
284
+ print "...ccc..." ,str (worker )
285
+ print "..ddd..." ,str (workers [0 ])
283
286
assert str (worker ) == str (workers [0 ])
284
287
assert worker != workers [0 ]
285
288
286
289
class StatTests (PyResTests ):
287
290
def test_incr (self ):
288
291
stat_obj = Stat ('test_stat' , self .resq )
289
292
stat_obj .incr ()
290
- assert self .redis .get ('stat:test_stat' ) == 1
293
+ assert self .redis .get ('resque: stat:test_stat' ) == 1
291
294
stat_obj .incr ()
292
- assert self .redis .get ('stat:test_stat' ) == 2
295
+ assert self .redis .get ('resque: stat:test_stat' ) == 2
293
296
stat_obj .incr (2 )
294
- assert self .redis .get ('stat:test_stat' ) == 4
297
+ assert self .redis .get ('resque: stat:test_stat' ) == 4
295
298
296
299
def test_decr (self ):
297
300
stat_obj = Stat ('test_stat' , self .resq )
298
301
stat_obj .incr ()
299
302
stat_obj .incr ()
300
- assert self .redis .get ('stat:test_stat' ) == 2
303
+ assert self .redis .get ('resque: stat:test_stat' ) == 2
301
304
stat_obj .decr ()
302
- assert self .redis .get ('stat:test_stat' ) == 1
305
+ assert self .redis .get ('resque: stat:test_stat' ) == 1
303
306
stat_obj .incr ()
304
307
stat_obj .decr (2 )
305
- assert self .redis .get ('stat:test_stat' ) == 0
308
+ assert self .redis .get ('resque: stat:test_stat' ) == 0
306
309
307
310
def test_get (self ):
308
311
stat_obj = Stat ('test_stat' , self .resq )
@@ -314,7 +317,9 @@ def test_clear(self):
314
317
stat_obj = Stat ('test_stat' , self .resq )
315
318
stat_obj .incr ()
316
319
stat_obj .incr ()
317
- assert self .redis .exists ('stat:test_stat' )
320
+ assert self .redis .exists ('resque: stat:test_stat' )
318
321
stat_obj .clear ()
319
- assert not self .redis .exists ('stat:test_stat' )
322
+ assert not self .redis .exists ('resque: stat:test_stat' )
320
323
324
+ if __name__ == "__main__" :
325
+ unittest .main ()
0 commit comments