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_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_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)
