dt/admin: Add get_enterprise

Signed-off-by: Oren Leiman <oren.leiman@redpanda.com>
oleiman · Sep 17, 2024 · b5c06fa · b5c06fa
1 parent 5fdc178
commit b5c06fa
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/rptest/services/admin.py b/tests/rptest/services/admin.py
@@ -355,6 +355,12 @@ class MigrationAction(Enum):
     cancel = "cancel"
 
 
+class EnterpriseLicenseStatus(Enum):
+    valid = "valid"
+    expired = "expired"
+    not_present = "not_present"
+
+
 class Admin:
     """
     Wrapper for Redpanda admin REST API.
@@ -820,6 +826,9 @@ def get_license(self, node=None, timeout=None):
     def put_license(self, license):
         return self._request("PUT", "features/license", data=license)
 
+    def get_enterprise_features(self):
+        return self._request("GET", "features/enterprise")
+
     def get_loggers(self, node):
         """
         Get the names of all loggers.