chromium/components/performance_manager/execution_context_priority/ad_frame_voter_unittest.cc

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

#include "components/performance_manager/execution_context_priority/ad_frame_voter.h"

#include "base/memory/raw_ptr.h"
#include "components/performance_manager/public/execution_context/execution_context.h"
#include "components/performance_manager/public/graph/graph.h"
#include "components/performance_manager/test_support/graph_test_harness.h"
#include "components/performance_manager/test_support/mock_graphs.h"
#include "components/performance_manager/test_support/voting.h"

namespace performance_manager {
namespace execution_context_priority {

DummyVoteObserver;

namespace {

const execution_context::ExecutionContext* GetExecutionContext(
    const FrameNode* frame_node) { … }

// Both the voting channel and the AdFrameVoter are expected live on the graph,
// without being actual GraphOwned objects. This class wraps both to allow this.
class GraphOwnedWrapper : public GraphOwned { … };

}  // namespace

class AdFrameVoterTest : public GraphTestHarness { … };

// Tests that the AdFrameVoter correctly casts a vote for an ad frame.
TEST_F(AdFrameVoterTest, SetIsAdFrameTrue) { … }

// Tests that the AdFrameVoter correctly does not cast a vote for an untagged
// frame.
TEST_F(AdFrameVoterTest, SetIsAdFrameFalse) { … }

}  // namespace execution_context_priority
}  // namespace performance_manager