@@ -2,6 +2,7 @@ package fr.free.nrw.commons.leaderboard
2
2
3
3
import android.accounts.Account
4
4
import android.content.Context
5
+ import android.os.Looper.getMainLooper
5
6
import android.view.LayoutInflater
6
7
import android.view.View
7
8
import android.view.ViewGroup
@@ -26,11 +27,14 @@ import org.junit.Test
26
27
import org.junit.runner.RunWith
27
28
import org.mockito.Mock
28
29
import org.mockito.Mockito.`when`
30
+ import org.mockito.Mockito.doReturn
29
31
import org.mockito.MockitoAnnotations
30
32
import org.powermock.reflect.Whitebox
31
33
import org.robolectric.Robolectric
32
34
import org.robolectric.RobolectricTestRunner
33
35
import org.robolectric.RuntimeEnvironment
36
+ import org.robolectric.Shadows
37
+ import org.robolectric.Shadows.shadowOf
34
38
import org.robolectric.annotation.Config
35
39
import org.robolectric.annotation.LooperMode
36
40
import org.robolectric.shadows.ShadowToast
@@ -222,14 +226,25 @@ class LeaderboardFragmentUnitTests {
222
226
@Test
223
227
@Throws(Exception ::class )
224
228
fun testMenuVisibilityOverrideVisibleWithContext () {
229
+ shadowOf(getMainLooper()).idle()
225
230
`when `(parentView.context).thenReturn(context)
226
231
val method: Method = LeaderboardFragment ::class .java.getDeclaredMethod(
227
232
" setMenuVisibility" ,
228
233
Boolean ::class .java
229
234
)
230
235
method.isAccessible = true
231
236
method.invoke(fragment, true )
232
- Assert .assertEquals(ShadowToast .getTextOfLatestToast().toString(), context.getString(R .string.leaderboard_unavailable_beta))
237
+ if (isBetaFlavour) {
238
+ Assert .assertEquals(
239
+ ShadowToast .getTextOfLatestToast().toString(),
240
+ context.getString(R .string.leaderboard_unavailable_beta)
241
+ )
242
+ } else {
243
+ Assert .assertNull(
244
+ ShadowToast .getTextOfLatestToast()
245
+ )
246
+ }
247
+
233
248
}
234
249
235
250
}
0 commit comments