chromium/src/third_party/blink/renderer/platform/heap/test/thread_state_scheduling_test.cc

// Copyright 2018 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.

#include "base/run_loop.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/features.h"
#include "third_party/blink/renderer/platform/heap/heap_test_utilities.h"
#include "third_party/blink/renderer/platform/heap/thread_state_scopes.h"
#include "third_party/blink/renderer/platform/scheduler/public/thread_scheduler.h"
#include "third_party/blink/renderer/platform/testing/runtime_enabled_features_test_helpers.h"
#include "third_party/blink/renderer/platform/wtf/functional.h"

namespace blink {

namespace {

void RunLoop() {
  base::RunLoop rl;
  // Push quit task.
  ThreadScheduler::Current()->V8TaskRunner()->PostNonNestableTask(
      FROM_HERE, WTF::Bind(rl.QuitWhenIdleClosure()));
  rl.Run();
}

}  // namespace

class ThreadStateSchedulingTest : public TestSupportingGC {
 public:
  void SetUp() override {
    state_ = ThreadState::Current();
    ClearOutOldGarbage();
  }

  void TearDown() override {
    PreciselyCollectGarbage();
    EXPECT_EQ(ThreadState::kNoGCScheduled, state_->GetGCState());
    EXPECT_FALSE(state_->IsMarkingInProgress());
    EXPECT_FALSE(state_->IsSweepingInProgress());
  }

  BlinkGC::GCReason LastReason() const {
    return state_->reason_for_scheduled_gc_;
  }

  void RunScheduledGC(BlinkGC::StackState stack_state) {
    state_->RunScheduledGC(stack_state);
  }

  ThreadState* state() { return state_; }

 private:
  ThreadState* state_;
};

TEST_F(ThreadStateSchedulingTest, RunIncrementalGCForTesting) {
  ThreadStateSchedulingTest* test = this;

  EXPECT_EQ(ThreadState::kNoGCScheduled, test->state()->GetGCState());
  test->state()->StartIncrementalMarking(
      BlinkGC::GCReason::kForcedGCForTesting);
  EXPECT_EQ(ThreadState::kIncrementalMarkingStepScheduled,
            test->state()->GetGCState());

  RunLoop();
  EXPECT_EQ(ThreadState::kNoGCScheduled, test->state()->GetGCState());
}

}  // namespace blink