From f44766ee54273ad21c96dc54a0da56b3cc836db4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ladislav=20Va=C5=A1ina?= <lvasina@redhat.com>
Date: Thu, 18 Apr 2024 08:39:30 +0200
Subject: [PATCH] Added entity that returns counts of the errata types (#1337)

---
 airgun/entities/host_new.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/airgun/entities/host_new.py b/airgun/entities/host_new.py
index 016c0997a..257b4d457 100644
--- a/airgun/entities/host_new.py
+++ b/airgun/entities/host_new.py
@@ -302,6 +302,30 @@ def get_errata_by_type(self, entity_name, type):
         view.content.errata.table.wait_displayed()
         return view.read(widget_names="content.errata.table")
 
+    def get_errata_type_counts(self, entity_name):
+        """
+        Get errata counts for each type of errata on selected host.
+
+        Args:
+            entity_name: Name of the host.
+        Returns:
+            errata_counts (dict): Dictionary with counts of each type of errata.
+        """
+
+        errata_types = ['Security', 'Bugfix', 'Enhancement']
+        errata_counts = {type: 0 for type in errata_types}
+        view = self.navigate_to(self, 'NewDetails', entity_name=entity_name)
+        view.wait_displayed()
+        view.content.errata.select()
+        for type in errata_types:
+            view.content.errata.wait_displayed()
+            view.content.errata.pagination.set_per_page(50)
+            view.content.errata.type_filter.fill(type)
+            self.browser.plugin.ensure_page_safe()
+            view.content.errata.table.wait_displayed()
+            errata_counts[type] = view.content.errata.table.row_count
+        return errata_counts
+
     def apply_erratas(self, entity_name, search=None):
         """Apply available errata on selected host based on searchbar result.