Fix shouldComponentUpdate of Map component

CHANGELOG.md

@@ -7,6 +7,9 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 
 ## [Unreleased]
 
+### Fixed
+- Fix `shouldComponentUpdate` of `Map` component
+
 ## [2.1.0] - 2018-08-22
 
 ### Added

src/geolocation/Map.js

@@ -15,12 +15,13 @@ class Map extends Component {
 
   shouldComponentUpdate(prevProps) {
     const rulesChanged = prevProps.rules.country !== this.props.rules.country
+    const loadingGoogleChanged = prevProps.loadingGoogle !== this.props.loadingGoogle
     const geoCoordsChanged = this.isDifferentGeoCoords(
       this.getCoordinatesFromProps(prevProps),
       this.getCoordinatesFromProps(this.props),
     )
 
-    return geoCoordsChanged || rulesChanged
+    return geoCoordsChanged || rulesChanged || loadingGoogleChanged
   }
 
   componentDidUpdate() {

src/geolocation/Map.test.js

@@ -86,4 +86,16 @@ describe('Map', () => {
 
     expect(shouldUpdate).toBe(true)
   })
+
+  it('should re-render if loadingGoogle changed', () => {
+    const currentProps = shallowInstance.props
+    const currentState = shallowInstance.state
+
+    const shouldUpdate = shallowInstance.shouldComponentUpdate(
+      { ...currentProps, loadingGoogle: true },
+      currentState,
+    )
+
+    expect(shouldUpdate).toBe(true)
+  })
 })