Mock storage service in tests

ScottLogic · Aug 5, 2024 · 7645c27 · 7645c27
1 parent 7969e63
commit 7645c27
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/app/carbon-estimator-form/carbon-estimator-form.component.spec.ts b/src/app/carbon-estimator-form/carbon-estimator-form.component.spec.ts
@@ -1,6 +1,19 @@
 import { ComponentFixture, TestBed } from '@angular/core/testing';
 
 import { CarbonEstimatorFormComponent } from './carbon-estimator-form.component';
+import { StorageService } from '../services/storage.service';
+
+class MockStorageService {
+  storage: Record<string, string> = {};
+
+  get(key: string): string | null {
+    return this.storage[key] || null;
+  }
+
+  set(key: string, value: string): void {
+    this.storage[key] = value;
+  }
+}
 
 describe('CarbonEstimatorFormComponent', () => {
   let component: CarbonEstimatorFormComponent;
@@ -9,6 +22,7 @@ describe('CarbonEstimatorFormComponent', () => {
   beforeEach(async () => {
     await TestBed.configureTestingModule({
       imports: [CarbonEstimatorFormComponent],
+      providers: [{ provide: StorageService, useClass: MockStorageService }],
     }).compileComponents();
 
     fixture = TestBed.createComponent(CarbonEstimatorFormComponent);