freezing_vote_decorator_unittest.cc 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. // Copyright 2020 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/performance_manager/decorators/freezing_vote_decorator.h"
  5. #include "components/performance_manager/freezing/freezing_vote_aggregator.h"
  6. #include "components/performance_manager/graph/page_node_impl.h"
  7. #include "components/performance_manager/public/freezing/freezing.h"
  8. #include "components/performance_manager/test_support/graph_test_harness.h"
  9. #include "testing/gtest/include/gtest/gtest.h"
  10. namespace performance_manager {
  11. namespace {
  12. static const freezing::FreezingVote kCannotFreezeVote(
  13. freezing::FreezingVoteValue::kCannotFreeze,
  14. "cannot freeze");
  15. static const freezing::FreezingVote kCanFreezeVote(
  16. freezing::FreezingVoteValue::kCanFreeze,
  17. "can freeze");
  18. } // namespace
  19. class FreezingVoteDecoratorTest : public GraphTestHarness {
  20. public:
  21. using Super = GraphTestHarness;
  22. FreezingVoteDecoratorTest() = default;
  23. ~FreezingVoteDecoratorTest() override = default;
  24. void SetUp() override {
  25. Super::SetUp();
  26. graph()->PassToGraph(std::make_unique<FreezingVoteDecorator>());
  27. }
  28. };
  29. TEST_F(FreezingVoteDecoratorTest, VotesAreForwarded) {
  30. auto page_node = CreateNode<PageNodeImpl>();
  31. EXPECT_FALSE(page_node->freezing_vote());
  32. freezing::FreezingVotingChannel voter =
  33. graph()
  34. ->GetRegisteredObjectAs<freezing::FreezingVoteAggregator>()
  35. ->GetVotingChannel();
  36. voter.SubmitVote(page_node.get(), kCannotFreezeVote);
  37. ASSERT_TRUE(page_node->freezing_vote().has_value());
  38. EXPECT_EQ(kCannotFreezeVote, page_node->freezing_vote().value());
  39. voter.ChangeVote(page_node.get(), kCanFreezeVote);
  40. ASSERT_TRUE(page_node->freezing_vote().has_value());
  41. EXPECT_EQ(kCanFreezeVote, page_node->freezing_vote().value());
  42. voter.ChangeVote(page_node.get(), kCannotFreezeVote);
  43. ASSERT_TRUE(page_node->freezing_vote().has_value());
  44. EXPECT_EQ(kCannotFreezeVote, page_node->freezing_vote().value());
  45. voter.InvalidateVote(page_node.get());
  46. EXPECT_FALSE(page_node->freezing_vote());
  47. }
  48. } // namespace performance_manager