media: Update PipelineIntegrationTestBase to support MemoryDataSource.

media/test/pipeline_integration_test_base.h

 // Copyright (c) 2012 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.
 
 #ifndef MEDIA_TEST_PIPELINE_INTEGRATION_TEST_BASE_H_
 #define MEDIA_TEST_PIPELINE_INTEGRATION_TEST_BASE_H_
 
 #include <stdint.h>
 
 #include "base/md5.h"
@@ skipping 45 matching lines @@
 // in the real AudioRendererImpl & SkCanvasVideoRenderer implementations used in
 // the browser. The renderers in this test don't actually write data to a
 // display or audio device. Both of these devices are simulated since they have
 // little effect on verifying pipeline behavior and allow tests to run faster
 // than real-time.
 class PipelineIntegrationTestBase {
  public:
   PipelineIntegrationTestBase();
   virtual ~PipelineIntegrationTestBase();
 
-  bool WaitUntilOnEnded();
-  PipelineStatus WaitUntilEndedOrError();
+  // Test types for advanced testing and benchmarking (e.g., underflow is
+  // disabled to ensure consistent hashes). May be combined using the bitwise
+  // or operator (and as such must have values that are powers of two).
+  enum TestTypeFlags { kNormal = 0, kHashed = 1, kClockless = 2 };
 
-  // Starts the pipeline (optionally with a CdmContext), returning the final
-  // status code after it has started. |filename| points at a test file located
-  // under media/test/data/.
+  // Starts the pipeline with a file specified by |filename|, optionally with a
+  // CdmContext or a |test_type|, returning the final status code after it has
+  // started. |filename| points at a test file located under media/test/data/.
   PipelineStatus Start(const std::string& filename);
   PipelineStatus Start(const std::string& filename, CdmContext* cdm_context);
+  PipelineStatus Start(const std::string& filename, uint8_t test_type);
 
-  // Starts the pipeline in a particular mode for advanced testing and
-  // benchmarking purposes (e.g., underflow is disabled to ensure consistent
-  // hashes).  May be combined using the bitwise or operator (and as such must
-  // have values that are powers of two).
-  enum TestTypeFlags { kHashed = 1, kClockless = 2 };
-  PipelineStatus Start(const std::string& filename, uint8_t test_type);
+  // Starts the pipeline with |data| (with |size| bytes). The |data| will be
+  // valid throughtout the lifetime of this test.
+  PipelineStatus Start(const uint8_t* data, size_t size, uint8_t test_type);
 
   void Play();
   void Pause();
   bool Seek(base::TimeDelta seek_time);
   bool Suspend();
   bool Resume(base::TimeDelta seek_time);
   void Stop();
   bool WaitUntilCurrentTimeIsAfter(const base::TimeDelta& wait_time);
 
+  bool WaitUntilOnEnded();
+  PipelineStatus WaitUntilEndedOrError();
+
   // Returns the MD5 hash of all video frames seen.  Should only be called once
   // after playback completes.  First time hashes should be generated with
   // --video-threads=1 to ensure correctness.  Pipeline must have been started
   // with hashing enabled.
   std::string GetVideoHash();
 
   // Returns the hash of all audio frames seen.  Should only be called once
   // after playback completes.  Pipeline must have been started with hashing
   // enabled.
   std::string GetAudioHash();
@@ skipping 15 matching lines @@
   scoped_ptr<NullVideoSink> video_sink_;
   bool ended_;
   PipelineStatus pipeline_status_;
   Demuxer::EncryptedMediaInitDataCB encrypted_media_init_data_cb_;
   VideoPixelFormat last_video_frame_format_;
   ColorSpace last_video_frame_color_space_;
   DummyTickClock dummy_clock_;
   AudioHardwareConfig hardware_config_;
   PipelineMetadata metadata_;
 
+  PipelineStatus StartInternal(scoped_ptr<DataSource> data_source,
+                               CdmContext* cdm_context,
+                               uint8_t test_type);
+
+  PipelineStatus StartWithFile(const std::string& filename,
+                               CdmContext* cdm_context,
+                               uint8_t test_type);
+
   void OnSeeked(base::TimeDelta seek_time, PipelineStatus status);
   void OnStatusCallback(PipelineStatus status);
   void DemuxerEncryptedMediaInitDataCB(EmeInitDataType type,
                                        const std::vector<uint8_t>& init_data);
   void set_encrypted_media_init_data_cb(
       const Demuxer::EncryptedMediaInitDataCB& encrypted_media_init_data_cb) {
     encrypted_media_init_data_cb_ = encrypted_media_init_data_cb;
   }
 
   void OnEnded();
   void OnError(PipelineStatus status);
   void QuitAfterCurrentTimeTask(const base::TimeDelta& quit_time);
 
   // Creates Demuxer and sets |demuxer_|.
-  void CreateDemuxer(const std::string& filename);
+  void CreateDemuxer(scoped_ptr<DataSource> data_source);
 
   // Creates and returns a Renderer.
   virtual scoped_ptr<Renderer> CreateRenderer();
 
   void OnVideoFramePaint(const scoped_refptr<VideoFrame>& frame);
 
   MOCK_METHOD1(OnMetadata, void(PipelineMetadata));
   MOCK_METHOD1(OnBufferingStateChanged, void(BufferingState));
   MOCK_METHOD1(DecryptorAttached, void(bool));
   MOCK_METHOD2(OnAddTextTrack,
                void(const TextTrackConfig& config,
                     const AddTextTrackDoneCB& done_cb));
   MOCK_METHOD0(OnWaitingForDecryptionKey, void(void));
 };
 
 }  // namespace media
 
 #endif  // MEDIA_TEST_PIPELINE_INTEGRATION_TEST_BASE_H_