123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116 |
- // Copyright 2017 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.
- #include "media/filters/frame_buffer_pool.h"
- #include "base/test/simple_test_tick_clock.h"
- #include "base/test/test_message_loop.h"
- #include "testing/gtest/include/gtest/gtest.h"
- namespace media {
- constexpr size_t kBufferSize = 1024;
- TEST(FrameBufferPool, BasicFunctionality) {
- base::TestMessageLoop message_loop;
- scoped_refptr<FrameBufferPool> pool = new FrameBufferPool();
- void* priv1 = nullptr;
- uint8_t* buf1 = pool->GetFrameBuffer(kBufferSize, &priv1);
- ASSERT_TRUE(priv1);
- ASSERT_TRUE(buf1);
- memset(buf1, 0, kBufferSize);
- void* priv2 = nullptr;
- uint8_t* buf2 = pool->GetFrameBuffer(kBufferSize, &priv2);
- ASSERT_TRUE(priv2);
- ASSERT_TRUE(buf2);
- EXPECT_NE(priv1, priv2);
- EXPECT_NE(buf1, buf2);
- memset(buf2, 0, kBufferSize);
- uint8_t* alpha = pool->AllocateAlphaPlaneForFrameBuffer(kBufferSize, priv1);
- ASSERT_TRUE(alpha);
- EXPECT_NE(alpha, buf1);
- EXPECT_NE(alpha, buf2);
- memset(alpha, 0, kBufferSize);
- EXPECT_EQ(2u, pool->get_pool_size_for_testing());
- // Frames are not released immediately, so this should still show two frames.
- pool->ReleaseFrameBuffer(priv2);
- priv2 = buf2 = nullptr;
- EXPECT_EQ(2u, pool->get_pool_size_for_testing());
- auto frame_release_cb = pool->CreateFrameCallback(priv1);
- // Shutdown should release all memory that's not held by VideoFrames.
- pool->Shutdown();
- EXPECT_EQ(1u, pool->get_pool_size_for_testing());
- memset(buf1, 0, kBufferSize);
- memset(alpha, 0, kBufferSize);
- // This will release all memory since we're in the shutdown state.
- std::move(frame_release_cb).Run();
- EXPECT_EQ(0u, pool->get_pool_size_for_testing());
- }
- TEST(FrameBufferPool, ForceAllocationError) {
- base::TestMessageLoop message_loop;
- scoped_refptr<FrameBufferPool> pool = new FrameBufferPool();
- pool->force_allocation_error_for_testing();
- void* priv1 = nullptr;
- uint8_t* buf1 = pool->GetFrameBuffer(kBufferSize, &priv1);
- ASSERT_FALSE(priv1);
- ASSERT_FALSE(buf1);
- pool->Shutdown();
- }
- TEST(FrameBufferPool, DeferredDestruction) {
- base::TestMessageLoop message_loop;
- scoped_refptr<FrameBufferPool> pool = new FrameBufferPool();
- base::SimpleTestTickClock test_clock;
- pool->set_tick_clock_for_testing(&test_clock);
- void* priv1 = nullptr;
- uint8_t* buf1 = pool->GetFrameBuffer(kBufferSize, &priv1);
- void* priv2 = nullptr;
- uint8_t* buf2 = pool->GetFrameBuffer(kBufferSize, &priv2);
- void* priv3 = nullptr;
- uint8_t* buf3 = pool->GetFrameBuffer(kBufferSize, &priv3);
- EXPECT_EQ(3u, pool->get_pool_size_for_testing());
- auto frame_release_cb = pool->CreateFrameCallback(priv1);
- pool->ReleaseFrameBuffer(priv1);
- priv1 = buf1 = nullptr;
- std::move(frame_release_cb).Run();
- // Frame buffers should not be immediately deleted upon return.
- EXPECT_EQ(3u, pool->get_pool_size_for_testing());
- // Advance some time, but not enough to trigger expiration.
- test_clock.Advance(base::Seconds(FrameBufferPool::kStaleFrameLimitSecs / 2));
- // We should still have 3 frame buffers in the pool at this point.
- frame_release_cb = pool->CreateFrameCallback(priv2);
- pool->ReleaseFrameBuffer(priv2);
- priv2 = buf2 = nullptr;
- std::move(frame_release_cb).Run();
- EXPECT_EQ(3u, pool->get_pool_size_for_testing());
- test_clock.Advance(base::Seconds(FrameBufferPool::kStaleFrameLimitSecs + 1));
- // All but this most recently released frame should remain now.
- frame_release_cb = pool->CreateFrameCallback(priv3);
- pool->ReleaseFrameBuffer(priv3);
- priv3 = buf3 = nullptr;
- std::move(frame_release_cb).Run();
- EXPECT_EQ(1u, pool->get_pool_size_for_testing());
- pool->Shutdown();
- }
- } // namespace media
|