Adding upstream version 110.0.1.upstream/110.0.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 196 insertions, 0 deletions

diff --git a/testing/raptor/test/test_cpu.py b/testing/raptor/test/test_cpu.py
new file mode 100644
index 0000000000..c60d5c6577
--- /dev/null
+++ b/testing/raptor/test/test_cpu.py
@@ -0,0 +1,196 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+import os
+import sys
+from unittest import mock
+
+import mozunit
+
+# need this so the raptor unit tests can find output & filter classes
+here = os.path.abspath(os.path.dirname(__file__))
+raptor_dir = os.path.join(os.path.dirname(here), "raptor")
+sys.path.insert(0, raptor_dir)
+
+import cpu
+from webextension import WebExtensionAndroid
+
+
+def test_no_device():
+    raptor = WebExtensionAndroid(
+        "geckoview",
+        "org.mozilla.org.mozilla.geckoview_example",
+        cpu_test=True,
+    )
+    raptor.device = None
+    resp = cpu.start_android_cpu_profiler(raptor)
+    assert resp is None
+
+
+def test_usage_with_invalid_data_returns_zero():
+    with mock.patch("mozdevice.adb.ADBDevice") as device:
+        with mock.patch("control_server.RaptorControlServer") as control_server:
+            # Create a device that returns invalid data
+            device.shell_output.side_effect = ["8.0.0", "geckoview"]
+            device._verbose = True
+
+            # Create a control server
+            control_server.cpu_test = True
+            control_server.device = device
+            raptor = WebExtensionAndroid("geckoview", "org.mozilla.geckoview_example")
+            raptor.config["cpu_test"] = True
+            raptor.control_server = control_server
+            raptor.device = device
+
+            cpu_profiler = cpu.AndroidCPUProfiler(raptor)
+            cpu_profiler.get_app_cpu_usage()
+
+            # Verify the call to submit data was made
+            avg_cpuinfo_data = {
+                "type": "cpu",
+                "test": "usage_with_invalid_data_returns_zero-avg",
+                "unit": "%",
+                "values": {"avg": 0},
+            }
+            min_cpuinfo_data = {
+                "type": "cpu",
+                "test": "usage_with_invalid_data_returns_zero-min",
+                "unit": "%",
+                "values": {"min": 0},
+            }
+            max_cpuinfo_data = {
+                "type": "cpu",
+                "test": "usage_with_invalid_data_returns_zero-max",
+                "unit": "%",
+                "values": {"max": 0},
+            }
+
+            cpu_profiler.generate_android_cpu_profile(
+                "usage_with_invalid_data_returns_zero"
+            )
+            control_server.submit_supporting_data.assert_has_calls(
+                [
+                    mock.call(avg_cpuinfo_data),
+                    mock.call(min_cpuinfo_data),
+                    mock.call(max_cpuinfo_data),
+                ]
+            )
+
+
+def test_usage_with_output():
+    with mock.patch("mozdevice.adb.ADBDevice") as device:
+        with mock.patch("control_server.RaptorControlServer") as control_server:
+            # Override the shell output with sample CPU usage details
+            filepath = os.path.abspath(os.path.dirname(__file__)) + "/files/"
+            with open(filepath + "top-info.txt", "r") as f:
+                test_data = f.read()
+            device.shell_output.side_effect = ["8.0.0", test_data]
+            device._verbose = True
+
+            # Create a control server
+            control_server.cpu_test = True
+            control_server.test_name = "cpuunittest"
+            control_server.device = device
+            control_server.app_name = "org.mozilla.geckoview_example"
+            raptor = WebExtensionAndroid("geckoview", "org.mozilla.geckoview_example")
+            raptor.device = device
+            raptor.config["cpu_test"] = True
+            raptor.control_server = control_server
+
+            cpu_profiler = cpu.AndroidCPUProfiler(raptor)
+            cpu_profiler.polls.append(cpu_profiler.get_app_cpu_usage())
+            cpu_profiler.polls.append(0)
+
+            # Verify the response contains our expected CPU % of 93.7
+            avg_cpuinfo_data = {
+                "type": "cpu",
+                "test": "usage_with_integer_cpu_info_output-avg",
+                "unit": "%",
+                "values": {"avg": 93.7 / 2},
+            }
+            min_cpuinfo_data = {
+                "type": "cpu",
+                "test": "usage_with_integer_cpu_info_output-min",
+                "unit": "%",
+                "values": {"min": 0},
+            }
+            max_cpuinfo_data = {
+                "type": "cpu",
+                "test": "usage_with_integer_cpu_info_output-max",
+                "unit": "%",
+                "values": {"max": 93.7},
+            }
+
+            cpu_profiler.generate_android_cpu_profile(
+                "usage_with_integer_cpu_info_output"
+            )
+            control_server.submit_supporting_data.assert_has_calls(
+                [
+                    mock.call(avg_cpuinfo_data),
+                    mock.call(min_cpuinfo_data),
+                    mock.call(max_cpuinfo_data),
+                ]
+            )
+
+
+def test_usage_with_fallback():
+    with mock.patch("mozdevice.adb.ADBDevice") as device:
+        with mock.patch("control_server.RaptorControlServer") as control_server:
+            device._verbose = True
+
+            # Return what our shell call to dumpsys would give us
+            shell_output = (
+                " 31093 u0_a196  10 -10   8% S    "
+                + "66 1392100K 137012K  fg org.mozilla.geckoview_example"
+            )
+
+            # We set the version to be less than Android 8
+            device.shell_output.side_effect = ["7.0.0", shell_output]
+
+            # Create a control server
+            control_server.cpu_test = True
+            control_server.test_name = "cpuunittest"
+            control_server.device = device
+            control_server.app_name = "org.mozilla.geckoview_example"
+            raptor = WebExtensionAndroid("geckoview", "org.mozilla.geckoview_example")
+            raptor.device = device
+            raptor.config["cpu_test"] = True
+            raptor.control_server = control_server
+
+            cpu_profiler = cpu.AndroidCPUProfiler(raptor)
+            cpu_profiler.polls.append(cpu_profiler.get_app_cpu_usage())
+            cpu_profiler.polls.append(0)
+
+            # Verify the response contains our expected CPU % of 8
+            # pylint --py3k W1619
+            avg_cpuinfo_data = {
+                "type": "cpu",
+                "test": "usage_with_fallback-avg",
+                "unit": "%",
+                "values": {"avg": 8 / 2},
+            }
+            min_cpuinfo_data = {
+                "type": "cpu",
+                "test": "usage_with_fallback-min",
+                "unit": "%",
+                "values": {"min": 0},
+            }
+            max_cpuinfo_data = {
+                "type": "cpu",
+                "test": "usage_with_fallback-max",
+                "unit": "%",
+                "values": {"max": 8},
+            }
+
+            cpu_profiler.generate_android_cpu_profile("usage_with_fallback")
+            control_server.submit_supporting_data.assert_has_calls(
+                [
+                    mock.call(avg_cpuinfo_data),
+                    mock.call(min_cpuinfo_data),
+                    mock.call(max_cpuinfo_data),
+                ]
+            )
+
+
+if __name__ == "__main__":
+    mozunit.main()