This repository has been archived by the owner on Feb 20, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
0c2d22f
commit a16bfb2
Showing
6 changed files
with
272 additions
and
28 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
76 changes: 76 additions & 0 deletions
76
app/src/test/java/org/mozilla/fenix/search/telemetry/BaseSearchTelemetryTest.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,76 @@ | ||
| package org.mozilla.fenix.search.telemetry | ||
|
|
||
| import io.mockk.mockk | ||
| import io.mockk.spyk | ||
| import io.mockk.verify | ||
| import mozilla.components.browser.state.store.BrowserStore | ||
| import mozilla.components.concept.engine.Engine | ||
| import org.json.JSONObject | ||
| import org.junit.Assert.assertEquals | ||
| import org.junit.Test | ||
|
|
||
| class BaseSearchTelemetryTest { | ||
|
|
||
| private lateinit var baseTelemetry: BaseSearchTelemetry | ||
| private lateinit var handler: BaseSearchTelemetry.SearchTelemetryMessageHandler | ||
|
|
||
| @org.junit.Before | ||
| fun setUp() { | ||
| baseTelemetry = spyk(object : BaseSearchTelemetry() { | ||
|
|
||
| override fun install(engine: Engine, store: BrowserStore) { | ||
| // mock, do nothing | ||
| } | ||
|
|
||
| override fun processMessage(message: JSONObject) { | ||
| // mock, do nothing | ||
| } | ||
| }) | ||
| handler = baseTelemetry.SearchTelemetryMessageHandler() | ||
| } | ||
|
|
||
| @Test | ||
| fun install() { | ||
| val engine = mockk<Engine>(relaxed = true) | ||
| val store = mockk<BrowserStore>(relaxed = true) | ||
| val id = "id" | ||
| val resourceUrl = "resourceUrl" | ||
| val messageId = "messageId" | ||
| val extensionInfo = ExtensionInfo(id, resourceUrl, messageId) | ||
|
|
||
| baseTelemetry.installWebExtension(engine, store, extensionInfo) | ||
|
|
||
| verify { | ||
| engine.installWebExtension( | ||
| id = id, | ||
| url = resourceUrl, | ||
| allowContentMessaging = true, | ||
| onSuccess = any(), | ||
| onError = any() | ||
| ) | ||
| } | ||
| } | ||
|
|
||
| @Test | ||
| fun `get provider for google url`() { | ||
| val url = "https://www.google.com/search?q=computers" | ||
|
|
||
| assertEquals("google", baseTelemetry.getProviderForUrl(url)?.name) | ||
| } | ||
|
|
||
| @Test | ||
| fun `message handler finds a valid json object`() { | ||
| val message = JSONObject() | ||
|
|
||
| handler.onMessage(message, mockk()) | ||
|
|
||
| verify { baseTelemetry.processMessage(message) } | ||
| } | ||
|
|
||
| @Test(expected = IllegalStateException::class) | ||
| fun `message handler finds no json object`() { | ||
| val message = "message" | ||
|
|
||
| handler.onMessage(message, mockk()) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
175 changes: 175 additions & 0 deletions
175
app/src/test/java/org/mozilla/fenix/search/telemetry/incontent/InContentTelemetryTest.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,175 @@ | ||
| package org.mozilla.fenix.search.telemetry.incontent | ||
|
|
||
| import io.mockk.mockk | ||
| import io.mockk.slot | ||
| import io.mockk.spyk | ||
| import io.mockk.verify | ||
| import mozilla.components.browser.state.store.BrowserStore | ||
| import mozilla.components.concept.engine.Engine | ||
| import org.json.JSONArray | ||
| import org.json.JSONObject | ||
| import org.junit.Assert | ||
| import org.junit.Before | ||
| import org.junit.Test | ||
| import org.junit.runner.RunWith | ||
| import org.mozilla.fenix.components.metrics.Event | ||
| import org.mozilla.fenix.components.metrics.MetricController | ||
| import org.mozilla.fenix.helpers.FenixRobolectricTestRunner | ||
| import org.mozilla.fenix.search.telemetry.ExtensionInfo | ||
| import org.mozilla.fenix.search.telemetry.incontent.InContentTelemetry.Companion.COOKIES_EXTENSION_ID | ||
| import org.mozilla.fenix.search.telemetry.incontent.InContentTelemetry.Companion.COOKIES_EXTENSION_RESOURCE_URL | ||
| import org.mozilla.fenix.search.telemetry.incontent.InContentTelemetry.Companion.COOKIES_MESSAGE_ID | ||
| import org.mozilla.fenix.search.telemetry.incontent.InContentTelemetry.Companion.COOKIES_MESSAGE_LIST_KEY | ||
| import org.mozilla.fenix.search.telemetry.incontent.InContentTelemetry.Companion.COOKIES_MESSAGE_SESSION_URL_KEY | ||
|
|
||
| @RunWith(FenixRobolectricTestRunner::class) | ||
| class InContentTelemetryTest { | ||
|
|
||
| private val metrics: MetricController = mockk(relaxed = true) | ||
| private lateinit var telemetry: InContentTelemetry | ||
|
|
||
| @Before | ||
| fun setUp() { | ||
| telemetry = spyk(InContentTelemetry(metrics)) | ||
| } | ||
|
|
||
| @Test | ||
| fun install() { | ||
| val engine = mockk<Engine>(relaxed = true) | ||
| val store = mockk<BrowserStore>(relaxed = true) | ||
| val extensionInfo = slot<ExtensionInfo>() | ||
|
|
||
| telemetry.install(engine, store) | ||
|
|
||
| verify { telemetry.installWebExtension(engine, store, capture(extensionInfo)) } | ||
| Assert.assertEquals(COOKIES_EXTENSION_ID, extensionInfo.captured.id) | ||
| Assert.assertEquals(COOKIES_EXTENSION_RESOURCE_URL, extensionInfo.captured.resourceUrl) | ||
| Assert.assertEquals(COOKIES_MESSAGE_ID, extensionInfo.captured.messageId) | ||
| } | ||
|
|
||
| @Test | ||
| fun processMessage() { | ||
| val first = JSONObject() | ||
| val second = JSONObject() | ||
| val array = JSONArray() | ||
| array.put(first) | ||
| array.put(second) | ||
| val message = JSONObject() | ||
| val url = "https://www.google.com/search?q=aaa" | ||
| message.put(COOKIES_MESSAGE_LIST_KEY, array) | ||
| message.put(COOKIES_MESSAGE_SESSION_URL_KEY, url) | ||
|
|
||
| telemetry.processMessage(message) | ||
|
|
||
| verify { telemetry.trackPartnerUrlTypeMetric(url, any()) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track google sap metric`() { | ||
| val url = "https://www.google.com/search?q=aaa&client=firefox-b-m" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("google.in-content.sap.firefox-b-m")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track duckduckgo sap metric`() { | ||
| val url = "https://duckduckgo.com/?q=aaa&t=fpas" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("duckduckgo.in-content.sap.fpas")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track baidu sap metric`() { | ||
| val url = "https://www.baidu.com/from=844b/s?wd=aaa&tn=34046034_firefox" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("baidu.in-content.sap.34046034_firefox")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track bing sap metric`() { | ||
| val url = "https://www.bing.com/search?q=aaa&pc=MOZMBA" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("bing.in-content.sap.mozmba")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track google sap-follow-on metric`() { | ||
| val url = "https://www.google.com/search?q=aaa&client=firefox-b-m&oq=random" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("google.in-content.sap-follow-on.firefox-b-m")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track baidu sap-follow-on metric`() { | ||
| val url = "https://www.baidu.com/from=844b/s?wd=aaa&tn=34046034_firefox&oq=random" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("baidu.in-content.sap-follow-on.34046034_firefox")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track bing sap-follow-on metric by cookies`() { | ||
| val url = "https://www.bing.com/search?q=aaa&pc=MOZMBA&form=QBRERANDOM" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, createCookieList()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("bing.in-content.sap-follow-on.mozmba")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track google organic metric`() { | ||
| val url = "https://www.google.com/search?q=aaa" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("google.in-content.organic.none")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track duckduckgo organic metric`() { | ||
| val url = "https://duckduckgo.com/?q=aaa" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("duckduckgo.in-content.organic.none")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track bing organic metric`() { | ||
| val url = "https://www.bing.com/search?q=aaa" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("bing.in-content.organic.none")) } | ||
| } | ||
|
|
||
| @Test | ||
| fun `track baidu organic metric`() { | ||
| val url = "https://www.baidu.com/from=844b/s?wd=aaa" | ||
|
|
||
| telemetry.trackPartnerUrlTypeMetric(url, listOf()) | ||
|
|
||
| verify { metrics.track(Event.SearchInContent("baidu.in-content.organic.none")) } | ||
| } | ||
|
|
||
| private fun createCookieList(): List<JSONObject> { | ||
| val first = JSONObject() | ||
| first.put("name", "SRCHS") | ||
| first.put("value", "PC=MOZMBA") | ||
| val second = JSONObject() | ||
| second.put("name", "RANDOM") | ||
| second.put("value", "RANDOM") | ||
| return listOf(first, second) | ||
| } | ||
| } |