def test_delete_issue(self): iss = monitor.WorkerIssue('min-procs-fail', 5, 78) yield self.stg.open_or_update_issue('worker-1', iss) res = yield self.redis.get(storage.issue_key('worker-1')) self.assertEqual(type(res), str) yield self.stg.delete_worker_issue('worker-1') res = yield self.redis.get(storage.issue_key('worker-1')) self.assertEqual(res, None)
def test_open_or_update_issue(self): obj = { 'issue_type': 'min-procs-fail', 'start_time': 5, 'procs_count': 78, } iss = monitor.WorkerIssue('min-procs-fail', 5, 78) yield self.stg.open_or_update_issue('foo', iss) res = yield self.redis.get(storage.issue_key('foo')) self.assertEqual(res, json.dumps(obj)) # now update the issue iss = monitor.WorkerIssue('min-procs-fail', 5, 77) obj['procs_count'] = 77 yield self.stg.open_or_update_issue('foo', iss) res = yield self.redis.get(storage.issue_key('foo')) self.assertEqual(res, json.dumps(obj))
def test_audit_fail(self): # here we test the verification of a worker who # who had less than min_procs check in yield self.worker.startWorker() fkredis = self.worker._redis attrs = self.gen_fake_attrs(time.time()) wkr_id = attrs['worker_id'] # process the fake message () yield self.worker.update(attrs) wkr = self.worker._workers[attrs['worker_id']] wkr.snapshot() yield wkr.audit(self.worker._storage) # test that an issue was opened self.assertEqual(wkr.procs_count, 1) key = issue_key(wkr_id) issue = json.loads((yield fkredis.get(key))) self.assertEqual(issue['issue_type'], 'min-procs-fail')
def test_audit_pass(self): # here we test the verification of a worker who # who had more than min_procs check in yield self.worker.startWorker() fkredis = self.worker._redis attrs = self.gen_fake_attrs(time.time()) wkr_id = attrs['worker_id'] # process the fake message () yield self.worker.update(attrs) attrs['pid'] = 2342 yield self.worker.update(attrs) wkr = self.worker._workers[attrs['worker_id']] wkr.snapshot() yield wkr.audit(self.worker._storage) # verify that no issue has been opened self.assertEqual(wkr.procs_count, 2) key = issue_key(wkr_id) issue = yield fkredis.get(key) self.assertEqual(issue, None)