1 files changed, 4 insertions, 0 deletions
diff --git a/chromecast/media/cma/test/mock_frame_provider.h b/chromecast/media/cma/test/mock_frame_provider.h
index 31aa305..b83503e 100644
--- a/chromecast/media/cma/test/mock_frame_provider.h
+++ b/chromecast/media/cma/test/mock_frame_provider.h
@@ -22,6 +22,7 @@ class MockFrameProvider : public CodedFrameProvider {
   void Configure(
       const std::vector<bool>& delayed_task_pattern,
       scoped_ptr<FrameGeneratorForTest> frame_generator);
+  void SetDelayFlush(bool delay_flush);
 
   // CodedFrameProvider implementation.
   void Read(const ReadCB& read_cb) override;
@@ -35,8 +36,11 @@ class MockFrameProvider : public CodedFrameProvider {
   // i.e. after receiving a Read request, either delivers a frame right away
   // or wait some time before delivering the frame.
   // |pattern_idx_| is the current index in the pattern.
+  // |delay_flush_| indicates whether to delay flush cb in Flush. Default is
+  // false.
   std::vector<bool> delayed_task_pattern_;
   size_t pattern_idx_;
+  bool delay_flush_;
 
   scoped_ptr<FrameGeneratorForTest> frame_generator_;