Include preview in tests

wylieconlon · wylieconlon · commit 7a9c5ce66ecc · 2020-03-20T15:23:27.000-04:00
diff --git a/x-pack/legacy/plugins/lens/public/editor_frame_service/editor_frame/layer_actions.test.ts b/x-pack/legacy/plugins/lens/public/editor_frame_service/editor_frame/layer_actions.test.ts
@@ -28,19 +28,21 @@ function createTestArgs(initialLayerIds: string[]) {
     appendLayer: (layerIds: unknown, layerId: string) => [...(layerIds as string[]), layerId],
   };
 
+  const datasourceStates = {
+    ds1: {
+      isLoading: false,
+      state: initialLayerIds.slice(0, 1),
+    },
+    ds2: {
+      isLoading: false,
+      state: initialLayerIds.slice(1),
+    },
+  };
+
   return {
     state: {
       activeDatasourceId: 'ds1',
-      datasourceStates: {
-        ds1: {
-          isLoading: false,
-          state: initialLayerIds.slice(0, 1),
-        },
-        ds2: {
-          isLoading: false,
-          state: initialLayerIds.slice(1),
-        },
-      },
+      datasourceStates,
       title: 'foo',
       visualization: {
         activeId: 'vis1',
@@ -53,6 +55,13 @@ function createTestArgs(initialLayerIds: string[]) {
       ds2: testDatasource('ds2'),
     },
     trackUiEvent,
+    stagedPreview: {
+      visualization: {
+        activeId: 'vis1',
+        state: initialLayerIds,
+      },
+      datasourceStates,
+    },
   };
 }
 
@@ -90,6 +99,7 @@ describe('removeLayer', () => {
     expect(newState.visualization.state).toEqual(['layer2']);
     expect(newState.datasourceStates.ds1.state).toEqual([]);
     expect(newState.datasourceStates.ds2.state).toEqual(['layer2']);
+    expect(newState.stagedPreview).not.toBeDefined();
     expect(trackUiEvent).toHaveBeenCalledWith('layer_removed');
   });
 });
@@ -111,6 +121,7 @@ describe('appendLayer', () => {
     expect(newState.visualization.state).toEqual(['layer1', 'layer2', 'foo']);
     expect(newState.datasourceStates.ds1.state).toEqual(['layer1', 'foo']);
     expect(newState.datasourceStates.ds2.state).toEqual(['layer2']);
+    expect(newState.stagedPreview).not.toBeDefined();
     expect(trackUiEvent).toHaveBeenCalledWith('layer_added');
   });
 });
diff --git a/x-pack/legacy/plugins/lens/public/editor_frame_service/editor_frame/layer_actions.ts b/x-pack/legacy/plugins/lens/public/editor_frame_service/editor_frame/layer_actions.ts
@@ -85,5 +85,6 @@ export function appendLayer({
       ...state.visualization,
       state: activeVisualization.appendLayer(state.visualization.state, layerId),
     },
+    stagedPreview: undefined,
   };
 }

Original file line number	Diff line number	Diff line change
`@@ -85,5 +85,6 @@ export function appendLayer({`
`85`	`85`	`...state.visualization,`
`86`	`86`	`state: activeVisualization.appendLayer(state.visualization.state, layerId),`
`87`	`87`	`},`
	`88`	`+ stagedPreview: undefined,`
`88`	`89`	`};`
`89`	`90`	`}`