From 96903b5fcd501e40bac7d51960a1d899f7e64d6d Mon Sep 17 00:00:00 2001
From: Aaron Louie <ajlouie@gmail.com>
Date: Mon, 18 May 2020 20:59:38 -0400
Subject: [PATCH] Adds unit tests

---
 .../reference-files.component.spec.ts         | 102 +++++++++++++++++-
 .../reference-files.component.ts              |   3 +-
 2 files changed, 101 insertions(+), 4 deletions(-)

diff --git a/src/app/reference-files/reference-files.component.spec.ts b/src/app/reference-files/reference-files.component.spec.ts
index 086f86e..8e16284 100644
--- a/src/app/reference-files/reference-files.component.spec.ts
+++ b/src/app/reference-files/reference-files.component.spec.ts
@@ -1,14 +1,69 @@
+import {HttpHeaders} from '@angular/common/http';
+import {HttpClientTestingModule, HttpTestingController} from '@angular/common/http/testing';
 import { async, ComponentFixture, TestBed } from '@angular/core/testing';
-
+import {MAT_DIALOG_DATA, MatDialogModule, MatDialogRef} from '@angular/material/dialog';
+import {MatIconModule} from '@angular/material/icon';
+import {MatSnackBarModule} from '@angular/material/snack-bar';
+import {BrowserDynamicTestingModule} from '@angular/platform-browser-dynamic/testing';
+import {BrowserAnimationsModule} from '@angular/platform-browser/animations';
+import {of} from 'rxjs';
+import {ApiService, FileMeta, FileType, MockEnvironment, mockFileMetaReference0} from 'sartography-workflow-lib';
+import {OpenFileDialogComponent} from '../_dialogs/open-file-dialog/open-file-dialog.component';
+import * as FileSaver from 'file-saver';
 import { ReferenceFilesComponent } from './reference-files.component';
+import createClone from 'rfdc';
 
 describe('ReferenceFilesComponent', () => {
+  let httpMock: HttpTestingController;
   let component: ReferenceFilesComponent;
   let fixture: ComponentFixture<ReferenceFilesComponent>;
 
+  // Mock file and response headers
+  const mockDocMeta: FileMeta = createClone()(mockFileMetaReference0);
+  mockDocMeta.type = FileType.XLSX;
+
+  const expectedFile = new File([], mockDocMeta.name, {
+    type: mockDocMeta.content_type,
+    lastModified: mockDocMeta.file.lastModified
+  });
+
+  const mockHeaders = new HttpHeaders()
+    .append('last-modified', expectedFile.lastModified.toString())
+    .append('content-type', mockDocMeta.content_type);
+
+  const mockArrayBuffer = new ArrayBuffer(8);
+
   beforeEach(async(() => {
     TestBed.configureTestingModule({
-      declarations: [ ReferenceFilesComponent ]
+      imports: [
+        BrowserAnimationsModule,
+        HttpClientTestingModule,
+        MatDialogModule,
+        MatIconModule,
+        MatSnackBarModule,
+      ],
+      declarations: [
+        OpenFileDialogComponent,
+        ReferenceFilesComponent,
+      ],
+      providers: [
+        ApiService,
+        {provide: 'APP_ENVIRONMENT', useClass: MockEnvironment},
+        {
+          provide: MatDialogRef,
+          useValue: {
+            close: (dialogResult: any) => {
+            }
+          }
+        },
+        {provide: MAT_DIALOG_DATA, useValue: []},
+      ]
+    }).overrideModule(BrowserDynamicTestingModule, {
+      set: {
+        entryComponents: [
+          OpenFileDialogComponent,
+        ]
+      }
     })
     .compileComponents();
   }));
@@ -16,10 +71,53 @@ describe('ReferenceFilesComponent', () => {
   beforeEach(() => {
     fixture = TestBed.createComponent(ReferenceFilesComponent);
     component = fixture.componentInstance;
+    httpMock = TestBed.inject(HttpTestingController);
     fixture.detectChanges();
+
+    const req = httpMock.expectOne('apiRoot/reference_file');
+    expect(req.request.method).toEqual('GET');
+    req.flush([mockFileMetaReference0]);
+    expect(component.referenceFiles.length).toEqual(1);
+  });
+
+  afterEach(() => {
+    httpMock.verify();
+    fixture.destroy();
   });
 
   it('should create', () => {
     expect(component).toBeTruthy();
   });
+
+  it('should update existing file from file dialog', () => {
+    const openDialogSpy = spyOn(component.dialog, 'open')
+      .and.returnValue({afterClosed: () => of({fileMetaId: mockFileMetaReference0.id, file: mockFileMetaReference0.file})} as any);
+    const _loadReferenceFilesSpy = spyOn((component as any), '_loadReferenceFiles').and.stub();
+
+    component.openFileDialog(mockFileMetaReference0);
+
+    const fReq = httpMock.expectOne(`apiRoot/reference_file/${mockDocMeta.name}`);
+    expect(fReq.request.method).toEqual('GET');
+    fReq.flush(mockArrayBuffer, {headers: mockHeaders});
+
+    const updateReq = httpMock.expectOne(`apiRoot/reference_file/${mockFileMetaReference0.name}`);
+    expect(updateReq.request.method).toEqual('PUT');
+    updateReq.flush(mockArrayBuffer, {headers: mockHeaders});
+
+    expect(openDialogSpy).toHaveBeenCalled();
+    expect(_loadReferenceFilesSpy).toHaveBeenCalled();
+  });
+
+  it('should save file', () => {
+    const fileSaverSpy = spyOn(FileSaver, 'saveAs').and.stub();
+
+    component.downloadFile(mockDocMeta);
+
+    const fReq = httpMock.expectOne(`apiRoot/reference_file/${mockDocMeta.name}`);
+    expect(fReq.request.method).toEqual('GET');
+    fReq.flush(mockArrayBuffer, {headers: mockHeaders});
+
+    expect(fileSaverSpy).toHaveBeenCalled();
+  });
+
 });
diff --git a/src/app/reference-files/reference-files.component.ts b/src/app/reference-files/reference-files.component.ts
index 5304b38..0235461 100644
--- a/src/app/reference-files/reference-files.component.ts
+++ b/src/app/reference-files/reference-files.component.ts
@@ -44,13 +44,12 @@ export class ReferenceFilesComponent implements OnInit {
 
       dialogRef.afterClosed().subscribe((data: OpenFileDialogData) => {
         if (data && data.file) {
-          console.log('data', data);
           this.apiService.updateReferenceFile(fm.name, data.file).subscribe(() => {
+          this.snackBar.open(`Updated file ${fm.name}.`, 'Ok', {duration: 3000});
             this._loadReferenceFiles();
           });
         }
       });
-
     });
   }