test: add tests for getControl methods

Author: stepankuzmin

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;
 }

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);

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);

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);

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);

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);

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);