fix: Adb.get_screen()

Author: dtopuzov

core/utils/device/adb.py

@@ -145,7 +145,12 @@ def is_text_visible(id, text, case_sensitive=False):
     @staticmethod
     def get_screen(id, file_path):
         File.clean(path=file_path)
-        Adb.__run_adb_command(command='exec-out screencap -p > ' + file_path, id=id, log_level=logging.INFO)
+        if Settings.OSType == OSType.WINDOWS:
+            Adb.__run_adb_command(command='exec-out screencap -p > ' + file_path, id=id, log_level=logging.INFO)
+        else:
+            Adb.__run_adb_command(command='shell rm /sdcard/screen.png', id=id)
+            Adb.__run_adb_command(command='shell screencap -p /sdcard/screen.png', id=id)
+            Adb.pull(id=id, source='/sdcard/screen.png', target=file_path)
         if File.exists(file_path):
             return
         else:

core/utils/device/device_manager.py

@@ -50,12 +50,12 @@ def stop():
             Process.kill('qemu-system-i38')
 
         @staticmethod
-        def start(emulator, use_snapshot=False):
+        def start(emulator, wipe_data=True):
             emulator_path = os.path.join(ANDROID_HOME, 'emulator', 'emulator')
             base_options = '-no-snapshot-save -no-boot-anim -no-audio'
-            options = '-port {0} -wipe-data -no-snapshot-load {1}'.format(emulator.port, base_options)
-            if use_snapshot:
-                options = '-port {0} {1}'.format(emulator.port, base_options)
+            options = '-port {0} {1}'.format(emulator.port, base_options)
+            if wipe_data:
+                options = '-port {0} -wipe-data {1}'.format(emulator.port, base_options)
             command = '{0} @{1} {2}'.format(emulator_path, emulator.avd, options)
             Log.info('Booting {0} with cmd:'.format(emulator.avd))
             Log.info(command)

core_tests/core/adb_tests.py

@@ -0,0 +1,38 @@
+import os
+
+from core.base_test.tns_test import TnsTest
+from core.log.log import Log
+from core.settings import Settings
+from core.utils.device.adb import Adb
+from core.utils.device.device_manager import DeviceManager
+# noinspection PyMethodMayBeStatic
+from core.utils.file_utils import File
+
+
+class AdbTests(TnsTest):
+    emu = None
+
+    @classmethod
+    def setUpClass(cls):
+        TnsTest.setUpClass()
+        DeviceManager.Emulator.stop()
+        cls.emu = DeviceManager.Emulator.start(Settings.Emulators.DEFAULT, wipe_data=False)
+
+    def setUp(self):
+        TnsTest.setUp(self)
+
+    def tearDown(self):
+        TnsTest.tearDown(self)
+
+    @classmethod
+    def tearDownClass(cls):
+        TnsTest.tearDownClass()
+
+    def test_01_adb_get_source(self):
+        page_source = Adb.get_page_source(id=self.emu.id)
+        assert '</hierarchy>' in page_source
+
+    def test_02_adb_get_screen(self):
+        path = os.path.join(Settings.TEST_OUT_HOME, 'temp.png')
+        File.clean(path)
+        Adb.get_screen(id=self.emu.id, file_path=path)