| // Copyright 2023 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <string> |
| #include <string_view> |
| |
| #include "base/functional/bind.h" |
| #include "base/run_loop.h" |
| #include "base/test/task_environment.h" |
| #include "media/formats/mp4/writable_box_definitions.h" |
| #include "media/muxers/mp4_muxer_context.h" |
| #include "media/muxers/output_position_tracker.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace media { |
| |
| TEST(Mp4MuxerContextTest, Default) { |
| base::test::SingleThreadTaskEnvironment task_environment; |
| std::string written_data; |
| base::RunLoop run_loop; |
| |
| auto output_position_tracker = |
| std::make_unique<OutputPositionTracker>(base::BindRepeating( |
| [](std::string* written_data, base::OnceClosure run_loop_quit, |
| std::string_view data) { |
| written_data->append(data); |
| std::move(run_loop_quit).Run(); |
| }, |
| &written_data, run_loop.QuitClosure())); |
| |
| Mp4MuxerContext mp4_context(std::move(output_position_tracker)); |
| EXPECT_FALSE(mp4_context.GetVideoTrack().has_value()); |
| EXPECT_FALSE(mp4_context.GetAudioTrack().has_value()); |
| |
| mp4_context.SetVideoTrack({1, 1111}); |
| mp4_context.SetAudioTrack({2, 2222}); |
| |
| std::string str1 = "abc"; |
| mp4_context.GetOutputPositionTracker().WriteString(str1); |
| run_loop.Run(); |
| |
| EXPECT_TRUE(mp4_context.GetVideoTrack().has_value()); |
| EXPECT_TRUE(mp4_context.GetAudioTrack().has_value()); |
| EXPECT_EQ(mp4_context.GetVideoTrack().value().index, 1u); |
| EXPECT_EQ(mp4_context.GetAudioTrack().value().index, 2u); |
| |
| EXPECT_TRUE(mp4_context.GetVideoTrack().has_value()); |
| EXPECT_TRUE(mp4_context.GetAudioTrack().has_value()); |
| EXPECT_EQ(mp4_context.GetVideoTrack().value().timescale, 1111u); |
| EXPECT_EQ(mp4_context.GetAudioTrack().value().timescale, 2222u); |
| EXPECT_EQ(str1, written_data); |
| } |
| |
| } // namespace media |