chore(state): add test cases for rxState options

michaelbe812 · michaelbe812 · commit 56d8c34ee6f5 · 2025-01-09T20:03:22.000+01:00
diff --git a/libs/state/spec/rx-state.spec.ts b/libs/state/spec/rx-state.spec.ts
@@ -1,4 +1,9 @@
-import { Component, isSignal, signal } from '@angular/core';
+import {
+  Component,
+  EnvironmentInjector,
+  isSignal,
+  signal,
+} from '@angular/core';
 import { fakeAsync, TestBed, tick } from '@angular/core/testing';
 import { selectSlice } from '@rx-angular/state/selections';
 import {
@@ -395,6 +400,24 @@ describe(rxState, () => {
       }).toThrowError('readOnlyState.connect is not a function');
     });
   });
+
+  describe('without injection context', () => {
+    it('should throw when invoked without injection context', () => {
+      const { rxState } = setUpWithoutInjectionContext();
+      expect(() => rxState(() => {}, {})).toThrow(
+        /rxState\(\) can only be used within an injection context/,
+      );
+    });
+
+    it('should be able to use a custom Injector', () => {
+      const { rxState } = setUpWithoutInjectionContext();
+
+      const envInjector = TestBed.inject(EnvironmentInjector);
+
+      const state = rxState(() => {}, { injector: envInjector });
+      expect(state).toBeDefined();
+    });
+  });
 });
 
 type ITestComponent<State extends object> = {
@@ -426,3 +449,12 @@ function setupComponent<State extends { count: number }>(
     component: fixture.componentInstance,
   };
 }
+
+function setUpWithoutInjectionContext() {
+  return {
+    rxState<T extends object>(...args: Parameters<typeof rxState<T>>) {
+      const state = rxState(...args);
+      return state;
+    },
+  };
+}
