begin_frame_source_test.h 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. // Copyright 2015 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. #ifndef COMPONENTS_VIZ_TEST_BEGIN_FRAME_SOURCE_TEST_H_
  5. #define COMPONENTS_VIZ_TEST_BEGIN_FRAME_SOURCE_TEST_H_
  6. #include "base/trace_event/traced_value.h"
  7. #include "components/viz/common/frame_sinks/begin_frame_source.h"
  8. #include "components/viz/test/begin_frame_args_test.h"
  9. #include "testing/gmock/include/gmock/gmock.h"
  10. #include "testing/gtest/include/gtest/gtest.h"
  11. // Macros to help set up expected calls on the MockBeginFrameObserver.
  12. #define EXPECT_BEGIN_FRAME_DROP(obs, source_id, sequence_number, frame_time, \
  13. deadline, interval) \
  14. EXPECT_CALL((obs), OnBeginFrame(CreateBeginFrameArgsForTesting( \
  15. BEGINFRAME_FROM_HERE, source_id, sequence_number, \
  16. frame_time, deadline, interval))) \
  17. .Times(1) \
  18. .InSequence((obs).sequence)
  19. #define EXPECT_BEGIN_FRAME_USED(obs, source_id, sequence_number, frame_time, \
  20. deadline, interval) \
  21. EXPECT_CALL((obs), OnBeginFrame(CreateBeginFrameArgsForTesting( \
  22. BEGINFRAME_FROM_HERE, source_id, sequence_number, \
  23. frame_time, deadline, interval))) \
  24. .InSequence((obs).sequence) \
  25. .WillOnce(::testing::SaveArg<0>(&((obs).last_begin_frame_args)))
  26. #define EXPECT_BEGIN_FRAME_USED_MISSED(obs, source_id, sequence_number, \
  27. frame_time, deadline, interval) \
  28. EXPECT_CALL( \
  29. (obs), OnBeginFrame(CreateBeginFrameArgsForTesting( \
  30. BEGINFRAME_FROM_HERE, source_id, sequence_number, frame_time, \
  31. deadline, interval, BeginFrameArgs::MISSED))) \
  32. .InSequence((obs).sequence) \
  33. .WillOnce(::testing::SaveArg<0>(&((obs).last_begin_frame_args)))
  34. #define EXPECT_BEGIN_FRAME_ARGS_DROP(obs, args) \
  35. EXPECT_CALL((obs), OnBeginFrame(args)).Times(1).InSequence((obs).sequence)
  36. #define EXPECT_BEGIN_FRAME_ARGS_USED(obs, args) \
  37. EXPECT_CALL((obs), OnBeginFrame(args)) \
  38. .InSequence((obs).sequence) \
  39. .WillOnce(::testing::SaveArg<0>(&((obs).last_begin_frame_args)))
  40. #define EXPECT_BEGIN_FRAME_SOURCE_PAUSED(obs, paused) \
  41. EXPECT_CALL((obs), OnBeginFrameSourcePausedChanged(paused)) \
  42. .Times(1) \
  43. .InSequence((obs).sequence)
  44. // Macros to send BeginFrameArgs on a FakeBeginFrameSink (and verify
  45. // resulting observer behaviour).
  46. #define SEND_BEGIN_FRAME(args_equal_to, source, sequence_number, frame_time, \
  47. deadline, interval) \
  48. { \
  49. BeginFrameArgs old_args = (source).TestLastUsedBeginFrameArgs(); \
  50. BeginFrameArgs new_args = CreateBeginFrameArgsForTesting( \
  51. BEGINFRAME_FROM_HERE, (source).source_id(), sequence_number, \
  52. frame_time, deadline, interval); \
  53. ASSERT_FALSE(old_args == new_args); \
  54. (source).TestOnBeginFrame(new_args); \
  55. EXPECT_EQ(args_equal_to, (source).TestLastUsedBeginFrameArgs()); \
  56. }
  57. // When dropping LastUsedBeginFrameArgs **shouldn't** change.
  58. #define SEND_BEGIN_FRAME_DROP(source, sequence_number, frame_time, deadline, \
  59. interval) \
  60. SEND_BEGIN_FRAME(old_args, source, sequence_number, frame_time, deadline, \
  61. interval);
  62. // When used LastUsedBeginFrameArgs **should** be updated.
  63. #define SEND_BEGIN_FRAME_USED(source, sequence_number, frame_time, deadline, \
  64. interval) \
  65. SEND_BEGIN_FRAME(new_args, source, sequence_number, frame_time, deadline, \
  66. interval);
  67. namespace viz {
  68. class MockBeginFrameObserver : public BeginFrameObserver {
  69. public:
  70. MOCK_METHOD1(OnBeginFrame, void(const BeginFrameArgs&));
  71. MOCK_CONST_METHOD0(LastUsedBeginFrameArgs, const BeginFrameArgs&());
  72. MOCK_METHOD1(OnBeginFrameSourcePausedChanged, void(bool));
  73. MOCK_CONST_METHOD0(WantsAnimateOnlyBeginFrames, bool());
  74. MOCK_CONST_METHOD0(IsRoot, bool());
  75. virtual void AsValueInto(base::trace_event::TracedValue* dict) const;
  76. // A value different from the normal default returned by a
  77. // BeginFrameObserver so it is easiable traced back here.
  78. static const BeginFrameArgs kDefaultBeginFrameArgs;
  79. MockBeginFrameObserver();
  80. ~MockBeginFrameObserver() override;
  81. BeginFrameArgs last_begin_frame_args;
  82. ::testing::Sequence sequence;
  83. };
  84. } // namespace viz
  85. #endif // COMPONENTS_VIZ_TEST_BEGIN_FRAME_SOURCE_TEST_H_