// 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/cluster_parser.h"

#include "gmock/gmock.h"
#include "gtest/gtest.h"

#include "test_utils/element_parser_test.h"
#include "webm/element.h"
#include "webm/id.h"

using testing::_;
using testing::DoAll;
using testing::InSequence;
using testing::NotNull;
using testing::Return;
using testing::SetArgPointee;

using webm::Action;
using webm::Ancestory;
using webm::BlockGroup;
using webm::Cluster;
using webm::ClusterParser;
using webm::ElementMetadata;
using webm::ElementParserTest;
using webm::Id;
using webm::SimpleBlock;
using webm::Status;

namespace {

class ClusterParserTest
    : public ElementParserTest<ClusterParser, Id::kCluster> {};

TEST_F(ClusterParserTest, DefaultParse) {
  {
    InSequence dummy;

    EXPECT_CALL(callback_, OnClusterBegin(metadata_, Cluster{}, NotNull()))
        .Times(1);

    EXPECT_CALL(callback_, OnClusterEnd(metadata_, Cluster{})).Times(1);
  }

  ParseAndVerify();
}

TEST_F(ClusterParserTest, DefaultValues) {
  SetReaderData({
      0xE7,  // ID = 0xE7 (Timecode).
      0x40, 0x00,  // Size = 0.

      0xAB,  // ID = 0xAB (PrevSize).
      0x40, 0x00,  // Size = 0.

      0xA3,  // ID = 0xA3 (SimpleBlock).
      0x85,  // Size = 5.
      0x81,  // Track number = 1.
      0x00, 0x00,  // Timecode = 0.
      0x00,  // Flags = 0.
      0x00,  // Frame 0.

      0xA0,  // ID = 0xA0 (BlockGroup).
      0x40, 0x00,  // Size = 0.
  });

  {
    InSequence dummy;

    Cluster cluster{};
    cluster.timecode.Set(0, true);
    cluster.previous_size.Set(0, true);

    EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull()))
        .Times(1);

    SimpleBlock simple_block{};
    simple_block.track_number = 1;
    simple_block.num_frames = 1;
    simple_block.is_visible = true;
    cluster.simple_blocks.emplace_back(simple_block, true);

    BlockGroup block_group{};
    cluster.block_groups.emplace_back(block_group, true);

    EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1);
  }

  ParseAndVerify();
}

TEST_F(ClusterParserTest, CustomValues) {
  SetReaderData({
      0xE7,  // ID = 0xE7 (Timecode).
      0x40, 0x01,  // Size = 1.
      0x01,  // Body (value = 1).

      0xAB,  // ID = 0xAB (PrevSize).
      0x40, 0x01,  // Size = 1.
      0x02,  // Body (value = 2).

      0xA3,  // ID = 0xA3 (SimpleBlock).
      0x85,  // Size = 5.
      0x81,  // Track number = 1.
      0x00, 0x00,  // Timecode = 0.
      0x00,  // Flags = 0.
      0x00,  // Frame 0.

      0xA3,  // ID = 0xA3 (SimpleBlock).
      0x85,  // Size = 5.
      0x82,  // Track number = 2.
      0x00, 0x00,  // Timecode = 0.
      0x00,  // Flags = 0.
      0x00,  // Frame 0.

      0xA0,  // ID = 0xA0 (BlockGroup).
      0x40, 0x04,  // Size = 4.

      0x9B,  //   ID = 0x9B (BlockDuration).
      0x40, 0x01,  //   Size = 1.
      0x01,  //   Body (value = 1).

      0xA0,  // ID = 0xA0 (BlockGroup).
      0x40, 0x04,  // Size = 4.

      0x9B,  //   ID = 0x9B (BlockDuration).
      0x40, 0x01,  //   Size = 1.
      0x02,  //   Body (value = 2).
  });

  {
    InSequence dummy;

    Cluster cluster{};
    cluster.timecode.Set(1, true);
    cluster.previous_size.Set(2, true);

    EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull()))
        .Times(1);

    SimpleBlock simple_block{};
    simple_block.num_frames = 1;
    simple_block.is_visible = true;

    simple_block.track_number = 1;
    cluster.simple_blocks.emplace_back(simple_block, true);
    simple_block.track_number = 2;
    cluster.simple_blocks.emplace_back(simple_block, true);

    BlockGroup block_group{};
    block_group.duration.Set(1, true);
    cluster.block_groups.emplace_back(block_group, true);
    block_group.duration.Set(2, true);
    cluster.block_groups.emplace_back(block_group, true);

    EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1);
  }

  ParseAndVerify();
}

TEST_F(ClusterParserTest, SkipOnClusterBegin) {
  {
    InSequence dummy;

    EXPECT_CALL(callback_, OnClusterBegin(_, _, _)).Times(0);

    EXPECT_CALL(callback_, OnBlockGroupBegin(_, NotNull())).Times(1);

    EXPECT_CALL(callback_, OnClusterEnd(_, _)).Times(1);
  }

  ElementMetadata child_metadata = {Id::kBlockGroup, 0, 0, 0};

  Ancestory ancestory;
  ASSERT_TRUE(Ancestory::ById(child_metadata.id, &ancestory));
  // Skip the Segment and Cluster ancestors.
  ancestory = ancestory.next().next();

  parser_.InitAfterSeek(ancestory, child_metadata);

  std::uint64_t num_bytes_read = 0;
  const Status status = parser_.Feed(&callback_, &reader_, &num_bytes_read);
  EXPECT_EQ(Status::kOkCompleted, status.code);
  EXPECT_EQ(reader_.size(), num_bytes_read);
}

TEST_F(ClusterParserTest, SkipSimpleBlock) {
  SetReaderData({
      0xA3,  // ID = 0xA3 (SimpleBlock).
      0x85,  // Size = 5.
      0x81,  // Track number = 1.
      0x00, 0x00,  // Timecode = 0.
      0x00,  // Flags = 0.
      0x00,  // Frame 0.
  });

  {
    InSequence dummy;

    Cluster cluster{};
    EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull()))
        .Times(1);

    SimpleBlock simple_block{};
    simple_block.num_frames = 1;
    simple_block.is_visible = true;
    simple_block.track_number = 1;

    EXPECT_CALL(callback_, OnSimpleBlockBegin(_, simple_block, NotNull()))
        .WillOnce(DoAll(SetArgPointee<2>(Action::kSkip),
                        Return(Status(Status::kOkCompleted))));

    EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1);
  }

  ParseAndVerify();
}

TEST_F(ClusterParserTest, SkipBlockGroup) {
  SetReaderData({
      0xA0,  // ID = 0xA0 (BlockGroup).
      0x40, 0x04,  // Size = 4.

      0x9B,  //   ID = 0x9B (BlockDuration).
      0x40, 0x01,  //   Size = 1.
      0x01,  //   Body (value = 1).
  });

  {
    InSequence dummy;

    Cluster cluster{};
    EXPECT_CALL(callback_, OnClusterBegin(metadata_, cluster, NotNull()))
        .Times(1);

    EXPECT_CALL(callback_, OnBlockGroupBegin(_, NotNull()))
        .WillOnce(DoAll(SetArgPointee<1>(Action::kSkip),
                        Return(Status(Status::kOkCompleted))));

    EXPECT_CALL(callback_, OnClusterEnd(metadata_, cluster)).Times(1);
  }

  ParseAndVerify();
}

}  // namespace