Fixing tests

plotly · Jan 8, 2019 · 385673b · 385673b
1 parent 2fb1397
commit 385673b
Showing 1 changed file with 25 additions and 10 deletions.
diff --git a/src/app/plotly/plotly.service.spec.ts b/src/app/plotly/plotly.service.spec.ts
@@ -32,20 +32,35 @@ describe('PlotlyService', () => {
         expect(service.getPlotly()).toBe(Plotlyjs);
     }));
 
-    it('should call plotly methods', inject([PlotlyService], (service: PlotlyService) => {
+    it('should call plotly.newPlot method', inject([PlotlyService], (service: PlotlyService) => {
         const plotly = service.getPlotly();
-        const methods: (keyof PlotlyService)[] = ['plot', 'update', 'newPlot'];
-        methods.forEach(methodName => {
-            spyOn(plotly, methodName).and.returnValue(new Promise(() => {}));
 
-            (service as any)[methodName]('one' as any, 'two' as any, 'three' as any, 'four' as any);
-            expect(plotly[methodName]).toHaveBeenCalledWith('one', 'two', 'three', 'four');
-        });
+        spyOn(plotly, 'newPlot').and.returnValue(new Promise(() => {}));
+        service.newPlot('one' as any, 'two' as any, 'three' as any, 'four' as any);
+        expect(plotly.newPlot).toHaveBeenCalledWith('one', 'two', 'three', 'four');
+    }));
 
-        spyOn(plotly.Plots, 'resize');
-        service.resize('one' as any);
-        expect(plotly.Plots.resize).toHaveBeenCalledWith('one');
+    it('should call plotly.plot method', inject([PlotlyService], (service: PlotlyService) => {
+        const plotly = service.getPlotly();
+
+        spyOn(plotly, 'plot').and.returnValue(new Promise(() => {}));
+        service.plot('one' as any, 'two' as any, 'three' as any, 'four' as any);
+        expect(plotly.plot).toHaveBeenCalledWith('one', 'two', 'three', 'four');
+    }));
+
+    it('should call plotly.update method', inject([PlotlyService], (service: PlotlyService) => {
+        const plotly = service.getPlotly();
+
+        spyOn(plotly, 'react').and.returnValue(new Promise(() => {}));
+        service.update('one' as any, 'two' as any, 'three' as any, 'four' as any);
+        expect(plotly.react).toHaveBeenCalledWith('one', 'two', 'three', 'four');
     }));
 
+    it('should call plotly.Plots.resize method', inject([PlotlyService], (service: PlotlyService) => {
+        const plotly = service.getPlotly();
 
+        spyOn(plotly.Plots, 'resize').and.returnValue(new Promise(() => {}));
+        service.resize('one' as any);
+        expect(plotly.Plots.resize).toHaveBeenCalledWith('one');
+    }));
 });