Skip to content

Commit 27185d2

Browse files
CrisBarreiroCrisBarreiro
committed
Fix tests
1 parent c7fdbf9 commit 27185d2

File tree

3 files changed

+134
-36
lines changed

3 files changed

+134
-36
lines changed

‎app/src/androidTest/java/com/duckduckgo/app/browser/BrowserWebViewClientTest.kt‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ import com.duckduckgo.app.statistics.pixels.Pixel
6767
import com.duckduckgo.autoconsent.api.Autoconsent
6868
import com.duckduckgo.autofill.api.BrowserAutofill
6969
import com.duckduckgo.autofill.api.InternalTestUserChecker
70+
import com.duckduckgo.browser.api.AddDocumentStartJavaScriptPlugin
7071
import com.duckduckgo.browser.api.JsInjectorPlugin
7172
import com.duckduckgo.browser.api.WebViewVersionProvider
7273
import com.duckduckgo.common.test.CoroutineTestRule

‎content-scope-scripts/content-scope-scripts-impl/src/test/java/com/duckduckgo/contentscopescripts/impl/ContentScopeScriptsAddDocumentStartJavaScriptTest.kt‎

Lines changed: 13 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,16 @@
11
package com.duckduckgo.contentscopescripts.impl
22

33
import android.webkit.WebView
4-
import com.duckduckgo.app.browser.api.WebViewCapabilityChecker
5-
import com.duckduckgo.browser.api.webviewcompat.WebViewCompatWrapper
64
import com.duckduckgo.common.test.CoroutineTestRule
75
import com.duckduckgo.contentscopescripts.api.contentscopeExperiments.ContentScopeExperiments
6+
import com.duckduckgo.js.messaging.api.AddDocumentStartJavaScript
7+
import com.duckduckgo.js.messaging.api.AddDocumentStartScriptDelegate
88
import kotlinx.coroutines.test.runTest
99
import org.junit.Before
1010
import org.junit.Rule
1111
import org.junit.Test
1212
import org.mockito.kotlin.any
1313
import org.mockito.kotlin.mock
14-
import org.mockito.kotlin.never
1514
import org.mockito.kotlin.verify
1615
import org.mockito.kotlin.whenever
1716

@@ -20,55 +19,33 @@ class ContentScopeScriptsAddDocumentStartJavaScriptTest {
2019
var coroutineRule = CoroutineTestRule()
2120

2221
private val mockWebViewCompatContentScopeScripts: WebViewCompatContentScopeScripts = mock()
23-
private val mockWebViewCapabilityChecker: WebViewCapabilityChecker = mock()
24-
private val mockWebViewCompatWrapper: WebViewCompatWrapper = mock()
2522
private val mockWebView: WebView = mock()
2623
private val mockActiveContentScopeExperiments: ContentScopeExperiments = mock()
24+
private val mockAddDocumentStartScriptDelegate: AddDocumentStartScriptDelegate = mock()
25+
private val mockAddDocumentStartJavaScript: AddDocumentStartJavaScript = mock()
2726

2827
private lateinit var testee: ContentScopeScriptsAddDocumentStartJavaScript
2928

3029
@Before
3130
fun setUp() = runTest {
3231
whenever(mockActiveContentScopeExperiments.getActiveExperiments()).thenReturn(listOf())
32+
whenever(mockAddDocumentStartScriptDelegate.createPlugin(any())).thenReturn(mockAddDocumentStartJavaScript)
3333
testee = ContentScopeScriptsAddDocumentStartJavaScript(
3434
mockWebViewCompatContentScopeScripts,
3535
mockActiveContentScopeExperiments,
36+
mockAddDocumentStartScriptDelegate,
3637
)
3738
}
3839

3940
@Test
40-
fun whenFeatureIsEnabledAndCapabilitySupportedThenCallScriptInjectionWithCorrectParams() =
41-
runTest {
42-
whenever(mockWebViewCompatContentScopeScripts.isEnabled()).thenReturn(true)
43-
whenever(mockWebViewCapabilityChecker.isSupported(any())).thenReturn(true)
44-
whenever(mockWebViewCompatContentScopeScripts.getScript(any())).thenReturn("script")
41+
fun whenAddDocumentStartJavaScriptCalledThenDelegateToCreatedPlugin() = runTest {
42+
testee.addDocumentStartJavaScript(mockWebView)
4543

46-
testee.addDocumentStartJavaScript(mockWebView)
47-
48-
verify(mockWebViewCompatWrapper).addDocumentStartJavaScript(mockWebView, "script", setOf("*"))
49-
}
50-
51-
@Test
52-
fun whenFeatureIsDisabledAndCapabilitySupportedThenDoNotCallScriptInjection() =
53-
runTest {
54-
whenever(mockWebViewCompatContentScopeScripts.isEnabled()).thenReturn(false)
55-
whenever(mockWebViewCapabilityChecker.isSupported(any())).thenReturn(true)
56-
whenever(mockWebViewCompatContentScopeScripts.getScript(any())).thenReturn("script")
57-
58-
testee.addDocumentStartJavaScript(mockWebView)
59-
60-
verify(mockWebViewCompatWrapper, never()).addDocumentStartJavaScript(any(), any(), any())
61-
}
44+
verify(mockAddDocumentStartJavaScript).addDocumentStartJavaScript(mockWebView)
45+
}
6246

6347
@Test
64-
fun whenFeatureIsEnabledAndCapabilityNotSupportedThenDoNotCallScriptInjection() =
65-
runTest {
66-
whenever(mockWebViewCompatContentScopeScripts.isEnabled()).thenReturn(true)
67-
whenever(mockWebViewCapabilityChecker.isSupported(any())).thenReturn(false)
68-
whenever(mockWebViewCompatContentScopeScripts.getScript(any())).thenReturn("script")
69-
70-
testee.addDocumentStartJavaScript(mockWebView)
71-
72-
verify(mockWebViewCompatWrapper, never()).addDocumentStartJavaScript(any(), any(), any())
73-
}
48+
fun whenConstructedThenCreatePluginWithCorrectStrategy() = runTest {
49+
verify(mockAddDocumentStartScriptDelegate).createPlugin(any())
50+
}
7451
}

‎js-messaging/js-messaging-impl/src/test/java/com/duckduckgo/js/messaging/impl/RealAddDocumentStartScriptDelegateTest.kt‎

Lines changed: 120 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,120 @@
1+
package com.duckduckgo.js.messaging.impl
2+
3+
import android.webkit.WebView
4+
import androidx.webkit.ScriptHandler
5+
import com.duckduckgo.app.browser.api.WebViewCapabilityChecker
6+
import com.duckduckgo.browser.api.webviewcompat.WebViewCompatWrapper
7+
import com.duckduckgo.common.test.CoroutineTestRule
8+
import com.duckduckgo.common.utils.DispatcherProvider
9+
import com.duckduckgo.js.messaging.api.AddDocumentStartJavaScript
10+
import com.duckduckgo.js.messaging.api.AddDocumentStartJavaScriptScriptStrategy
11+
import kotlinx.coroutines.test.runTest
12+
import org.junit.Before
13+
import org.junit.Rule
14+
import org.junit.Test
15+
import org.mockito.kotlin.any
16+
import org.mockito.kotlin.mock
17+
import org.mockito.kotlin.never
18+
import org.mockito.kotlin.verify
19+
import org.mockito.kotlin.whenever
20+
21+
class RealAddDocumentStartScriptDelegateTest {
22+
23+
@get:Rule
24+
var coroutineRule = CoroutineTestRule()
25+
26+
private val mockWebViewCapabilityChecker: WebViewCapabilityChecker = mock()
27+
private val mockWebViewCompatWrapper: WebViewCompatWrapper = mock()
28+
private val mockWebView: WebView = mock()
29+
private val mockScriptHandler: ScriptHandler = mock()
30+
private val mockDispatcherProvider: DispatcherProvider = mock()
31+
32+
private lateinit var testee: RealAddDocumentStartScriptDelegate
33+
private lateinit var plugin: AddDocumentStartJavaScript
34+
35+
@Before
36+
fun setUp() = runTest {
37+
whenever(mockDispatcherProvider.main()).thenReturn(coroutineRule.testDispatcher)
38+
testee = RealAddDocumentStartScriptDelegate(
39+
mockWebViewCapabilityChecker,
40+
coroutineRule.testScope,
41+
mockDispatcherProvider,
42+
mockWebViewCompatWrapper,
43+
)
44+
}
45+
46+
@Test
47+
fun whenFeatureEnabledAndCapabilitySupportedThenInjectScript() = runTest {
48+
val mockStrategy = createMockStrategy(canInject = true, scriptString = "test script")
49+
plugin = testee.createPlugin(mockStrategy)
50+
whenever(mockWebViewCapabilityChecker.isSupported(any())).thenReturn(true)
51+
whenever(mockWebViewCompatWrapper.addDocumentStartJavaScript(any(), any(), any())).thenReturn(mockScriptHandler)
52+
53+
plugin.addDocumentStartJavaScript(mockWebView)
54+
55+
verify(mockWebViewCompatWrapper).addDocumentStartJavaScript(mockWebView, "test script", setOf("*"))
56+
}
57+
58+
@Test
59+
fun whenFeatureDisabledThenDoNotInjectScript() = runTest {
60+
val mockStrategy = createMockStrategy(canInject = false, scriptString = "test script")
61+
plugin = testee.createPlugin(mockStrategy)
62+
whenever(mockWebViewCapabilityChecker.isSupported(any())).thenReturn(true)
63+
64+
plugin.addDocumentStartJavaScript(mockWebView)
65+
66+
verify(mockWebViewCompatWrapper, never()).addDocumentStartJavaScript(any(), any(), any())
67+
}
68+
69+
@Test
70+
fun whenCapabilityNotSupportedThenDoNotInjectScript() = runTest {
71+
val mockStrategy = createMockStrategy(canInject = true, scriptString = "test script")
72+
plugin = testee.createPlugin(mockStrategy)
73+
whenever(mockWebViewCapabilityChecker.isSupported(any())).thenReturn(false)
74+
75+
plugin.addDocumentStartJavaScript(mockWebView)
76+
77+
verify(mockWebViewCompatWrapper, never()).addDocumentStartJavaScript(any(), any(), any())
78+
}
79+
80+
@Test
81+
fun whenScriptStringSameAsCurrentThenDoNotReinject() = runTest {
82+
val mockStrategy = createMockStrategy(canInject = true, scriptString = "test script")
83+
plugin = testee.createPlugin(mockStrategy)
84+
whenever(mockWebViewCapabilityChecker.isSupported(any())).thenReturn(true)
85+
whenever(mockWebViewCompatWrapper.addDocumentStartJavaScript(any(), any(), any())).thenReturn(mockScriptHandler)
86+
87+
plugin.addDocumentStartJavaScript(mockWebView)
88+
plugin.addDocumentStartJavaScript(mockWebView)
89+
90+
verify(mockWebViewCompatWrapper).addDocumentStartJavaScript(mockWebView, "test script", setOf("*"))
91+
}
92+
93+
@Test
94+
fun whenScriptStringDifferentThenRemoveOldAndInjectNew() = runTest {
95+
val mockStrategy: AddDocumentStartJavaScriptScriptStrategy = mock()
96+
whenever(mockStrategy.canInject()).thenReturn(true)
97+
whenever(mockStrategy.getScriptString()).thenReturn("script 1")
98+
whenever(mockStrategy.allowedOriginRules).thenReturn(setOf("*"))
99+
plugin = testee.createPlugin(mockStrategy)
100+
whenever(mockWebViewCapabilityChecker.isSupported(any())).thenReturn(true)
101+
whenever(mockWebViewCompatWrapper.addDocumentStartJavaScript(any(), any(), any())).thenReturn(mockScriptHandler)
102+
103+
plugin.addDocumentStartJavaScript(mockWebView)
104+
105+
whenever(mockStrategy.getScriptString()).thenReturn("script 2")
106+
107+
plugin.addDocumentStartJavaScript(mockWebView)
108+
109+
verify(mockScriptHandler).remove()
110+
verify(mockWebViewCompatWrapper).addDocumentStartJavaScript(mockWebView, "script 2", setOf("*"))
111+
}
112+
113+
private fun createMockStrategy(canInject: Boolean, scriptString: String): AddDocumentStartJavaScriptScriptStrategy {
114+
return object : AddDocumentStartJavaScriptScriptStrategy {
115+
override suspend fun canInject(): Boolean = canInject
116+
override suspend fun getScriptString(): String = scriptString
117+
override val allowedOriginRules: Set<String> = setOf("*")
118+
}
119+
}
120+
}

0 commit comments

Comments
 (0)