def test_skip_collection_for_errorful_cluster(self, get_info_mock, *_):
error_cluster = self.env.create(
api=False,
nodes_kwargs=[{"roles": ["controller"], "online": False}],
cluster_kwargs={"name": "error",
"status": consts.CLUSTER_STATUSES.operational}
)
normal_cluster = self.env.create(
api=False,
nodes_kwargs=[{"roles": ["controller"], "online": True}],
cluster_kwargs={"name": "normal",
"status": consts.CLUSTER_STATUSES.operational}
)
get_info_mock.return_value = self.vms_info
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last_for_error_clsr = OpenStackWorkloadStats.get_last_by(
error_cluster["id"], consts.OSWL_RESOURCE_TYPES.vm)
self.assertIsNone(last_for_error_clsr)
last_for_normal_clsr = OpenStackWorkloadStats.get_last_by(
normal_cluster["id"], consts.OSWL_RESOURCE_TYPES.vm)
self.assertIsNotNone(last_for_normal_clsr)
upd_time = last_for_normal_clsr.updated_time
res_data = {
'added': [{'time': upd_time.isoformat(), 'id': 1}],
'removed': [],
'modified': [],
'current': self.vms_info}
self.assertEqual(last_for_normal_clsr.resource_data, res_data)
def test_clear_data_for_removed_cluster(self, get_info_mock, *_):
cls_id, res_data = self.collect_for_operational_cluster(get_info_mock)
cls = Cluster.get_by_uid(cls_id)
Cluster.delete(cls)
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
removed = dict(self.vms_info[0])
removed['time'] = last.updated_time.isoformat()
res_data.update({'removed': [removed], 'current': []})
# current data is cleared when cluster is deleted
self.assertEqual(last.resource_data, res_data)
def test_clear_data_for_removed_cluster(self, get_info_mock, *_):
cls_id, res_data = self.collect_for_operational_cluster(get_info_mock)
cls = Cluster.get_by_uid(cls_id)
Cluster.delete(cls)
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
removed = dict(self.vms_info[0])
removed['time'] = last.updated_time.isoformat()
res_data.update({
'removed': [removed],
'current': []})
# current data is cleared when cluster is deleted
self.assertEqual(last.resource_data, res_data)
def test_skip_collection_for_errorful_cluster(self, get_info_mock, *_):
error_cluster = self.env.create(api=False,
nodes_kwargs=[{
"roles": ["controller"],
"online": False
}],
cluster_kwargs={
"name":
"error",
"status":
consts.CLUSTER_STATUSES.operational
})
normal_cluster = self.env.create(
api=False,
nodes_kwargs=[{
"roles": ["controller"],
"online": True
}],
cluster_kwargs={
"name": "normal",
"status": consts.CLUSTER_STATUSES.operational
})
get_info_mock.return_value = self.vms_info
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last_for_error_clsr = OpenStackWorkloadStats.get_last_by(
error_cluster["id"], consts.OSWL_RESOURCE_TYPES.vm)
self.assertIsNone(last_for_error_clsr)
last_for_normal_clsr = OpenStackWorkloadStats.get_last_by(
normal_cluster["id"], consts.OSWL_RESOURCE_TYPES.vm)
self.assertIsNotNone(last_for_normal_clsr)
upd_time = last_for_normal_clsr.updated_time
res_data = {
'added': [{
'time': upd_time.isoformat(),
'id': 1
}],
'removed': [],
'modified': [],
'current': self.vms_info
}
self.assertEqual(last_for_normal_clsr.resource_data, res_data)
def collect_for_operational_cluster(self, get_info_mock):
cluster = self.env.create_cluster(
api=False,
status=consts.CLUSTER_STATUSES.operational
)
cls_id = cluster.id
get_info_mock.return_value = self.vms_info
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
upd_time = last.updated_time
res_data = {
'added': [{'time': upd_time.isoformat(), 'id': 1}],
'removed': [],
'modified': [],
'current': self.vms_info}
self.assertEqual(last.resource_data, res_data)
return cls_id, res_data
def collect_for_operational_cluster(self, get_info_mock):
cluster = self.env.create_cluster(
api=False, status=consts.CLUSTER_STATUSES.operational)
cls_id = cluster.id
get_info_mock.return_value = self.vms_info
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
upd_time = last.updated_time
res_data = {
'added': [{
'time': upd_time.isoformat(),
'id': 1
}],
'removed': [],
'modified': [],
'current': self.vms_info
}
self.assertEqual(last.resource_data, res_data)
return cls_id, res_data
def test_removed_several_times(self, get_info_mock, *_):
cls_id, res_data = self.collect_for_operational_cluster(get_info_mock)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
self.assertItemsEqual(self.vms_info, last.resource_data['current'])
# reset cluster
get_info_mock.return_value = []
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
removed = dict(self.vms_info[0])
removed['time'] = last.updated_time.isoformat()
removed_data = [removed]
# check data is not duplicated in removed on several collects
for _ in xrange(10):
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
self.assertEqual(removed_data, last.resource_data['removed'])
# cluster is operational
# checking 'removed' is don't changed
get_info_mock.return_value = self.vms_info
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
self.assertEqual(removed_data, last.resource_data['removed'])
# reset cluster again
# checking only id and time added to 'removed'
get_info_mock.return_value = []
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
removed_data.append({
'id': removed_data[0]['id'],
'time': last.updated_time.isoformat()
})
self.assertListEqual(removed_data, last.resource_data['removed'])
def test_removed_several_times(self, get_info_mock, *_):
cls_id, res_data = self.collect_for_operational_cluster(get_info_mock)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
self.assertItemsEqual(self.vms_info, last.resource_data['current'])
# reset cluster
get_info_mock.return_value = []
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
removed = dict(self.vms_info[0])
removed['time'] = last.updated_time.isoformat()
removed_data = [removed]
# check data is not duplicated in removed on several collects
for _ in xrange(10):
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
self.assertEqual(removed_data, last.resource_data['removed'])
# cluster is operational
# checking 'removed' is don't changed
get_info_mock.return_value = self.vms_info
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
self.assertEqual(removed_data, last.resource_data['removed'])
# reset cluster again
# checking only id and time added to 'removed'
get_info_mock.return_value = []
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
last = OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
removed_data.append({
'id': removed_data[0]['id'],
'time': last.updated_time.isoformat()
})
self.assertListEqual(removed_data, last.resource_data['removed'])
def update_cluster_status_and_oswl_data(self, cls_id, status):
cls = Cluster.get_by_uid(cls_id)
Cluster.update(cls, {'status': status})
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
return OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)
def update_cluster_status_and_oswl_data(self, cls_id, status):
cls = Cluster.get_by_uid(cls_id)
Cluster.update(cls, {'status': status})
oswl_collect_once(consts.OSWL_RESOURCE_TYPES.vm)
return OpenStackWorkloadStats.get_last_by(
cls_id, consts.OSWL_RESOURCE_TYPES.vm)