// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <memory>

#include "base/android/jni_android.h"
#include "base/android/library_loader/library_loader_hooks.h"
#include "base/check_op.h"
#include "base/no_destructor.h"
#include "base/test/launcher/test_launcher.h"
#include "chrome/app/android/chrome_jni_onload.h"
#include "chrome/test/base/chrome_test_launcher.h"
#include "content/public/app/content_jni_onload.h"
#include "content/public/app/content_main.h"
#include "content/public/test/nested_message_pump_android.h"
#include "content/public/test/network_service_test_helper.h"
#include "ui/base/test/ui_controls.h"

bool NativeInitializationHook(base::android::LibraryProcessType process_type) {
  // This needs to be done before base::TestSuite::Initialize() is called,
  // as it also tries to set MessagePumpForUIFactory.
  base::MessagePump::OverrideMessagePumpForUIFactory(
      []() -> std::unique_ptr<base::MessagePump> {
        return std::make_unique<content::NestedMessagePumpAndroid>();
      });

  if (!content::android::OnJNIOnLoadInit()) {
    return false;
  }

  content::SetContentMainDelegate(new ChromeTestChromeMainDelegate());

  // Setup a working test environment for the network service in case it's used.
  // Only create this object in the utility process, so that its members don't
  // interfere with other test objects in the browser process.
  static base::NoDestructor<std::unique_ptr<content::NetworkServiceTestHelper>>
      network_service_test_helper(content::NetworkServiceTestHelper::Create());

  // Allow ui_controls testing APIs in android_browsertests as tests
  // are run in serial unlike browser_tests on other platforms.
  ui_controls::EnableUIControls();
  return true;
}

// This is called by the VM when the shared library is first loaded.
JNI_EXPORT jint JNI_OnLoad(JavaVM* vm, void* reserved) {
  base::android::InitVM(vm);
  base::android::SetNativeInitializationHook(NativeInitializationHook);

  return JNI_VERSION_1_4;
}