// Copyright (c) 2013 Google Inc. 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/mp4/composition_offset_iterator.h"

#include "base/memory/scoped_ptr.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace media {
namespace mp4 {

const CompositionOffset kCompositionOffsets[] =
    {{10, 8}, {9, 5}, {25, 7}, {48, 63}, {8, 2}};

class CompositionOffsetIteratorTest : public testing::Test {
 public:
  CompositionOffsetIteratorTest() {
    // Build composition offset table from kCompositionOffsets.
    uint32 composition_offset = 0;
    uint32 length = sizeof(kCompositionOffsets) / sizeof(CompositionOffset);
    for (uint32 i = 0; i < length; ++i) {
      for (uint32 j = 0; j < kCompositionOffsets[i].sample_count; ++j) {
        composition_offset_table_.push_back(
            kCompositionOffsets[i].sample_offset);
      }
    }

    composition_time_to_sample_.composition_offset.assign(
        kCompositionOffsets, kCompositionOffsets + length);
    composition_offset_iterator_.reset(
        new CompositionOffsetIterator(composition_time_to_sample_));
  }

 protected:
  std::vector<uint32> composition_offset_table_;
  CompositionTimeToSample composition_time_to_sample_;
  scoped_ptr<CompositionOffsetIterator> composition_offset_iterator_;

 private:
  DISALLOW_COPY_AND_ASSIGN(CompositionOffsetIteratorTest);
};

TEST_F(CompositionOffsetIteratorTest, EmptyCompositionTime) {
  CompositionTimeToSample composition_time_to_sample;
  CompositionOffsetIterator iterator(composition_time_to_sample);
  EXPECT_FALSE(iterator.IsValid());
  EXPECT_EQ(0, iterator.NumSamples());
}

TEST_F(CompositionOffsetIteratorTest, NumSamples) {
  ASSERT_EQ(composition_offset_table_.size(),
            composition_offset_iterator_->NumSamples());
}

TEST_F(CompositionOffsetIteratorTest, AdvanceSample) {
  ASSERT_EQ(composition_offset_table_[0],
            composition_offset_iterator_->sample_offset());
  for (uint32 sample = 1; sample < composition_offset_table_.size(); ++sample) {
    ASSERT_TRUE(composition_offset_iterator_->AdvanceSample());
    ASSERT_EQ(composition_offset_table_[sample],
              composition_offset_iterator_->sample_offset());
    ASSERT_TRUE(composition_offset_iterator_->IsValid());
  }
  ASSERT_FALSE(composition_offset_iterator_->AdvanceSample());
  ASSERT_FALSE(composition_offset_iterator_->IsValid());
}

TEST_F(CompositionOffsetIteratorTest, SampleOffset) {
  for (uint32 sample = 0; sample < composition_offset_table_.size(); ++sample) {
    ASSERT_EQ(composition_offset_table_[sample],
              composition_offset_iterator_->SampleOffset(sample+1));
  }
}

}  // namespace mp4
}  // namespace media