From 9c6c09621cf06384179322433378c446e1985c56 Mon Sep 17 00:00:00 2001
From: Ankush Bose <46471379+bosankus@users.noreply.github.com>
Date: Sat, 24 Dec 2022 14:34:17 +0530
Subject: [PATCH] 5125: Fix LocationPickerActivityUnitTests (#5126)

* 5121: Fix RecentLanguagesAdapterUnitTest
- all tests are passing
- additionally removed 'context' variable from AdvanceQueryFragmentUnitTests.kt as it is not used, and not removed on previous PR #5119

* 5125: Fix LocationPickerActivityUnitTest
- all tests are passing
---
 .../locationpicker/LocationPickerActivityUnitTests.kt      | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/app/src/test/kotlin/fr/free/nrw/commons/locationpicker/LocationPickerActivityUnitTests.kt b/app/src/test/kotlin/fr/free/nrw/commons/locationpicker/LocationPickerActivityUnitTests.kt
index e010b26f4..461882c3b 100644
--- a/app/src/test/kotlin/fr/free/nrw/commons/locationpicker/LocationPickerActivityUnitTests.kt
+++ b/app/src/test/kotlin/fr/free/nrw/commons/locationpicker/LocationPickerActivityUnitTests.kt
@@ -80,6 +80,9 @@ class LocationPickerActivityUnitTests {
     @Mock
     private lateinit var smallToolbarText: TextView
 
+    @Mock
+    private lateinit var fabCenterOnLocation: FloatingActionButton
+
     @Mock
     private lateinit var tvAttribution: AppCompatTextView
 
@@ -89,7 +92,7 @@ class LocationPickerActivityUnitTests {
     @Before
     fun setUp() {
         MockitoAnnotations.initMocks(this)
-        context = RuntimeEnvironment.application.applicationContext
+        context = RuntimeEnvironment.getApplication().applicationContext
         activity = Robolectric.buildActivity(LocationPickerActivity::class.java).get()
 
         Whitebox.setInternalState(activity, "mapboxMap", mapboxMap)
@@ -103,6 +106,7 @@ class LocationPickerActivityUnitTests {
         Whitebox.setInternalState(activity, "shadow", shadow)
         Whitebox.setInternalState(activity, "largeToolbarText", largeToolbarText)
         Whitebox.setInternalState(activity, "smallToolbarText", smallToolbarText)
+        Whitebox.setInternalState(activity, "fabCenterOnLocation", fabCenterOnLocation)
         Whitebox.setInternalState(activity, "tvAttribution", tvAttribution)
     }
 
@@ -206,6 +210,7 @@ class LocationPickerActivityUnitTests {
         verify(droppedMarkerLayer, times(1)).setProperties(any())
         verify(largeToolbarText, times(1)).text = "Choose a location"
         verify(smallToolbarText, times(1)).text = "Pan and zoom to adjust"
+        verify(fabCenterOnLocation, times(1)).visibility = View.VISIBLE
         verify(mapboxMap, times(1)).addOnCameraMoveStartedListener(activity)
         verify(mapboxMap, times(1)).addOnCameraIdleListener(activity)
     }