def test_get_tag_cluster_filter(self):
"""Test that tags from a cluster are returned with the cluster filter."""
url = "?filter[cluster]=OCP-on-AWS"
query_params = self.mocked_query_params(url, OCPAWSTagView)
handler = OCPAWSTagQueryHandler(query_params)
with tenant_context(self.tenant):
tag_keys = (OCPAWSTagsSummary.objects.filter(
report_period__cluster_id__contains="OCP-on-AWS").values(
"key").distinct().all())
tag_keys = [tag.get("key") for tag in tag_keys]
result = handler.get_tag_keys()
self.assertEqual(sorted(result), sorted(tag_keys))
def test_get_tag_keys(self):
"""Test that all OCP-on-AWS tag keys are returned."""
url = "?filter[time_scope_units]=month&filter[time_scope_value]=-1&filter[resolution]=monthly"
query_params = self.mocked_query_params(url, OCPAWSTagView)
handler = OCPAWSTagQueryHandler(query_params)
with tenant_context(self.tenant):
tag_keys = (OCPAWSCostLineItemDailySummary.objects.annotate(
tag_keys=JSONBObjectKeys("tags")).values(
"tag_keys").distinct().all())
tag_keys = [tag.get("tag_keys") for tag in tag_keys]
result = handler.get_tag_keys()
self.assertEqual(sorted(result), sorted(tag_keys))
