From 3c8c425843985d2bd461690aed876dfe0d850279 Mon Sep 17 00:00:00 2001
From: Yura Bondarenko <yura.bondarenko@atmire.com>
Date: Fri, 17 Dec 2021 17:05:53 +0100
Subject: [PATCH] 86016: Add unit tests for correlationId reducer & service

---
 .../correlation-id.reducer.spec.ts            | 23 +++++
 .../correlation-id.service.spec.ts            | 83 +++++++++++++++++++
 2 files changed, 106 insertions(+)
 create mode 100644 src/app/correlation-id/correlation-id.reducer.spec.ts
 create mode 100644 src/app/correlation-id/correlation-id.service.spec.ts

diff --git a/src/app/correlation-id/correlation-id.reducer.spec.ts b/src/app/correlation-id/correlation-id.reducer.spec.ts
new file mode 100644
index 0000000000..c784def1d9
--- /dev/null
+++ b/src/app/correlation-id/correlation-id.reducer.spec.ts
@@ -0,0 +1,23 @@
+import { correlationIdReducer } from './correlation-id.reducer';
+import { SetCorrelationIdAction } from './correlation-id.actions';
+
+describe('correlationIdReducer', () => {
+  it('should set the correlatinId with SET action', () => {
+    const initialState = null;
+    const currentState = correlationIdReducer(initialState, new SetCorrelationIdAction('new ID'));
+
+    expect(currentState).toBe('new ID');
+  });
+
+  it('should leave correlatinId unchanged otherwise', () => {
+    const initialState = null;
+
+    let currentState = correlationIdReducer(initialState, { type: 'unknown' } as any);
+    expect(currentState).toBe(null);
+
+    currentState = correlationIdReducer(currentState, new SetCorrelationIdAction('new ID'));
+    currentState = correlationIdReducer(currentState, { type: 'unknown' } as any);
+
+    expect(currentState).toBe('new ID');
+  });
+});
diff --git a/src/app/correlation-id/correlation-id.service.spec.ts b/src/app/correlation-id/correlation-id.service.spec.ts
new file mode 100644
index 0000000000..003924c123
--- /dev/null
+++ b/src/app/correlation-id/correlation-id.service.spec.ts
@@ -0,0 +1,83 @@
+import { CorrelationIdService } from './correlation-id.service';
+import { CookieServiceMock } from '../shared/mocks/cookie.service.mock';
+import { UUIDService } from '../core/shared/uuid.service';
+import { MockStore, provideMockStore } from '@ngrx/store/testing';
+import { TestBed } from '@angular/core/testing';
+import { Store, StoreModule } from '@ngrx/store';
+import { appReducers, AppState, storeModuleConfig } from '../app.reducer';
+import { SetCorrelationIdAction } from './correlation-id.actions';
+
+fdescribe('CorrelationIdService', () => {
+  let service: CorrelationIdService;
+
+  let cookieService;
+  let uuidService;
+  let store;
+
+  beforeEach(async () => {
+    await TestBed.configureTestingModule({
+      imports: [
+        StoreModule.forRoot(appReducers, storeModuleConfig),
+      ],
+    }).compileComponents();
+  });
+
+  beforeEach(() => {
+    cookieService = new CookieServiceMock();
+    uuidService = new UUIDService();
+    store = TestBed.inject(Store) as MockStore<AppState>;
+    service = new CorrelationIdService(cookieService, uuidService, store);
+  });
+
+  describe('getCorrelationId', () => {
+    it('should get from from store', () => {
+      expect(service.getCorrelationId()).toBe(null);
+      store.dispatch(new SetCorrelationIdAction('some value'));
+      expect(service.getCorrelationId()).toBe('some value');
+    });
+  });
+
+
+  describe('initCorrelationId', () => {
+    const cookieCID = 'cookie CID';
+    const storeCID = 'store CID';
+
+    it('should set cookie and store values to a newly generated value if neither ex', () => {
+      service.initCorrelationId();
+
+      expect(cookieService.get('CORRELATION-ID')).toBeTruthy();
+      expect(service.getCorrelationId()).toBeTruthy();
+      expect(cookieService.get('CORRELATION-ID')).toEqual(service.getCorrelationId());
+    });
+
+    it('should set store value to cookie value if present', () => {
+      expect(service.getCorrelationId()).toBe(null);
+
+      cookieService.set('CORRELATION-ID', cookieCID);
+
+      service.initCorrelationId();
+
+      expect(cookieService.get('CORRELATION-ID')).toBe(cookieCID);
+      expect(service.getCorrelationId()).toBe(cookieCID);
+    });
+
+    it('should set cookie value to store value if present', () => {
+      store.dispatch(new SetCorrelationIdAction(storeCID));
+
+      service.initCorrelationId();
+
+      expect(cookieService.get('CORRELATION-ID')).toBe(storeCID);
+      expect(service.getCorrelationId()).toBe(storeCID);
+    });
+
+    it('should set store value to cookie value if both are present', () => {
+      cookieService.set('CORRELATION-ID', cookieCID);
+      store.dispatch(new SetCorrelationIdAction(storeCID));
+
+      service.initCorrelationId();
+
+      expect(cookieService.get('CORRELATION-ID')).toBe(cookieCID);
+      expect(service.getCorrelationId()).toBe(cookieCID);
+    });
+  });
+});