add unit test

elastic · Mar 9, 2021 · 62743ff · 62743ff
1 parent e919bf2
commit 62743ff
Showing 1 changed file with 52 additions and 7 deletions.
diff --git a/x-pack/plugins/maps/public/classes/layers/tiled_vector_layer/tiled_vector_layer.test.tsx b/x-pack/plugins/maps/public/classes/layers/tiled_vector_layer/tiled_vector_layer.test.tsx
@@ -37,7 +37,8 @@ const defaultConfig = {
 
 function createLayer(
   layerOptions: Partial<VectorLayerDescriptor> = {},
-  sourceOptions: Partial<TiledSingleLayerVectorSourceDescriptor> = {}
+  sourceOptions: Partial<TiledSingleLayerVectorSourceDescriptor> = {},
+  isTimeAware: boolean = false
 ): TiledVectorLayer {
   const sourceDescriptor: TiledSingleLayerVectorSourceDescriptor = {
     type: SOURCE_TYPES.MVT_SINGLE_LAYER,
@@ -47,6 +48,14 @@ function createLayer(
     ...sourceOptions,
   };
   const mvtSource = new MVTSingleLayerVectorSource(sourceDescriptor);
+  if (isTimeAware) {
+    mvtSource.isTimeAware = async () => {
+      return true;
+    };
+    mvtSource.getApplyGlobalTime = () => {
+      return true;
+    };
+  }
 
   const defaultLayerOptions = {
     ...layerOptions,
@@ -130,13 +139,49 @@ describe('syncData', () => {
     sinon.assert.notCalled(syncContext2.stopLoading);
   });
 
+  it('Should resync when changes to syncContext', async () => {
+    const layer1: TiledVectorLayer = createLayer({}, {}, true);
+    const syncContext1 = new MockSyncContext({
+      dataFilters: {
+        timeFilters: {
+          from: 'now',
+          to: '15m',
+          mode: 'relative',
+        },
+      },
+    });
+
+    await layer1.syncData(syncContext1);
+
+    const dataRequestDescriptor: DataRequestDescriptor = {
+      data: { ...defaultConfig },
+      dataId: 'source',
+    };
+    const layer2: TiledVectorLayer = createLayer(
+      {
+        __dataRequests: [dataRequestDescriptor],
+      },
+      {},
+      true
+    );
+    const syncContext2 = new MockSyncContext({
+      dataFilters: {
+        timeFilters: {
+          from: 'now',
+          to: '30m',
+          mode: 'relative',
+        },
+      },
+    });
+    await layer2.syncData(syncContext2);
+    // @ts-expect-error
+    sinon.assert.calledOnce(syncContext2.startLoading);
+    // @ts-expect-error
+    sinon.assert.calledOnce(syncContext2.stopLoading);
+  });
+
   describe('Should resync when changes to source params: ', () => {
-    [
-      { layerName: 'barfoo' },
-      { urlTemplate: 'https://sub.example.com/{z}/{x}/{y}.pbf' },
-      { minSourceZoom: 1 },
-      { maxSourceZoom: 12 },
-    ].forEach((changes) => {
+    [{ layerName: 'barfoo' }, { minSourceZoom: 1 }, { maxSourceZoom: 12 }].forEach((changes) => {
       it(`change in ${Object.keys(changes).join(',')}`, async () => {
         const layer1: TiledVectorLayer = createLayer({}, {});
         const syncContext1 = new MockSyncContext({ dataFilters: {} });