def test_strip_label_column_name(self):
"""Test that the tag column name is stripped from results."""
# '?'
query_params = FakeQueryParameters({},
report_type='cpu',
tenant=self.tenant)
handler = OCPReportQueryHandler(query_params.mock_qp)
tag_column = handler._mapper.provider_map.get('tag_column')
data = [{
f'{tag_column}__tag_key1': 'value'
}, {
f'{tag_column}__tag_key2': 'value'
}]
group_by = [
'date', f'{tag_column}__tag_key1', f'{tag_column}__tag_key2'
]
expected_data = [{'tag_key1': 'value'}, {'tag_key2': 'value'}]
expected_group_by = ['date', 'tag_key1', 'tag_key2']
result_data, result_group_by = handler.strip_label_column_name(
data, group_by)
self.assertEqual(result_data, expected_data)
self.assertEqual(result_group_by, expected_group_by)
def test_strip_label_column_name(self):
"""Test that the tag column name is stripped from results."""
url = '?'
query_params = self.mocked_query_params(url, OCPCpuView)
handler = OCPReportQueryHandler(query_params)
tag_column = handler._mapper.provider_map.get('tag_column')
data = [
{
f'{tag_column}__tag_key1': 'value'
},
{
f'{tag_column}__tag_key2': 'value'
},
]
group_by = [
'date', f'{tag_column}__tag_key1', f'{tag_column}__tag_key2'
]
expected_data = [{'tag_key1': 'value'}, {'tag_key2': 'value'}]
expected_group_by = ['date', 'tag_key1', 'tag_key2']
result_data, result_group_by = handler.strip_label_column_name(
data, group_by)
self.assertEqual(result_data, expected_data)
self.assertEqual(result_group_by, expected_group_by)
def test_strip_label_column_name(self):
"""Test that the tag column name is stripped from results."""
query_params = {}
handler = OCPReportQueryHandler(query_params, '', self.tenant,
**{'report_type': 'cpu'})
tag_column = handler._mapper.provider_map.get('tag_column')
data = [{
f'{tag_column}__tag_key1': 'value'
}, {
f'{tag_column}__tag_key2': 'value'
}]
group_by = [
'date', f'{tag_column}__tag_key1', f'{tag_column}__tag_key2'
]
expected_data = [{'tag_key1': 'value'}, {'tag_key2': 'value'}]
expected_group_by = ['date', 'tag_key1', 'tag_key2']
result_data, result_group_by = handler.strip_label_column_name(
data, group_by)
self.assertEqual(result_data, expected_data)
self.assertEqual(result_group_by, expected_group_by)
