test: add tests for getControl methods

urbica · Apr 8, 2019 · ab7e852 · ab7e852
1 parent 7bcd8d7
commit ab7e852
Show file tree

Hide file tree

Showing 7 changed files with 29 additions and 7 deletions.
diff --git a/src/__mocks__/mapbox-gl.js b/src/__mocks__/mapbox-gl.js
@@ -137,6 +137,10 @@ function GeolocateControl() {
   return this;
 }
 
+GeolocateControl.prototype.on = function on(listener, fn) {
+  fn({ target: this });
+};
+
 function NavigationControl() {
   return this;
 }

diff --git a/src/components/AttributionControl/index.test.js b/src/components/AttributionControl/index.test.js
@@ -15,7 +15,9 @@ test('AttributionControl#render', () => {
     </MapGL>
   );
 
-  expect(wrapper.find('AttributionControl').exists()).toBe(true);
+  const control = wrapper.find('AttributionControl');
+  expect(control.exists()).toBe(true);
+  expect(control.instance().getControl()).toBeTruthy();
 
   wrapper.unmount();
   expect(wrapper.find('AttributionControl').exists()).toBe(false);

diff --git a/src/components/FullscreenControl/index.test.js b/src/components/FullscreenControl/index.test.js
@@ -11,7 +11,9 @@ test('render', () => {
     </MapGL>
   );
 
-  expect(wrapper.find('FullscreenControl').exists()).toBe(true);
+  const control = wrapper.find('FullscreenControl');
+  expect(control.exists()).toBe(true);
+  expect(control.instance().getControl()).toBeTruthy();
 
   wrapper.unmount();
   expect(wrapper.find('FullscreenControl').exists()).toBe(false);

diff --git a/src/components/GeolocateControl/index.test.js b/src/components/GeolocateControl/index.test.js
@@ -7,11 +7,19 @@ import MapGL, { GeolocateControl } from '../..';
 test('render', () => {
   const wrapper = mount(
     <MapGL latitude={0} longitude={0} zoom={0}>
-      <GeolocateControl position="top-right" />
+      <GeolocateControl
+        position="top-right"
+        onError={jest.fn()}
+        onGeolocate={jest.fn()}
+        onTrackUserLocationEnd={jest.fn()}
+        onTrackUserLocationStart={jest.fn()}
+      />
     </MapGL>
   );
 
-  expect(wrapper.find('GeolocateControl').exists()).toBe(true);
+  const control = wrapper.find('GeolocateControl');
+  expect(control.exists()).toBe(true);
+  expect(control.instance().getControl()).toBeTruthy();
 
   wrapper.unmount();
   expect(wrapper.find('GeolocateControl').exists()).toBe(false);

diff --git a/src/components/LanguageControl/index.test.js b/src/components/LanguageControl/index.test.js
@@ -11,7 +11,9 @@ test('render', () => {
     </MapGL>
   );
 
-  expect(wrapper.find('LanguageControl').exists()).toBe(true);
+  const control = wrapper.find('LanguageControl');
+  expect(control.exists()).toBe(true);
+  expect(control.instance().getControl()).toBeTruthy();
 
   wrapper.unmount();
   expect(wrapper.find('LanguageControl').exists()).toBe(false);

diff --git a/src/components/NavigationControl/index.test.js b/src/components/NavigationControl/index.test.js
@@ -11,7 +11,9 @@ test('render', () => {
     </MapGL>
   );
 
-  expect(wrapper.find('NavigationControl').exists()).toBe(true);
+  const control = wrapper.find('NavigationControl');
+  expect(control.exists()).toBe(true);
+  expect(control.instance().getControl()).toBeTruthy();
 
   wrapper.unmount();
   expect(wrapper.find('NavigationControl').exists()).toBe(false);

diff --git a/src/components/ScaleControl/index.test.js b/src/components/ScaleControl/index.test.js
@@ -11,7 +11,9 @@ test('render', () => {
     </MapGL>
   );
 
-  expect(wrapper.find('ScaleControl').exists()).toBe(true);
+  const control = wrapper.find('ScaleControl');
+  expect(control.exists()).toBe(true);
+  expect(control.instance().getControl()).toBeTruthy();
 
   wrapper.unmount();
   expect(wrapper.find('ScaleControl').exists()).toBe(false);