1 files changed, 22 insertions, 5 deletions
diff --git a/simpleperf/environment.cpp b/simpleperf/environment.cpp
index 3591626e..ab537dcc 100644
--- a/simpleperf/environment.cpp
+++ b/simpleperf/environment.cpp
@@ -808,18 +808,35 @@ class SimpleperfAppRunner : public InAppRunner {
 
 }  // namespace
 
+static bool allow_run_as = true;
+static bool allow_simpleperf_app_runner = true;
+
+void SetRunInAppToolForTesting(bool run_as, bool simpleperf_app_runner) {
+  allow_run_as = run_as;
+  allow_simpleperf_app_runner = simpleperf_app_runner;
+}
+
 bool RunInAppContext(const std::string& app_package_name, const std::string& cmd,
                      const std::vector<std::string>& args, size_t workload_args_size,
                      const std::string& output_filepath, bool need_tracepoint_events) {
-  std::unique_ptr<InAppRunner> in_app_runner(new RunAs(app_package_name));
-  if (!in_app_runner->Prepare()) {
+  std::unique_ptr<InAppRunner> in_app_runner;
+  if (allow_run_as) {
+    in_app_runner.reset(new RunAs(app_package_name));
+    if (!in_app_runner->Prepare()) {
+      in_app_runner = nullptr;
+    }
+  }
+  if (!in_app_runner && allow_simpleperf_app_runner) {
     in_app_runner.reset(new SimpleperfAppRunner(app_package_name));
     if (!in_app_runner->Prepare()) {
-      LOG(ERROR) << "Package " << app_package_name
-          << " doesn't exist or isn't debuggable/profileable.";
-      return false;
+      in_app_runner = nullptr;
     }
   }
+  if (!in_app_runner) {
+    LOG(ERROR) << "Package " << app_package_name
+               << " doesn't exist or isn't debuggable/profileable.";
+    return false;
+  }
   return in_app_runner->RunCmdInApp(cmd, args, workload_args_size, output_filepath,
                                     need_tracepoint_events);
 }