From af74d00b058a392a148f745d8e696d172a10144d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz=20Paczos?= <lukas.paczos@gmail.com>
Date: Wed, 3 Apr 2019 17:38:25 +0200
Subject: [android] add unit tests for #14171

---
 .../com/mapbox/mapboxsdk/maps/MapboxMapTest.kt     |  11 ++
 .../java/com/mapbox/mapboxsdk/maps/StyleTest.kt    | 112 +++++++++++++++++++--
 2 files changed, 114 insertions(+), 9 deletions(-)

diff --git a/platform/android/MapboxGLAndroidSDK/src/test/java/com/mapbox/mapboxsdk/maps/MapboxMapTest.kt b/platform/android/MapboxGLAndroidSDK/src/test/java/com/mapbox/mapboxsdk/maps/MapboxMapTest.kt
index 2d68612c70..dbb4bf1116 100644
--- a/platform/android/MapboxGLAndroidSDK/src/test/java/com/mapbox/mapboxsdk/maps/MapboxMapTest.kt
+++ b/platform/android/MapboxGLAndroidSDK/src/test/java/com/mapbox/mapboxsdk/maps/MapboxMapTest.kt
@@ -112,4 +112,15 @@ class MapboxMapTest {
     mapboxMap.onDestroy()
     verify { cameraChangeDispatcher.onDestroy() }
   }
+
+  @Test
+  fun testStyleClearedOnDestroy() {
+    val style = mockk<Style>(relaxed = true)
+    val builder = mockk<Style.Builder>(relaxed = true)
+    every { builder.build(nativeMapView) } returns style
+    mapboxMap.setStyle(builder)
+
+    mapboxMap.onDestroy()
+    verify(exactly = 1) { style.clear() }
+  }
 }
\ No newline at end of file
diff --git a/platform/android/MapboxGLAndroidSDK/src/test/java/com/mapbox/mapboxsdk/maps/StyleTest.kt b/platform/android/MapboxGLAndroidSDK/src/test/java/com/mapbox/mapboxsdk/maps/StyleTest.kt
index 855987b233..a845d23875 100644
--- a/platform/android/MapboxGLAndroidSDK/src/test/java/com/mapbox/mapboxsdk/maps/StyleTest.kt
+++ b/platform/android/MapboxGLAndroidSDK/src/test/java/com/mapbox/mapboxsdk/maps/StyleTest.kt
@@ -3,8 +3,10 @@ package com.mapbox.mapboxsdk.maps
 import android.graphics.Bitmap
 import android.graphics.drawable.ShapeDrawable
 import com.mapbox.mapboxsdk.constants.MapboxConstants
+import com.mapbox.mapboxsdk.style.layers.CannotAddLayerException
 import com.mapbox.mapboxsdk.style.layers.SymbolLayer
 import com.mapbox.mapboxsdk.style.layers.TransitionOptions
+import com.mapbox.mapboxsdk.style.sources.CannotAddSourceException
 import com.mapbox.mapboxsdk.style.sources.GeoJsonSource
 import io.mockk.every
 import io.mockk.mockk
@@ -25,16 +27,8 @@ class StyleTest {
 
     @Before
     fun setup() {
-        nativeMapView = mockk()
+        nativeMapView = mockk(relaxed = true)
         mapboxMap = MapboxMap(nativeMapView, null, null, null, null, null)
-        every { nativeMapView.styleUrl = any() } answers {}
-        every { nativeMapView.styleJson = any() } answers {}
-        every { nativeMapView.addLayerBelow(any(), any()) } answers {}
-        every { nativeMapView.addLayerAbove(any(), any()) } answers {}
-        every { nativeMapView.addLayerAt(any(), any()) } answers {}
-        every { nativeMapView.addSource(any()) } answers {}
-        every { nativeMapView.addImages(any()) } answers {}
-        every { nativeMapView.transitionOptions = any() } answers {}
         every { nativeMapView.isDestroyed } returns false
         mapboxMap.injectLocationComponent(spyk())
     }
@@ -314,4 +308,104 @@ class StyleTest {
         mapboxMap.notifyStyleLoaded()
         verify(exactly = 1) { nativeMapView.addImages(any()) }
     }
+
+    @Test
+    fun testSourceSkippedIfAdditionFails() {
+        val source1 = mockk<GeoJsonSource>(relaxed = true)
+        every { source1.id } returns "source1"
+        val source2 = mockk<GeoJsonSource>(relaxed = true)
+        every { source2.id } returns "source1" // same ID
+
+        val builder = Style.Builder().withSource(source1)
+        mapboxMap.setStyle(builder)
+        mapboxMap.notifyStyleLoaded()
+
+        every { nativeMapView.addSource(any()) } throws CannotAddSourceException("Duplicate ID")
+
+        try {
+            mapboxMap.style!!.addSource(source2)
+        } catch (ex: Exception) {
+            Assert.assertEquals("Source that failed to be added shouldn't be cached", source1, mapboxMap.style!!.getSource("source1"))
+        }
+    }
+
+    @Test
+    fun testLayerSkippedIfAdditionFails() {
+        val layer1 = mockk<SymbolLayer>(relaxed = true)
+        every { layer1.id } returns "layer1"
+        val layer2 = mockk<SymbolLayer>(relaxed = true)
+        every { layer2.id } returns "layer1" // same ID
+
+        val builder = Style.Builder().withLayer(layer1)
+        mapboxMap.setStyle(builder)
+        mapboxMap.notifyStyleLoaded()
+
+        every { nativeMapView.addLayer(any()) } throws CannotAddLayerException("Duplicate ID")
+
+        try {
+            mapboxMap.style!!.addLayer(layer2)
+        } catch (ex: Exception) {
+            Assert.assertEquals("Layer that failed to be added shouldn't be cached", layer1, mapboxMap.style!!.getLayer("layer1"))
+        }
+    }
+
+    @Test
+    fun testLayerSkippedIfAdditionBelowFails() {
+        val layer1 = mockk<SymbolLayer>(relaxed = true)
+        every { layer1.id } returns "layer1"
+        val layer2 = mockk<SymbolLayer>(relaxed = true)
+        every { layer2.id } returns "layer1" // same ID
+
+        val builder = Style.Builder().withLayer(layer1)
+        mapboxMap.setStyle(builder)
+        mapboxMap.notifyStyleLoaded()
+
+        every { nativeMapView.addLayerBelow(any(), "") } throws CannotAddLayerException("Duplicate ID")
+
+        try {
+            mapboxMap.style!!.addLayerBelow(layer2, "")
+        } catch (ex: Exception) {
+            Assert.assertEquals("Layer that failed to be added shouldn't be cached", layer1, mapboxMap.style!!.getLayer("layer1"))
+        }
+    }
+
+    @Test
+    fun testLayerSkippedIfAdditionAboveFails() {
+        val layer1 = mockk<SymbolLayer>(relaxed = true)
+        every { layer1.id } returns "layer1"
+        val layer2 = mockk<SymbolLayer>(relaxed = true)
+        every { layer2.id } returns "layer1" // same ID
+
+        val builder = Style.Builder().withLayer(layer1)
+        mapboxMap.setStyle(builder)
+        mapboxMap.notifyStyleLoaded()
+
+        every { nativeMapView.addLayerAbove(any(), "") } throws CannotAddLayerException("Duplicate ID")
+
+        try {
+            mapboxMap.style!!.addLayerAbove(layer2, "")
+        } catch (ex: Exception) {
+            Assert.assertEquals("Layer that failed to be added shouldn't be cached", layer1, mapboxMap.style!!.getLayer("layer1"))
+        }
+    }
+
+    @Test
+    fun testLayerSkippedIfAdditionAtFails() {
+        val layer1 = mockk<SymbolLayer>(relaxed = true)
+        every { layer1.id } returns "layer1"
+        val layer2 = mockk<SymbolLayer>(relaxed = true)
+        every { layer2.id } returns "layer1" // same ID
+
+        val builder = Style.Builder().withLayer(layer1)
+        mapboxMap.setStyle(builder)
+        mapboxMap.notifyStyleLoaded()
+
+        every { nativeMapView.addLayerAt(any(), 5) } throws CannotAddLayerException("Duplicate ID")
+
+        try {
+            mapboxMap.style!!.addLayerAt(layer2, 5)
+        } catch (ex: Exception) {
+            Assert.assertEquals("Layer that failed to be added shouldn't be cached", layer1, mapboxMap.style!!.getLayer("layer1"))
+        }
+    }
 }
-- 
cgit v1.2.1