Add a new incremental parsing API
Change-Id: I6b921766836d58df0281fb23b2add3f62a478e14
This commit is contained in:
Michael Bradshaw
366
webm_parser/tests/master_value_parser_test.cc
Normal file
366
webm_parser/tests/master_value_parser_test.cc
Normal file
@@ -0,0 +1,366 @@
|
||||
// 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
|
||||
Reference in New Issue
Block a user