Fixes unit test for UsageComponent

Author: vbakke

src/app/component/usage/usage.component.html

@@ -1,5 +1,5 @@
 <app-top-header section="Usage"></app-top-header>
 <app-readme-to-html
-    class="usage-{{ page }}"
-    MDFile="./assets/Markdown Files/{{ page }}.md">
+  class="usage-{{ page }}"
+  MDFile="./assets/Markdown Files/{{ page }}.md">
 </app-readme-to-html>

src/app/component/usage/usage.component.spec.ts

@@ -1,6 +1,8 @@
 import { ComponentFixture, TestBed } from '@angular/core/testing';
 
 import { UsageComponent } from './usage.component';
+import { ActivatedRoute } from '@angular/router';
+import { of } from 'rxjs';
 
 describe('UsageComponent', () => {
   let component: UsageComponent;
@@ -12,13 +14,28 @@ describe('UsageComponent', () => {
     }).compileComponents();
   });
 
-  beforeEach(() => {
+  it('should create', () => {
+    TestBed.overrideProvider(ActivatedRoute, {
+      useValue: { params: of({}) },
+    });
+
     fixture = TestBed.createComponent(UsageComponent);
     component = fixture.componentInstance;
     fixture.detectChanges();
-  });
 
-  it('should create', () => {
     expect(component).toBeTruthy();
+    expect(component.page).toBe('USAGE');
+  });
+
+  it('should load page', () => {
+    TestBed.overrideProvider(ActivatedRoute, {
+      useValue: { params: of({ page: 'test-page' }) },
+    });
+
+    fixture = TestBed.createComponent(UsageComponent);
+    component = fixture.componentInstance;
+    fixture.detectChanges();
+
+    expect(component.page).toBe('test-page');
   });
 });

src/app/component/usage/usage.component.ts

@@ -11,11 +11,14 @@ export class UsageComponent implements OnInit {
   constructor(private route: ActivatedRoute) {}
 
   ngOnInit() {
-    this.route.params.subscribe(params => {
-      let page = params['page'];
-      if (page.match(/^[\w.-]+$/)) {
-        this.page = page;
-      }
-    });
+    if (this.route && this.route.params) {
+      this.route.params.subscribe(params => {
+        let page = params['page'];
+        // CWE-79 - sanitize input
+        if (page.match(/^[\w.-]+$/)) {
+          this.page = page;
+        }
+      });
+    }
   }
 }