def _busy_test(self, xbusy, count=5):
activities = [
activity(activity_id=i, type_tag='Restore', state='Completed')
for i in range(count)
]
extra_jobs_status = 'Complete' if not xbusy else 'InProgress'
activities.extend(
activity(activity_id=i, type_tag='Backup', state=extra_jobs_status)
for i in range(count))
HTTPretty.register_uri(HTTPretty.GET,
self.url,
body=json.dumps(activities))
self.assertEqual(self.agent.busy, xbusy)
def _busy_test(self, xbusy, count=5):
activities = [activity(activity_id=i,
type_tag='Restore',
state='Completed')
for i in range(count)]
extra_jobs_status = 'Complete' if not xbusy else 'InProgress'
activities.extend(activity(activity_id=i,
type_tag='Backup',
state=extra_jobs_status)
for i in range(count))
HTTPretty.register_uri(HTTPretty.GET, self.url,
body=json.dumps(activities))
self.assertEqual(self.agent.busy, xbusy)
def _activity_test(self, method, xtype, xstatus, xcount):
activities = [activity(activity_id=i,
type_tag=xtype,
state=xstatus)
for i in range(xcount)]
HTTPretty.register_uri(HTTPretty.GET, self.url,
body=json.dumps(activities))
result = list(getattr(self.agent, method))
self.assertEqual(len(result), xcount)
for a in result:
self.assertEqual(a.type, xtype)
self.assertEqual(a.state, xstatus)
def _activity_test(self, method, xtype, xstatus, xcount):
activities = [
activity(activity_id=i, type_tag=xtype, state=xstatus)
for i in range(xcount)
]
HTTPretty.register_uri(HTTPretty.GET,
self.url,
body=json.dumps(activities))
result = list(getattr(self.agent, method))
self.assertEqual(len(result), xcount)
for a in result:
self.assertEqual(a.type, xtype)
self.assertEqual(a.state, xstatus)
def setUp(self):
self.activity = activity.from_dict(mock_activity.activity())
