Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(17)

Side by Side Diff: webrtc/common_video/i420_buffer_pool_unittest.cc

Issue 2906053002: Update I420Buffer to new VideoFrameBuffer interface (Closed)
Patch Set: Make const versions of Get functions Created 3 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 /* 1 /*
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source 5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 #include <string> 11 #include <string>
12 12
13 #include "webrtc/common_video/include/i420_buffer_pool.h" 13 #include "webrtc/common_video/include/i420_buffer_pool.h"
14 #include "webrtc/test/gtest.h" 14 #include "webrtc/test/gtest.h"
15 15
16 namespace webrtc { 16 namespace webrtc {
17 17
18 TEST(TestI420BufferPool, SimpleFrameReuse) { 18 TEST(TestI420BufferPool, SimpleFrameReuse) {
19 I420BufferPool pool; 19 I420BufferPool pool;
20 rtc::scoped_refptr<VideoFrameBuffer> buffer = pool.CreateBuffer(16, 16); 20 rtc::scoped_refptr<I420BufferInterface> buffer = pool.CreateBuffer(16, 16);
21 EXPECT_EQ(16, buffer->width()); 21 EXPECT_EQ(16, buffer->width());
22 EXPECT_EQ(16, buffer->height()); 22 EXPECT_EQ(16, buffer->height());
23 // Extract non-refcounted pointers for testing. 23 // Extract non-refcounted pointers for testing.
24 const uint8_t* y_ptr = buffer->DataY(); 24 const uint8_t* y_ptr = buffer->DataY();
25 const uint8_t* u_ptr = buffer->DataU(); 25 const uint8_t* u_ptr = buffer->DataU();
26 const uint8_t* v_ptr = buffer->DataV(); 26 const uint8_t* v_ptr = buffer->DataV();
27 // Release buffer so that it is returned to the pool. 27 // Release buffer so that it is returned to the pool.
28 buffer = nullptr; 28 buffer = nullptr;
29 // Check that the memory is resued. 29 // Check that the memory is resued.
30 buffer = pool.CreateBuffer(16, 16); 30 buffer = pool.CreateBuffer(16, 16);
31 EXPECT_EQ(y_ptr, buffer->DataY()); 31 EXPECT_EQ(y_ptr, buffer->DataY());
32 EXPECT_EQ(u_ptr, buffer->DataU()); 32 EXPECT_EQ(u_ptr, buffer->DataU());
33 EXPECT_EQ(v_ptr, buffer->DataV()); 33 EXPECT_EQ(v_ptr, buffer->DataV());
34 EXPECT_EQ(16, buffer->width()); 34 EXPECT_EQ(16, buffer->width());
35 EXPECT_EQ(16, buffer->height()); 35 EXPECT_EQ(16, buffer->height());
36 } 36 }
37 37
38 TEST(TestI420BufferPool, FailToReuse) { 38 TEST(TestI420BufferPool, FailToReuse) {
39 I420BufferPool pool; 39 I420BufferPool pool;
40 rtc::scoped_refptr<VideoFrameBuffer> buffer = pool.CreateBuffer(16, 16); 40 rtc::scoped_refptr<I420BufferInterface> buffer = pool.CreateBuffer(16, 16);
41 // Extract non-refcounted pointers for testing. 41 // Extract non-refcounted pointers for testing.
42 const uint8_t* u_ptr = buffer->DataU(); 42 const uint8_t* u_ptr = buffer->DataU();
43 const uint8_t* v_ptr = buffer->DataV(); 43 const uint8_t* v_ptr = buffer->DataV();
44 // Release buffer so that it is returned to the pool. 44 // Release buffer so that it is returned to the pool.
45 buffer = nullptr; 45 buffer = nullptr;
46 // Check that the pool doesn't try to reuse buffers of incorrect size. 46 // Check that the pool doesn't try to reuse buffers of incorrect size.
47 buffer = pool.CreateBuffer(32, 16); 47 buffer = pool.CreateBuffer(32, 16);
48 EXPECT_EQ(32, buffer->width()); 48 EXPECT_EQ(32, buffer->width());
49 EXPECT_EQ(16, buffer->height()); 49 EXPECT_EQ(16, buffer->height());
50 EXPECT_NE(u_ptr, buffer->DataU()); 50 EXPECT_NE(u_ptr, buffer->DataU());
51 EXPECT_NE(v_ptr, buffer->DataV()); 51 EXPECT_NE(v_ptr, buffer->DataV());
52 } 52 }
53 53
54 TEST(TestI420BufferPool, FrameValidAfterPoolDestruction) { 54 TEST(TestI420BufferPool, FrameValidAfterPoolDestruction) {
55 rtc::scoped_refptr<I420Buffer> buffer; 55 rtc::scoped_refptr<I420Buffer> buffer;
56 { 56 {
57 I420BufferPool pool; 57 I420BufferPool pool;
58 buffer = pool.CreateBuffer(16, 16); 58 buffer = pool.CreateBuffer(16, 16);
59 } 59 }
60 EXPECT_EQ(16, buffer->width()); 60 EXPECT_EQ(16, buffer->width());
61 EXPECT_EQ(16, buffer->height()); 61 EXPECT_EQ(16, buffer->height());
62 // Try to trigger use-after-free errors by writing to y-plane. 62 // Try to trigger use-after-free errors by writing to y-plane.
63 memset(buffer->MutableDataY(), 0xA5, 16 * buffer->StrideY()); 63 memset(buffer->MutableDataY(), 0xA5, 16 * buffer->StrideY());
64 } 64 }
65 65
66 TEST(TestI420BufferPool, MaxNumberOfBuffers) { 66 TEST(TestI420BufferPool, MaxNumberOfBuffers) {
67 I420BufferPool pool(false, 1); 67 I420BufferPool pool(false, 1);
68 rtc::scoped_refptr<VideoFrameBuffer> buffer1 = pool.CreateBuffer(16, 16); 68 rtc::scoped_refptr<I420BufferInterface> buffer1 = pool.CreateBuffer(16, 16);
69 EXPECT_NE(nullptr, buffer1.get()); 69 EXPECT_NE(nullptr, buffer1.get());
70 EXPECT_EQ(nullptr, pool.CreateBuffer(16, 16).get()); 70 EXPECT_EQ(nullptr, pool.CreateBuffer(16, 16).get());
71 } 71 }
72 72
73 } // namespace webrtc 73 } // namespace webrtc
OLDNEW
« no previous file with comments | « webrtc/api/video/video_frame_buffer.cc ('k') | webrtc/common_video/i420_video_frame_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698