diff --git a/tuskar_ui/infrastructure/flavors/tabs.py b/tuskar_ui/infrastructure/flavors/tabs.py index 5f9f15b48..aa022f037 100644 --- a/tuskar_ui/infrastructure/flavors/tabs.py +++ b/tuskar_ui/infrastructure/flavors/tabs.py @@ -44,6 +44,9 @@ class FlavorsTab(horizon.tabs.TableTab): template_name = ("horizon/common/_detail_table.html") preload = False + def get_items_count(self): + return len(self.get_flavors_data()) + def get_flavors_data(self): flavors = api.flavor.Flavor.list(self.request) flavors.sort(key=lambda np: (np.vcpus, np.ram, np.disk)) @@ -127,6 +130,9 @@ class FlavorSuggestionsTab(horizon.tabs.TableTab): template_name = ("horizon/common/_detail_table.html") preload = False + def get_items_count(self): + return len(self.get_flavor_suggestions_data()) + def get_flavor_suggestions_data(self): return list(get_flavor_suggestions(self.request)) @@ -138,3 +144,4 @@ class FlavorTabs(horizon.tabs.TabGroup): FlavorSuggestionsTab, ) sticky = True + template_name = "horizon/common/_items_count_tab_group.html" diff --git a/tuskar_ui/infrastructure/flavors/tests.py b/tuskar_ui/infrastructure/flavors/tests.py index 8eae006c5..bb9d3767e 100644 --- a/tuskar_ui/infrastructure/flavors/tests.py +++ b/tuskar_ui/infrastructure/flavors/tests.py @@ -85,7 +85,7 @@ class FlavorsTest(test.BaseAdminViewTests): res = self.client.get(INDEX_URL) self.assertEqual(plans_mock.call_count, 1) self.assertEqual(roles_mock.call_count, 4) - self.assertEqual(flavors_mock.call_count, 2) + self.assertEqual(flavors_mock.call_count, 5) self.assertEqual(servers_mock.call_count, 1) self.assertTemplateUsed(res, 'infrastructure/flavors/index.html')