MySegmentsCacheInLocal.spec.ts

import { MySegmentsCacheInLocal } from '../MySegmentsCacheInLocal';
import { KeyBuilderCS, myLargeSegmentsKeyBuilder } from '../../KeyBuilderCS';
import { loggerMock } from '../../../logger/__tests__/sdkLogger.mock';
import { storages, PREFIX } from './wrapper.mock';
import { IMySegmentsResponse } from '../../../dtos/types';

test.each(storages)('SEGMENT CACHE / in LocalStorage', (storage) => {
  const caches = [
    new MySegmentsCacheInLocal(loggerMock, new KeyBuilderCS(PREFIX, 'user'), storage),
    new MySegmentsCacheInLocal(loggerMock, myLargeSegmentsKeyBuilder(PREFIX, 'user'), storage)
  ];

  caches.forEach(cache => {
    cache.clear();

    expect(cache.resetSegments({ k: [{ n: 'mocked-segment' }, { n: 'mocked-segment-2' }], cn: 123 })).toBe(true);
    expect(cache.getChangeNumber()).toBe(123);
    expect(cache.resetSegments({ k: [{ n: 'mocked-segment' }, { n: 'mocked-segment-2' }] })).toBe(false);
    expect(cache.getChangeNumber()).toBe(-1);

    expect(cache.isInSegment('mocked-segment')).toBe(true);
    expect(cache.getRegisteredSegments()).toEqual(['mocked-segment', 'mocked-segment-2']);
    expect(cache.getKeysCount()).toBe(1);
  });

  caches.forEach(cache => {
    cache.resetSegments({
      added: [],
      removed: ['mocked-segment']
    } as IMySegmentsResponse);

    expect(cache.isInSegment('mocked-segment')).toBe(false);
    expect(cache.getRegisteredSegments()).toEqual(['mocked-segment-2']);
    expect(cache.getKeysCount()).toBe(1);
  });

  expect(storage.getItem(PREFIX + '.user.segment.mocked-segment-2')).toBe('1');
  expect(storage.getItem(PREFIX + '.user.segment.mocked-segment')).toBe(null);
  expect(storage.getItem(PREFIX + '.user.largeSegment.mocked-segment-2')).toBe('1');
  expect(storage.getItem(PREFIX + '.user.largeSegment.mocked-segment')).toBe(null);
});

test('SEGMENT CACHE / Special case: localStorage failure should not throw an exception', () => {
  const cache = new MySegmentsCacheInLocal(loggerMock, new KeyBuilderCS(PREFIX, 'user2'), localStorage);

  // mock localStorage failure
  const setItemSpy = jest.spyOn(localStorage, 'setItem').mockImplementation(() => { throw new Error('localStorage failure'); });
  setItemSpy.mockClear();

  expect(cache.resetSegments({ k: [{ n: 'mocked-segment' }, { n: 'mocked-segment-2' }], cn: 123 })).toBe(false);
  expect(setItemSpy).toHaveBeenCalledTimes(1);

  setItemSpy.mockRestore();
});