| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 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 "remoting/ios/bridge/frame_consumer_bridge.h" | |
| 6 | |
| 7 #include <queue> | |
| 8 #include <gtest/gtest.h> | |
| 9 | |
| 10 #include "base/bind.h" | |
| 11 #include "base/memory/scoped_ptr.h" | |
| 12 #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" | |
| 13 #include "third_party/webrtc/modules/desktop_capture/desktop_region.h" | |
| 14 | |
| 15 namespace { | |
| 16 const webrtc::DesktopSize kFrameSize(100, 100); | |
| 17 const webrtc::DesktopVector kDpi(100, 100); | |
| 18 | |
| 19 const webrtc::DesktopRect FrameRect() { | |
| 20 return webrtc::DesktopRect::MakeSize(kFrameSize); | |
| 21 } | |
| 22 | |
| 23 webrtc::DesktopRegion FrameRegion() { | |
| 24 return webrtc::DesktopRegion(FrameRect()); | |
| 25 } | |
| 26 | |
| 27 void FrameDelivery(const webrtc::DesktopSize& view_size, | |
| 28 webrtc::DesktopFrame* buffer, | |
| 29 const webrtc::DesktopRegion& region) { | |
| 30 ASSERT_TRUE(view_size.equals(kFrameSize)); | |
| 31 ASSERT_TRUE(region.Equals(FrameRegion())); | |
| 32 }; | |
| 33 | |
| 34 } // namespace | |
| 35 | |
| 36 namespace remoting { | |
| 37 | |
| 38 class FrameProducerTester : public FrameProducer { | |
| 39 public: | |
| 40 virtual ~FrameProducerTester() {}; | |
| 41 | |
| 42 virtual void DrawBuffer(webrtc::DesktopFrame* buffer) OVERRIDE { | |
| 43 frames.push(buffer); | |
| 44 }; | |
| 45 | |
| 46 virtual void InvalidateRegion(const webrtc::DesktopRegion& region) OVERRIDE { | |
| 47 NOTIMPLEMENTED(); | |
| 48 }; | |
| 49 | |
| 50 virtual void RequestReturnBuffers(const base::Closure& done) OVERRIDE { | |
| 51 // Don't have to actually return the buffers. This function is really | |
| 52 // saying don't use the references anymore, they are now invalid. | |
| 53 while (!frames.empty()) { | |
| 54 frames.pop(); | |
| 55 } | |
| 56 done.Run(); | |
| 57 }; | |
| 58 | |
| 59 virtual void SetOutputSizeAndClip(const webrtc::DesktopSize& view_size, | |
| 60 const webrtc::DesktopRect& clip_area) | |
| 61 OVERRIDE { | |
| 62 viewSize = view_size; | |
| 63 clipArea = clip_area; | |
| 64 }; | |
| 65 | |
| 66 std::queue<webrtc::DesktopFrame*> frames; | |
| 67 webrtc::DesktopSize viewSize; | |
| 68 webrtc::DesktopRect clipArea; | |
| 69 }; | |
| 70 | |
| 71 class FrameConsumerBridgeTest : public ::testing::Test { | |
| 72 protected: | |
| 73 virtual void SetUp() OVERRIDE { | |
| 74 frameProducer_.reset(new FrameProducerTester()); | |
| 75 frameConsumer_.reset(new FrameConsumerBridge(base::Bind(&FrameDelivery))); | |
| 76 frameConsumer_->Initialize(frameProducer_.get()); | |
| 77 } | |
| 78 virtual void TearDown() OVERRIDE {} | |
| 79 | |
| 80 scoped_ptr<FrameProducerTester> frameProducer_; | |
| 81 scoped_ptr<FrameConsumerBridge> frameConsumer_; | |
| 82 }; | |
| 83 | |
| 84 TEST(FrameConsumerBridgeTest_NotInitialized, CreateAndRelease) { | |
| 85 scoped_ptr<FrameConsumerBridge> frameConsumer_( | |
| 86 new FrameConsumerBridge(base::Bind(&FrameDelivery))); | |
| 87 ASSERT_TRUE(frameConsumer_.get() != NULL); | |
| 88 frameConsumer_.reset(); | |
| 89 ASSERT_TRUE(frameConsumer_.get() == NULL); | |
| 90 } | |
| 91 | |
| 92 TEST_F(FrameConsumerBridgeTest, ApplyBuffer) { | |
| 93 webrtc::DesktopFrame* frame = NULL; | |
| 94 ASSERT_EQ(0, frameProducer_->frames.size()); | |
| 95 frameConsumer_->SetSourceSize(kFrameSize, kDpi); | |
| 96 ASSERT_EQ(1, frameProducer_->frames.size()); | |
| 97 | |
| 98 // Return the frame, and ensure we get it back | |
| 99 frame = frameProducer_->frames.front(); | |
| 100 frameProducer_->frames.pop(); | |
| 101 ASSERT_EQ(0, frameProducer_->frames.size()); | |
| 102 frameConsumer_->ApplyBuffer( | |
| 103 kFrameSize, FrameRect(), frame, FrameRegion(), FrameRegion()); | |
| 104 ASSERT_EQ(1, frameProducer_->frames.size()); | |
| 105 ASSERT_TRUE(frame == frameProducer_->frames.front()); | |
| 106 ASSERT_TRUE(frame->data() == frameProducer_->frames.front()->data()); | |
| 107 | |
| 108 // Change the SourceSize, we should get a new frame, but when the old frame is | |
| 109 // submitted we will not get it back. | |
| 110 frameConsumer_->SetSourceSize(webrtc::DesktopSize(1, 1), kDpi); | |
| 111 ASSERT_EQ(2, frameProducer_->frames.size()); | |
| 112 frame = frameProducer_->frames.front(); | |
| 113 frameProducer_->frames.pop(); | |
| 114 ASSERT_EQ(1, frameProducer_->frames.size()); | |
| 115 frameConsumer_->ApplyBuffer( | |
| 116 kFrameSize, FrameRect(), frame, FrameRegion(), FrameRegion()); | |
| 117 ASSERT_EQ(1, frameProducer_->frames.size()); | |
| 118 } | |
| 119 | |
| 120 TEST_F(FrameConsumerBridgeTest, SetSourceSize) { | |
| 121 frameConsumer_->SetSourceSize(webrtc::DesktopSize(0, 0), | |
| 122 webrtc::DesktopVector(0, 0)); | |
| 123 ASSERT_TRUE(frameProducer_->viewSize.equals(webrtc::DesktopSize(0, 0))); | |
| 124 ASSERT_TRUE(frameProducer_->clipArea.equals( | |
| 125 webrtc::DesktopRect::MakeLTRB(0, 0, 0, 0))); | |
| 126 ASSERT_EQ(1, frameProducer_->frames.size()); | |
| 127 ASSERT_TRUE( | |
| 128 frameProducer_->frames.front()->size().equals(webrtc::DesktopSize(0, 0))); | |
| 129 | |
| 130 frameConsumer_->SetSourceSize(kFrameSize, kDpi); | |
| 131 ASSERT_TRUE(frameProducer_->viewSize.equals(kFrameSize)); | |
| 132 ASSERT_TRUE(frameProducer_->clipArea.equals(FrameRect())); | |
| 133 ASSERT_EQ(2, frameProducer_->frames.size()); | |
| 134 frameProducer_->frames.pop(); | |
| 135 ASSERT_TRUE(frameProducer_->frames.front()->size().equals(kFrameSize)); | |
| 136 } | |
| 137 | |
| 138 } // namespace remoting | |
| OLD | NEW |