1 files changed, 91 insertions, 0 deletions
diff --git a/media/base/mock_reader.h b/media/base/mock_reader.h
new file mode 100644
index 0000000..ad8db88
--- /dev/null
+++ b/media/base/mock_reader.h
@@ -0,0 +1,91 @@
+// Copyright (c) 2009 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_BASE_MOCK_READER_H_
+#define MEDIA_BASE_MOCK_READER_H_
+
+#include <string>
+
+#include "base/ref_counted.h"
+#include "base/waitable_event.h"
+#include "media/base/filters.h"
+
+namespace media {
+
+// Ref counted object so we can create callbacks for asynchronous Read()
+// methods for any filter type.
+template <class FilterType, class BufferType>
+class MockReader :
+    public base::RefCountedThreadSafe<MockReader<FilterType, BufferType> > {
+ public:
+  MockReader()
+      : called_(false),
+        expecting_call_(false),
+        wait_for_read_(false, false) {
+  }
+
+  virtual ~MockReader() {
+  }
+
+  // Prepares this object for another read.
+  void Reset() {
+    DCHECK(!expecting_call_);
+    expecting_call_ = false;
+    called_ = false;
+    buffer_ = NULL;
+    wait_for_read_.Reset();
+  }
+
+  // Executes an asynchronous read on the given filter.
+  void Read(FilterType* filter) {
+    DCHECK(!expecting_call_);
+    called_ = false;
+    expecting_call_ = true;
+    filter->Read(NewCallback(this, &MockReader::OnReadComplete));
+  }
+
+  // Waits 500ms for the read callback to be completed.  Returns true if the
+  // read was completed, false otherwise.
+  bool WaitForRead() {
+    return wait_for_read_.TimedWait(base::TimeDelta::FromMilliseconds(500));
+  }
+
+  // Mock accessors.
+  BufferType* buffer() { return buffer_; }
+  bool called() { return called_; }
+  bool expecting_call() { return expecting_call_; }
+
+ private:
+  void OnReadComplete(BufferType* buffer) {
+    DCHECK(!called_);
+    DCHECK(expecting_call_);
+    expecting_call_ = false;
+    called_ = true;
+    buffer_ = buffer;
+    wait_for_read_.Signal();
+  }
+
+  // Reference to the buffer provided in the callback.
+  scoped_refptr<BufferType> buffer_;
+
+  // Whether or not the callback was executed.
+  bool called_;
+
+  // Whether or not this reader was expecting a callback.
+  bool expecting_call_;
+
+  // Used by tests to wait for the callback to be executed.
+  base::WaitableEvent wait_for_read_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockReader);
+};
+
+// Commonly used reader types.
+typedef MockReader<DemuxerStream, Buffer> DemuxerStreamReader;
+typedef MockReader<AudioDecoder, Buffer> AudioDecoderReader;
+typedef MockReader<VideoDecoder, VideoFrame> VideoDecoderReader;
+
+}  // namespace media
+
+#endif  // MEDIA_BASE_MOCK_READER_H_