chore: Add support for using injectionTokens in global overrides

Signed-off-by: Sebastian Malton <sebastian@malton.name>

packages/utility-features/test-utils/src/get-global-override.ts

@@ -2,17 +2,30 @@
  * Copyright (c) OpenLens Authors. All rights reserved.
  * Licensed under MIT License. See LICENSE in root directory for more information.
  */
-import type { Injectable, Instantiate } from "@ogre-tools/injectable";
+import type { Injectable, InjectionToken, Instantiate } from "@ogre-tools/injectable";
 
-export interface GlobalOverride<Instance, Param> {
-  injectable: Injectable<Instance, unknown, Param>;
-  overridingInstantiate: Instantiate<Instance, Param>;
+export type GlobalOverride<
+  InjectionInstance extends InjectionTokenInstance,
+  InjectionTokenInstance,
+  InstantiationParam,
+> = ReturnType<typeof getGlobalOverride<InjectionInstance, InjectionTokenInstance, InstantiationParam>>;
+
+export function getGlobalOverride<
+  InjectionInstance extends InjectionTokenInstance,
+  InjectionTokenInstance,
+  InstantiationParam,
+>(
+  injectable:
+    | InjectionToken<InjectionInstance, InstantiationParam>
+    | Injectable<
+        InjectionInstance,
+        InjectionTokenInstance,
+        InstantiationParam
+      >,
+  overridingInstantiate: Instantiate<InjectionInstance, InstantiationParam>,
+) {
+  return {
+    injectable,
+    overridingInstantiate,
+  };
 }
-
-export const getGlobalOverride = <Instance, Param>(
-  injectable: Injectable<Instance, unknown, Param>,
-  overridingInstantiate: (typeof injectable)["instantiate"],
-) => ({
-  injectable,
-  overridingInstantiate,
-});