#357 - Group template unit tests (#697)

Add unit tests for the "group template" feature

Author: ecolternv

src/service/core/tests/BUILD

@@ -40,6 +40,25 @@ osmo_py_test(
     visibility = ["//visibility:public"],
 )
 
+osmo_py_test(
+    name = "test_group_templates",
+    srcs = ["test_group_templates.py"],
+    deps = [
+        ":fixture",
+        "//src/lib/utils:common",
+        "//src/lib/utils:osmo_errors",
+        "//src/lib/utils:priority",
+        "//src/service/core:service_lib",
+        "//src/tests/common",
+        "//src/tests/common/storage",
+    ],
+    size = "large",
+    tags = [
+        "requires-network",
+    ],
+    visibility = ["//visibility:public"],
+)
+
 osmo_py_library(
     name = "fixture",
     srcs = ["fixture.py"],

src/service/core/tests/fixture.py

@@ -19,6 +19,7 @@
 import logging
 import io
 import time
+from typing import Any, Dict
 
 from fastapi import testclient
 
@@ -139,9 +140,23 @@ def create_test_backend(self, database=None, backend_name='test_backend'):
         agent_helpers.create_backend(
             database, backend_name, backend_messages.InitBody(**backend))
 
+    def create_test_group_template(self, name: str, group_template: Dict[str, Any]) -> None:
+        """Helper function to create a group template.
+
+        Args:
+            name: Name of the group template
+            group_template: The group template dict (must contain apiVersion, kind, metadata.name)
+        """
+        config_service.put_group_template(
+            name=name,
+            request=config_objects.PutGroupTemplateRequest(configs=group_template),
+            username='test@nvidia.com',
+        )
+
     def create_test_pool(self, pool_name='test_pool', description='test_description',
                          default_platform='test_platform', backend='test_backend',
-                         common_pod_template=None, enable_maintenance=False):
+                         common_pod_template=None, common_group_templates=None,
+                         enable_maintenance=False):
         """Helper function to create a test pool with configurable parameters.
 
         Args:
@@ -150,6 +165,7 @@ def create_test_pool(self, pool_name='test_pool', description='test_description'
             default_platform: Default platform for the pool
             backend: Backend for the pool
             common_pod_template: List of pod templates to use (defaults to None)
+            common_group_templates: List of group template names to use (defaults to None)
             enable_maintenance: Whether maintenance mode is enabled
 
         Returns:
@@ -169,6 +185,9 @@ def create_test_pool(self, pool_name='test_pool', description='test_description'
         if common_pod_template:
             pool_config['common_pod_template'] = common_pod_template
 
+        if common_group_templates:
+            pool_config['common_group_templates'] = common_group_templates
+
         config_service.put_pool(
             name=pool_name,
             request=config_objects.PutPoolRequest(