1 files changed, 72 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/frontend/src/app/shared/services/feature-toggles-guard.service.spec.ts b/src/pybind/mgr/dashboard/frontend/src/app/shared/services/feature-toggles-guard.service.spec.ts
new file mode 100644
index 000000000..883139986
--- /dev/null
+++ b/src/pybind/mgr/dashboard/frontend/src/app/shared/services/feature-toggles-guard.service.spec.ts
@@ -0,0 +1,72 @@
+import { Component, NgZone } from '@angular/core';
+import { fakeAsync, TestBed, tick } from '@angular/core/testing';
+import { ActivatedRouteSnapshot, Router, Routes } from '@angular/router';
+import { RouterTestingModule } from '@angular/router/testing';
+
+import { of as observableOf } from 'rxjs';
+
+import { DashboardNotFoundError } from '~/app/core/error/error';
+import { configureTestBed } from '~/testing/unit-test-helper';
+import { FeatureTogglesGuardService } from './feature-toggles-guard.service';
+import { FeatureTogglesService } from './feature-toggles.service';
+
+describe('FeatureTogglesGuardService', () => {
+  let service: FeatureTogglesGuardService;
+  let fakeFeatureTogglesService: FeatureTogglesService;
+  let router: Router;
+  let ngZone: NgZone;
+
+  @Component({ selector: 'cd-cephfs', template: '' })
+  class CephfsComponent {}
+
+  @Component({ selector: 'cd-404', template: '' })
+  class NotFoundComponent {}
+
+  const routes: Routes = [
+    { path: 'cephfs', component: CephfsComponent },
+    { path: '404', component: NotFoundComponent }
+  ];
+
+  configureTestBed({
+    imports: [RouterTestingModule.withRoutes(routes)],
+    providers: [
+      { provide: FeatureTogglesService, useValue: { get: null } },
+      FeatureTogglesGuardService
+    ],
+    declarations: [CephfsComponent, NotFoundComponent]
+  });
+
+  beforeEach(() => {
+    service = TestBed.inject(FeatureTogglesGuardService);
+    fakeFeatureTogglesService = TestBed.inject(FeatureTogglesService);
+    ngZone = TestBed.inject(NgZone);
+    router = TestBed.inject(Router);
+  });
+
+  it('should be created', () => {
+    expect(service).toBeTruthy();
+  });
+
+  function testCanActivate(path: string, feature_toggles_map: object) {
+    let result: boolean;
+    spyOn(fakeFeatureTogglesService, 'get').and.returnValue(observableOf(feature_toggles_map));
+
+    ngZone.run(() => {
+      service
+        .canActivate(<ActivatedRouteSnapshot>{ routeConfig: { path: path } })
+        .subscribe((val) => (result = val));
+    });
+    tick();
+
+    return result;
+  }
+
+  it('should allow the feature if enabled', fakeAsync(() => {
+    expect(testCanActivate('cephfs', { cephfs: true })).toBe(true);
+    expect(router.url).toBe('/');
+  }));
+
+  it('should throw error if disable', fakeAsync(() => {
+    expect(() => testCanActivate('cephfs', { cephfs: false })).toThrowError(DashboardNotFoundError);
+  }));
+});