From cad6c642368b51e862cee5a3f9899e54524a1da5 Mon Sep 17 00:00:00 2001
From: sv-ohorvath <oana.horvath@softvision.com>
Date: Mon, 29 Aug 2022 17:04:11 +0300
Subject: [PATCH] No issue: disable TCP CFR in screenshots tests

---
 .../java/org/mozilla/fenix/screenshots/MenuScreenShotTest.kt | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/app/src/androidTest/java/org/mozilla/fenix/screenshots/MenuScreenShotTest.kt b/app/src/androidTest/java/org/mozilla/fenix/screenshots/MenuScreenShotTest.kt
index 97fa608b5b..b6b6f85baa 100644
--- a/app/src/androidTest/java/org/mozilla/fenix/screenshots/MenuScreenShotTest.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/screenshots/MenuScreenShotTest.kt
@@ -24,6 +24,7 @@ import org.junit.Test
 import org.mozilla.fenix.HomeActivity
 import org.mozilla.fenix.R
 import org.mozilla.fenix.helpers.AndroidAssetDispatcher
+import org.mozilla.fenix.helpers.FeatureSettingsHelper
 import org.mozilla.fenix.helpers.HomeActivityTestRule
 import org.mozilla.fenix.helpers.TestAssetHelper
 import org.mozilla.fenix.helpers.TestHelper.mDevice
@@ -39,6 +40,7 @@ import tools.fastlane.screengrab.locale.LocaleTestRule
 class MenuScreenShotTest : ScreenshotTest() {
     private lateinit var mockWebServer: MockWebServer
     private lateinit var mDevice: UiDevice
+    private val featureSettingsHelper = FeatureSettingsHelper()
 
     @Rule
     @JvmField
@@ -54,10 +56,13 @@ class MenuScreenShotTest : ScreenshotTest() {
             dispatcher = AndroidAssetDispatcher()
             start()
         }
+
+        featureSettingsHelper.setTCPCFREnabled(false)
     }
 
     @After
     fun tearDown() {
+        featureSettingsHelper.resetAllFeatureFlags()
         mActivityTestRule.getActivity().finishAndRemoveTask()
         mockWebServer.shutdown()
     }