| // Copyright (c) 2016 The WebM project authors. All Rights Reserved. |
| // |
| // Use of this source code is governed by a BSD-style license |
| // that can be found in the LICENSE file in the root of the source |
| // tree. An additional intellectual property rights grant can be found |
| // in the file PATENTS. All contributing project authors may |
| // be found in the AUTHORS file in the root of the source tree. |
| #include "src/master_value_parser.h" |
| |
| #include "gmock/gmock.h" |
| #include "gtest/gtest.h" |
| |
| #include "src/int_parser.h" |
| #include "test_utils/element_parser_test.h" |
| #include "webm/element.h" |
| #include "webm/id.h" |
| #include "webm/status.h" |
| |
| using testing::_; |
| using testing::DoAll; |
| using testing::InSequence; |
| using testing::NotNull; |
| using testing::Return; |
| using testing::SetArgPointee; |
| |
| using webm::Action; |
| using webm::Callback; |
| using webm::Cluster; |
| using webm::ElementMetadata; |
| using webm::ElementParserTest; |
| using webm::Id; |
| using webm::MasterValueParser; |
| using webm::Status; |
| using webm::UnsignedIntParser; |
| |
| namespace { |
| |
| // An instance of MasterValueParser that can actually be used in the tests to |
| // parse Cluster structures. |
| class FakeClusterParser : public MasterValueParser<Cluster> { |
| public: |
| FakeClusterParser() |
| : MasterValueParser( |
| MakeChild<UnsignedIntParser>(Id::kTimecode, &Cluster::timecode), |
| MakeChild<UnsignedIntParser>(Id::kPrevSize, &Cluster::previous_size) |
| .UseAsStartEvent()) {} |
| |
| protected: |
| Status OnParseStarted(Callback* callback, Action* action) override { |
| return callback->OnClusterBegin(metadata(Id::kCluster), value(), action); |
| } |
| |
| Status OnParseCompleted(Callback* callback) override { |
| return callback->OnClusterEnd(metadata(Id::kCluster), value()); |
| } |
| }; |
| |
| class MasterValueParserTest |
| : public ElementParserTest<FakeClusterParser, Id::kCluster> {}; |
| |
| TEST_F(MasterValueParserTest, InvalidId) { |
| SetReaderData({ |
| 0x00, // Invalid ID. |
| }); |
| |
| ParseAndExpectResult(Status::kInvalidElementId); |
| } |
| |
| TEST_F(MasterValueParserTest, InvalidSize) { |
| SetReaderData({ |
| 0xE7, // ID = 0xE7 (Timecode). |
| 0xFF, // Size = unknown. |
| }); |
| |
| ParseAndExpectResult(Status::kInvalidElementSize); |
| } |
| |
| TEST_F(MasterValueParserTest, DefaultParse) { |
| { |
| InSequence dummy; |
| |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, Cluster{}, NotNull())) |
| .Times(1); |
| |
| EXPECT_CALL(callback_, OnClusterEnd(metadata_, Cluster{})).Times(1); |
| } |
| |
| ParseAndVerify(); |
| } |
| |
| TEST_F(MasterValueParserTest, DefaultValues) { |
| SetReaderData({ |
| 0xE7, // ID = 0xE7 (Timecode). |
| 0x80, // Size = 0. |
| |
| 0xAB, // ID = 0xAB (PrevSize). |
| 0x80, // Size = 0. |
| }); |
| |
| { |
| InSequence dummy; |
| |
| Cluster cluster{}; |
| cluster.timecode.Set(0, true); |
| |
| ElementMetadata child_metadata = {Id::kTimecode, 2, 0, 0}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| child_metadata = {Id::kPrevSize, 2, 0, 2}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull())) |
| .Times(1); |
| |
| cluster.previous_size.Set(0, true); |
| EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1); |
| } |
| |
| ParseAndVerify(); |
| } |
| |
| TEST_F(MasterValueParserTest, CustomValues) { |
| SetReaderData({ |
| 0xE7, // ID = 0xE7 (Timecode). |
| 0x82, // Size = 2. |
| 0x04, 0x00, // Body (value = 1024). |
| |
| 0xAB, // ID = 0xAB (PrevSize). |
| 0x40, 0x01, // Size = 1. |
| 0x01, // Body (value = 1). |
| }); |
| |
| { |
| InSequence dummy; |
| |
| Cluster cluster{}; |
| cluster.timecode.Set(1024, true); |
| |
| ElementMetadata child_metadata = {Id::kTimecode, 2, 2, 0}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| child_metadata = {Id::kPrevSize, 3, 1, 4}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull())) |
| .Times(1); |
| |
| cluster.previous_size.Set(1, true); |
| EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1); |
| } |
| |
| ParseAndVerify(); |
| } |
| |
| TEST_F(MasterValueParserTest, IncrementalParse) { |
| SetReaderData({ |
| 0xE7, // ID = 0xE7 (Timecode). |
| 0x40, 0x02, // Size = 2. |
| 0x04, 0x00, // Body (value = 1024). |
| |
| 0x40, 0x00, // ID = 0x4000 (Unknown). |
| 0x40, 0x02, // Size = 2. |
| 0x00, 0x00, // Body. |
| |
| 0xAB, // ID = 0xAB (PrevSize). |
| 0x40, 0x02, // Size = 2. |
| 0x00, 0x01, // Body (value = 1). |
| }); |
| |
| { |
| InSequence dummy; |
| |
| Cluster cluster{}; |
| cluster.timecode.Set(1024, true); |
| |
| ElementMetadata child_metadata = {Id::kTimecode, 3, 2, 0}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| child_metadata = {static_cast<Id>(0x4000), 4, 2, 5}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| EXPECT_CALL(callback_, |
| OnUnknownElement(child_metadata, NotNull(), NotNull())) |
| .Times(3); |
| |
| child_metadata = {Id::kPrevSize, 3, 2, 11}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull())) |
| .Times(1); |
| |
| cluster.previous_size.Set(1, true); |
| EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1); |
| } |
| |
| IncrementalParseAndVerify(); |
| } |
| |
| TEST_F(MasterValueParserTest, StartEventWithoutPreviousSize) { |
| SetReaderData({ |
| 0xE7, // ID = 0xE7 (Timecode). |
| 0x80, // Size = 0. |
| }); |
| |
| { |
| InSequence dummy; |
| |
| Cluster cluster{}; |
| |
| const ElementMetadata child_metadata = {Id::kTimecode, 2, 0, 0}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| cluster.timecode.Set(0, true); |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull())) |
| .Times(1); |
| |
| EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1); |
| } |
| |
| ParseAndVerify(); |
| } |
| |
| TEST_F(MasterValueParserTest, StartEventWithPreviousSize) { |
| SetReaderData({ |
| 0xAB, // ID = 0xAB (PrevSize). |
| 0x80, // Size = 0. |
| |
| 0xE7, // ID = 0xE7 (Timecode). |
| 0x80, // Size = 0. |
| }); |
| |
| { |
| InSequence dummy; |
| |
| Cluster cluster{}; |
| |
| ElementMetadata child_metadata = {Id::kPrevSize, 2, 0, 0}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull())) |
| .Times(1); |
| |
| child_metadata = {Id::kTimecode, 2, 0, 2}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| cluster.previous_size.Set(0, true); |
| cluster.timecode.Set(0, true); |
| EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1); |
| } |
| |
| ParseAndVerify(); |
| } |
| |
| TEST_F(MasterValueParserTest, PartialStartThenRead) { |
| SetReaderData({ |
| 0xAB, // ID = 0xAB (PrevSize). |
| 0x81, // Size = 1. |
| 0x01, // Body (value = 1). |
| |
| 0xE7, // ID = 0xE7 (Timecode). |
| 0x81, // Size = 1. |
| 0x01, // Body (value = 1). |
| }); |
| |
| { |
| InSequence dummy; |
| |
| Cluster cluster{}; |
| |
| ElementMetadata child_metadata = {Id::kPrevSize, 2, 1, 0}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull())) |
| .WillOnce(Return(Status(Status::kOkPartial))) |
| .WillOnce(DoAll(SetArgPointee<2>(Action::kRead), |
| Return(Status(Status::kOkCompleted)))); |
| |
| child_metadata = {Id::kTimecode, 2, 1, 3}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| cluster.previous_size.Set(1, true); |
| cluster.timecode.Set(1, true); |
| EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1); |
| } |
| |
| IncrementalParseAndVerify(); |
| } |
| |
| TEST_F(MasterValueParserTest, EmptySkip) { |
| const Cluster cluster{}; |
| |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull())) |
| .WillOnce(DoAll(SetArgPointee<2>(Action::kSkip), |
| Return(Status(Status::kOkCompleted)))); |
| |
| EXPECT_CALL(callback_, OnElementBegin(_, _)).Times(0); |
| EXPECT_CALL(callback_, OnClusterEnd(_, _)).Times(0); |
| |
| ParseAndVerify(); |
| |
| EXPECT_EQ(cluster, parser_.value()); |
| } |
| |
| TEST_F(MasterValueParserTest, Skip) { |
| SetReaderData({ |
| 0xAB, // ID = 0xAB (PrevSize). |
| 0x81, // Size = 1. |
| 0x01, // Body (value = 1). |
| |
| 0xE7, // ID = 0xE7 (Timecode). |
| 0x81, // Size = 1. |
| 0x01, // Body (value = 1). |
| }); |
| |
| const Cluster cluster{}; |
| |
| { |
| InSequence dummy; |
| |
| const ElementMetadata child_metadata = {Id::kPrevSize, 2, 1, 0}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull())) |
| .WillOnce(DoAll(SetArgPointee<2>(Action::kSkip), |
| Return(Status(Status::kOkCompleted)))); |
| |
| EXPECT_CALL(callback_, OnClusterEnd(_, _)).Times(0); |
| } |
| |
| ParseAndVerify(); |
| |
| EXPECT_EQ(cluster, parser_.value()); |
| } |
| |
| TEST_F(MasterValueParserTest, PartialStartThenSkip) { |
| SetReaderData({ |
| 0xAB, // ID = 0xAB (PrevSize). |
| 0x81, // Size = 1. |
| 0x01, // Body (value = 1). |
| |
| 0xE7, // ID = 0xE7 (Timecode). |
| 0x81, // Size = 1. |
| 0x01, // Body (value = 1). |
| }); |
| |
| const Cluster cluster{}; |
| |
| { |
| InSequence dummy; |
| |
| const ElementMetadata child_metadata = {Id::kPrevSize, 2, 1, 0}; |
| EXPECT_CALL(callback_, OnElementBegin(child_metadata, NotNull())).Times(1); |
| |
| EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull())) |
| .WillOnce(Return(Status(Status::kOkPartial))) |
| .WillOnce(DoAll(SetArgPointee<2>(Action::kSkip), |
| Return(Status(Status::kOkCompleted)))); |
| |
| EXPECT_CALL(callback_, OnClusterEnd(_, _)).Times(0); |
| } |
| |
| IncrementalParseAndVerify(); |
| |
| EXPECT_EQ(cluster, parser_.value()); |
| } |
| |
| } // namespace |