src/gn/ffi/test_with_scope.h

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

#ifndef TOOLS_GN_FFI_TEST_WITH_SCOPE_H_
#define TOOLS_GN_FFI_TEST_WITH_SCOPE_H_

#include <cstdlib>
#include <memory>
#include <mutex>

#include "base/command_line.h"
#include "gn/scheduler.h"
#include "gn/test_with_scope.h"

inline std::unique_ptr<TestWithScope> NewTestWithScope() {
  static std::once_flag init_flag;
  std::call_once(init_flag, []() {
    if (!base::CommandLine::InitializedForCurrentProcess()) {
      int argc = 1;
      const char* argv[] = {"gn_rust_tests", nullptr};
      base::CommandLine::Init(argc, argv);
      // Since this is used in rust, rather than trying to attach the lifetime
      // to anything, we just attach it to the process itself.
      std::atexit([]() {
        if (base::CommandLine::InitializedForCurrentProcess()) {
          base::CommandLine::Reset();
        }
      });
    }
  });
  // Initialize a single Scheduler instance that lives for the lifetime of the
  // test process. This ensures g_scheduler is always set and never cleared
  // between tests.
  static Scheduler scheduler;
  return std::make_unique<TestWithScope>();
}

#endif  // TOOLS_GN_FFI_TEST_WITH_SCOPE_H_