TECH: Try to fix screenshot tests

KasperskyLab · Aug 22, 2023 · c5ff5b4 · c5ff5b4
1 parent 9305ec4
commit c5ff5b4
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 34 deletions.
diff --git a/.../kotlin/com/kaspersky/kaspressample/docloc_tests/advanced/AdvancedScreenshotSampleTest.kt b/.../kotlin/com/kaspersky/kaspressample/docloc_tests/advanced/AdvancedScreenshotSampleTest.kt
@@ -2,6 +2,7 @@ package com.kaspersky.kaspressample.docloc_tests.advanced
 
 import android.Manifest
 import android.graphics.Color
+import androidx.fragment.app.testing.FragmentScenario
 import androidx.fragment.app.testing.launchFragmentInContainer
 import androidx.test.rule.GrantPermissionRule
 import com.kaspersky.kaspressample.docloc.ScreenshotSampleFragment
@@ -16,7 +17,7 @@ import org.junit.Test
  * For more information see DocLoc wiki page.
  */
 class AdvancedScreenshotSampleTest : DocLocScreenshotTestCase(locales = "en,ru") {
-    private lateinit var view: ScreenshotSampleView
+    private lateinit var scenario: FragmentScenario<ScreenshotSampleFragment>
 
     @get:Rule
     val runtimePermissionRule: GrantPermissionRule = GrantPermissionRule.grant(
@@ -27,26 +28,26 @@ class AdvancedScreenshotSampleTest : DocLocScreenshotTestCase(locales = "en,ru")
     @ScreenShooterTest
     @Test
     fun test() = before {
-        val scenario = launchFragmentInContainer<ScreenshotSampleFragment>()
-        scenario.onFragment {
-            view = getUiSafeProxy(it as ScreenshotSampleView)
-        }
+        scenario = launchFragmentInContainer()
     }.after {
     }.run {
-        step("1. Launch feature screen") {
-            view.setCounterValue(0)
-            view.setBackgroundColor(Color.WHITE)
-            captureScreenshot("1. Startup")
-        }
+        scenario.onFragment {
+            val view = getUiSafeProxy(it as ScreenshotSampleView)
+            step("1. Launch feature screen") {
+                view.setCounterValue(0)
+                view.setBackgroundColor(Color.WHITE)
+                captureScreenshot("1. Startup")
+            }
 
-        step("2. Increase counter by 5") {
-            view.setCounterValue(5)
-            captureScreenshot("2. Value has been increased by 5")
-        }
+            step("2. Increase counter by 5") {
+                view.setCounterValue(5)
+                captureScreenshot("2. Value has been increased by 5")
+            }
 
-        step("3. Set red background color") {
-            view.setBackgroundColor(Color.RED)
-            captureScreenshot("3. Background has been set to red")
+            step("3. Set red background color") {
+                view.setBackgroundColor(Color.RED)
+                captureScreenshot("3. Background has been set to red")
+            }
         }
     }
 }
diff --git a/...n/com/kaspersky/kaspressample/docloc_tests/advanced/AdvancedScreenshotSampleTestLegacy.kt b/...n/com/kaspersky/kaspressample/docloc_tests/advanced/AdvancedScreenshotSampleTestLegacy.kt
@@ -2,6 +2,7 @@ package com.kaspersky.kaspressample.docloc_tests.advanced
 
 import android.Manifest
 import android.graphics.Color
+import androidx.fragment.app.testing.FragmentScenario
 import androidx.fragment.app.testing.launchFragmentInContainer
 import androidx.test.rule.GrantPermissionRule
 import com.kaspersky.kaspressample.docloc.ScreenshotSampleFragment
@@ -20,7 +21,7 @@ class AdvancedScreenshotSampleTestLegacy : DocLocScreenshotTestCase(
     screenshotsDirectory = File("screenshots"),
     locales = "en,ru"
 ) {
-    private lateinit var view: ScreenshotSampleView
+    private lateinit var scenario: FragmentScenario<ScreenshotSampleFragment>
 
     @get:Rule
     val runtimePermissionRule: GrantPermissionRule = GrantPermissionRule.grant(
@@ -31,26 +32,26 @@ class AdvancedScreenshotSampleTestLegacy : DocLocScreenshotTestCase(
     @ScreenShooterTest
     @Test
     fun test() = before {
-        val scenario = launchFragmentInContainer<ScreenshotSampleFragment>()
-        scenario.onFragment {
-            view = getUiSafeProxy(it as ScreenshotSampleView)
-        }
+        scenario = launchFragmentInContainer<ScreenshotSampleFragment>()
     }.after {
     }.run {
-        step("1. Launch feature screen") {
-            view.setCounterValue(0)
-            view.setBackgroundColor(Color.WHITE)
-            captureScreenshot("1. Startup")
-        }
+        scenario.onFragment {
+            val view = getUiSafeProxy(it as ScreenshotSampleView)
+            step("1. Launch feature screen") {
+                view.setCounterValue(0)
+                view.setBackgroundColor(Color.WHITE)
+                captureScreenshot("1. Startup")
+            }
 
-        step("2. Increase counter by 5") {
-            view.setCounterValue(5)
-            captureScreenshot("2. Value has been increased by 5")
-        }
+            step("2. Increase counter by 5") {
+                view.setCounterValue(5)
+                captureScreenshot("2. Value has been increased by 5")
+            }
 
-        step("3. Set red background color") {
-            view.setBackgroundColor(Color.RED)
-            captureScreenshot("3. Background has been set to red")
+            step("3. Set red background color") {
+                view.setBackgroundColor(Color.RED)
+                captureScreenshot("3. Background has been set to red")
+            }
         }
     }
 }