src/content/renderer/media/video_source_handler_unittest.cc

   1 // Copyright (c) 2013 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
   5 #include <string>
   6
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "content/renderer/media/media_stream_extra_data.h"
   9 #include "content/renderer/media/media_stream_registry_interface.h"
  10 #include "content/renderer/media/mock_media_stream_dependency_factory.h"
  11 #include "content/renderer/media/mock_media_stream_registry.h"
  12 #include "content/renderer/media/video_source_handler.h"
  13 #include "testing/gmock/include/gmock/gmock.h"
  14 #include "testing/gtest/include/gtest/gtest.h"
  15 #include "third_party/WebKit/public/platform/WebMediaStreamTrack.h"
  16 #include "third_party/WebKit/public/platform/WebString.h"
  17 #include "third_party/libjingle/source/talk/media/base/videorenderer.h"
  18 #include "third_party/libjingle/source/talk/media/webrtc/webrtcvideoframe.h"
  19
  20 using cricket::VideoFrame;
  21
  22 namespace content {
  23
  24 static const std::string kTestStreamUrl = "stream_url";
  25 static const std::string kTestVideoTrackId = "video_track_id";
  26 static const std::string kUnknownStreamUrl = "unknown_stream_url";
  27
  28 class FakeFrameReader : public FrameReaderInterface {
  29  public:
  30   virtual bool GotFrame(VideoFrame* frame) OVERRIDE {
  31     last_frame_.reset(frame);
  32     return true;
  33   }
  34
  35   const VideoFrame* last_frame() {
  36     return last_frame_.get();
  37   }
  38
  39  private:
  40   scoped_ptr<VideoFrame> last_frame_;
  41 };
  42
  43 class VideoSourceHandlerTest : public ::testing::Test {
  44  public:
  45   VideoSourceHandlerTest() : registry_(&dependency_factory_) {
  46     handler_.reset(new VideoSourceHandler(&registry_));
  47     registry_.Init(kTestStreamUrl);
  48     registry_.AddVideoTrack(kTestVideoTrackId);
  49   }
  50
  51  protected:
  52   scoped_ptr<VideoSourceHandler> handler_;
  53   MockMediaStreamDependencyFactory dependency_factory_;
  54   MockMediaStreamRegistry registry_;
  55 };
  56
  57 TEST_F(VideoSourceHandlerTest, OpenClose) {
  58   FakeFrameReader reader;
  59   // Unknow url will return false.
  60   EXPECT_FALSE(handler_->Open(kUnknownStreamUrl, &reader));
  61   EXPECT_TRUE(handler_->Open(kTestStreamUrl, &reader));
  62   cricket::WebRtcVideoFrame test_frame;
  63   int width = 640;
  64   int height = 360;
  65   int64 et = 123456;
  66   int64 ts = 789012;
  67   test_frame.InitToBlack(width, height, 1, 1, et, ts);
  68   cricket::VideoRenderer* receiver = handler_->GetReceiver(&reader);
  69   ASSERT(receiver != NULL);
  70   receiver->RenderFrame(&test_frame);
  71
  72   const VideoFrame* frame = reader.last_frame();
  73   ASSERT_TRUE(frame != NULL);
  74
  75   // Compare |frame| to |test_frame|.
  76   EXPECT_EQ(test_frame.GetWidth(), frame->GetWidth());
  77   EXPECT_EQ(test_frame.GetHeight(), frame->GetHeight());
  78   EXPECT_EQ(test_frame.GetElapsedTime(), frame->GetElapsedTime());
  79   EXPECT_EQ(test_frame.GetTimeStamp(), frame->GetTimeStamp());
  80   EXPECT_EQ(test_frame.GetYPlane(), frame->GetYPlane());
  81   EXPECT_EQ(test_frame.GetUPlane(), frame->GetUPlane());
  82   EXPECT_EQ(test_frame.GetVPlane(), frame->GetVPlane());
  83
  84   EXPECT_FALSE(handler_->Close(NULL));
  85   EXPECT_TRUE(handler_->Close(&reader));
  86   EXPECT_TRUE(handler_->GetReceiver(&reader) == NULL);
  87 }
  88
  89 TEST_F(VideoSourceHandlerTest, OpenWithoutClose) {
  90   FakeFrameReader reader;
  91   EXPECT_TRUE(handler_->Open(kTestStreamUrl, &reader));
  92 }
  93
  94 }  // namespace content