From 399ff718c4f9ae3f18cca550f005eb3b2ea5a316 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz=20Paczos?= <lpaczos@duckduckgo.com>
Date: Tue, 1 Apr 2025 12:11:13 +0200
Subject: [PATCH] adjust DefaultBrowserPromptsExperimentImplTest

---
 .../DefaultBrowserPromptsExperimentImplTest.kt       | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/app/src/test/java/com/duckduckgo/app/browser/defaultbrowsing/prompts/DefaultBrowserPromptsExperimentImplTest.kt b/app/src/test/java/com/duckduckgo/app/browser/defaultbrowsing/prompts/DefaultBrowserPromptsExperimentImplTest.kt
index ad4ca0b70cd4..6a36233c4d5f 100644
--- a/app/src/test/java/com/duckduckgo/app/browser/defaultbrowsing/prompts/DefaultBrowserPromptsExperimentImplTest.kt
+++ b/app/src/test/java/com/duckduckgo/app/browser/defaultbrowsing/prompts/DefaultBrowserPromptsExperimentImplTest.kt
@@ -909,7 +909,7 @@ class DefaultBrowserPromptsExperimentImplTest {
     @Test
     fun `if message dialog shown, then send a pixel`() = runTest {
         val expectedParams = mapOf(
-            PIXEL_PARAM_KEY_VARIANT to "variant_2",
+            PIXEL_PARAM_KEY_VARIANT to "variant2",
             PIXEL_PARAM_KEY_STAGE to "stage_1",
         )
         val dataStoreMock = createDataStoreFake(
@@ -930,7 +930,7 @@ class DefaultBrowserPromptsExperimentImplTest {
     @Test
     fun `if message dialog canceled, then send a pixel`() = runTest {
         val expectedParams = mapOf(
-            PIXEL_PARAM_KEY_VARIANT to "variant_2",
+            PIXEL_PARAM_KEY_VARIANT to "variant2",
             PIXEL_PARAM_KEY_STAGE to "stage_1",
         )
         val dataStoreMock = createDataStoreFake(
@@ -951,7 +951,7 @@ class DefaultBrowserPromptsExperimentImplTest {
     @Test
     fun `if message dialog not now clicked, then send a pixel`() = runTest {
         val expectedParams = mapOf(
-            PIXEL_PARAM_KEY_VARIANT to "variant_2",
+            PIXEL_PARAM_KEY_VARIANT to "variant2",
             PIXEL_PARAM_KEY_STAGE to "stage_1",
         )
         val dataStoreMock = createDataStoreFake(
@@ -972,7 +972,7 @@ class DefaultBrowserPromptsExperimentImplTest {
     @Test
     fun `if message dialog confirmation clicked, then send a pixel`() = runTest {
         val expectedParams = mapOf(
-            PIXEL_PARAM_KEY_VARIANT to "variant_2",
+            PIXEL_PARAM_KEY_VARIANT to "variant2",
             PIXEL_PARAM_KEY_STAGE to "stage_1",
         )
         val dataStoreMock = createDataStoreFake(
@@ -993,7 +993,7 @@ class DefaultBrowserPromptsExperimentImplTest {
     @Test
     fun `if menu item clicked, then send a pixel`() = runTest {
         val expectedParams = mapOf(
-            PIXEL_PARAM_KEY_VARIANT to "variant_2",
+            PIXEL_PARAM_KEY_VARIANT to "variant2",
             PIXEL_PARAM_KEY_STAGE to "stage_1",
         )
         val dataStoreMock = createDataStoreFake(
@@ -1064,7 +1064,7 @@ class DefaultBrowserPromptsExperimentImplTest {
 
     private fun mockActiveCohort(cohortName: AdditionalPromptsCohortName): Cohort {
         val cohort = Cohort(
-            name = cohortName.name,
+            name = cohortName.cohortName,
             weight = 1,
             enrollmentDateET = fakeEnrollmentDateETString,
         )