QA install script additions (#2762)

Author: rarescosma

scripts/qa/boilerplate.py

@@ -29,12 +29,12 @@ class AppsConfig:
 
     path: Path
 
-    def set(self, key: str, value: Jsonable) -> None:
+    def set(self, key: str, value: Jsonable, merge: bool = True) -> None:
         """Set a config key (with merge)"""
         _run(
             "yq",
             "-i",
-            f".{key} = .{key} * {json.dumps(value)}",
+            f".{key} = .{key} * {json.dumps(value)}" if merge else f".{key} = {json.dumps(value)}",
             self.path.resolve().as_posix(),
         )
 

scripts/qa/config.json.sample

@@ -52,5 +52,6 @@
     "apiServer": [],
     "nodes": [],
     "ingress": []
-  }
+  },
+  "networkPlugin": "set-me"
 }

scripts/qa/config.py

@@ -8,7 +8,18 @@
 import json
 import sys
 from contextlib import suppress
-from typing import Any, Optional, Type, TypedDict, Union, cast, get_args, get_origin, get_type_hints
+from typing import (
+    Any,
+    Literal,
+    Optional,
+    Type,
+    TypedDict,
+    Union,
+    cast,
+    get_args,
+    get_origin,
+    get_type_hints,
+)
 
 AwsSecrets = TypedDict(
     "AwsSecrets",
@@ -60,6 +71,8 @@
     total=False,
 )
 
+NetworkPlugin = Literal["calico", "cilium"]
+
 
 class Config(TypedDict):
     """Holds the parsed configuration"""
@@ -75,6 +88,8 @@ class Config(TypedDict):
     wcSubnets: Optional[Subnets]
     scSubnets: Optional[Subnets]
 
+    networkPlugin: Optional[NetworkPlugin]
+
 
 def dig(config: Config, path: str) -> str | int | float | bool | list | dict | None:
     """Dig a dotted expression out of the config"""
@@ -140,6 +155,9 @@ def get_default_value(f_type: Type) -> Any:
                 return set()
             if origin is tuple:
                 return ()
+            if origin is Literal:
+                non_none_types = get_non_none_types(f_type)
+                return get_default_value(type(non_none_types[0])) if non_none_types else None
 
         # Default fallback
         return None

scripts/qa/install_apps.py

@@ -48,6 +48,10 @@ def configure_apps(
 ) -> None:
     """Configure apps"""
 
+    # Configure network plugin
+    if (network_plugin := args.config["networkPlugin"]) is not None:
+        common_config.set("networkPlugin.type", network_plugin, merge=False)
+
     # Configure platform administrators
     common_config.set("clusterAdmin", {"users": [], "groups": [args.config["adminGroup"]]})
 
@@ -127,7 +131,7 @@ def configure_apps(
     )
 
     # Configure Falco BPF driver
-    common_config.set("falco.driver", {"kind": "modern-bpf"})
+    common_config.set("falco.driver", {"kind": "modern_ebpf"})
 
     # Configure Grafana with test requirements
     grafana_cfg = {