Fix jest test

elastic · Sep 4, 2019 · f7adb55 · f7adb55
1 parent b579701
commit f7adb55
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/src/legacy/core_plugins/data/public/timefilter/time_history.ts b/src/legacy/core_plugins/data/public/timefilter/time_history.ts
@@ -22,7 +22,7 @@ import { TimeRange } from 'src/plugins/data/public';
 import { PersistedLog } from 'ui/persisted_log';
 
 export class TimeHistory {
-  private history: PersistedLog;
+  private history: PersistedLog<TimeRange>;
 
   constructor() {
     const historyOptions = {

diff --git a/src/legacy/core_plugins/kibana/public/dashboard/dashboard_state.test.ts b/src/legacy/core_plugins/kibana/public/dashboard/dashboard_state.test.ts
@@ -27,6 +27,7 @@ import { TimeRange } from 'src/plugins/data/public';
 import { ViewMode } from '../../../embeddable_api/public/np_ready/public';
 
 import { dataPluginMock } from '../../../../core_plugins/data/public/mocks';
+import { InputTimeRange } from 'ui/timefilter';
 const dataSetupMock = dataPluginMock.createSetup();
 
 describe('DashboardState', function() {
@@ -36,11 +37,8 @@ describe('DashboardState', function() {
   let mockTime: TimeRange = { to: 'now', from: 'now-15m' };
   const mockTimefilter = dataSetupMock.timefilter!.timefilter;
 
-  mockTimefilter.setTime.mockImplementation((time: any) => {
-    mockTime = {
-      from: time.from.toISOString(),
-      to: time.to.toISOString(),
-    };
+  mockTimefilter.setTime.mockImplementation((time: InputTimeRange) => {
+    mockTime = time as TimeRange;
   });
   mockTimefilter.getTime.mockImplementation(() => {
     return mockTime;