// Copyright 2014 Google LLC. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

#include <packager/mpd/base/mpd_builder.h>

#include <memory>

#include <gmock/gmock.h>
#include <gtest/gtest.h>

#include <packager/mpd/base/adaptation_set.h>
#include <packager/mpd/base/period.h>
#include <packager/mpd/base/representation.h>
#include <packager/mpd/test/mpd_builder_test_helper.h>
#include <packager/utils/clock.h>
#include <packager/utils/test_clock.h>
#include <packager/version/version.h>

using ::testing::HasSubstr;

namespace shaka {

template <DashProfile profile>
class MpdBuilderTest : public ::testing::Test {
 public:
  MpdBuilderTest() : mpd_(MpdOptions()), representation_() {
    mpd_.mpd_options_.dash_profile = profile;
  }
  ~MpdBuilderTest() override {}

  void SetUp() override {
    const double kPeriodStartTimeSeconds = 0.0;
    period_ = mpd_.GetOrCreatePeriod(kPeriodStartTimeSeconds);
    ASSERT_TRUE(period_);
  }

  MpdOptions* mutable_mpd_options() { return &mpd_.mpd_options_; }

  void CheckMpd(const std::string& expected_output_file) {
    std::string mpd_doc;
    ASSERT_TRUE(mpd_.ToString(&mpd_doc));
    ASSERT_TRUE(ValidateMpdSchema(mpd_doc));

    ASSERT_NO_FATAL_FAILURE(
        ExpectMpdToEqualExpectedOutputFile(mpd_doc, expected_output_file));
  }

  void AddSegmentToPeriod(double segment_start_time_seconds,
                          double segment_duration_seconds,
                          Period* period) {
    MediaInfo media_info = GetTestMediaInfo(kFileNameVideoMediaInfo1);
    // Not relevant in this test.
    const bool kContentProtectionFlag = true;
    const size_t kBytes = 1000;

    AdaptationSet* adaptation_set =
        period->GetOrCreateAdaptationSet(media_info, kContentProtectionFlag);
    Representation* representation =
        adaptation_set->AddRepresentation(media_info);
    representation->AddNewSegment(
        segment_start_time_seconds * media_info.reference_time_scale(),
        segment_duration_seconds * media_info.reference_time_scale(), kBytes);
  }

 protected:
  // Creates a new AdaptationSet and adds a Representation element using
  // |media_info|.
  void AddRepresentation(const MediaInfo& media_info) {
    AdaptationSet* adaptation_set =
        period_->GetOrCreateAdaptationSet(media_info, true);
    ASSERT_TRUE(adaptation_set);

    Representation* representation =
        adaptation_set->AddRepresentation(media_info);
    ASSERT_TRUE(representation);

    representation_ = representation;
  }

  MpdBuilder mpd_;

  // We usually need only one representation.
  Representation* representation_;  // Owned by |mpd_|.

 private:
  Period* period_ = nullptr;

  DISALLOW_COPY_AND_ASSIGN(MpdBuilderTest);
};

class OnDemandMpdBuilderTest : public MpdBuilderTest<DashProfile::kOnDemand> {};

class LiveMpdBuilderTest : public MpdBuilderTest<DashProfile::kLive> {
 public:
  ~LiveMpdBuilderTest() override {}

  // Anchors availabilityStartTime so that the test result doesn't depend on the
  // current time.
  void SetUp() override {
    SetPackagerVersionForTesting("<tag>-<hash>-<test>");
    mpd_.mpd_options_.mpd_type = MpdType::kDynamic;
    mpd_.availability_start_time_ = "2011-12-25T12:30:00";
    InjectTestClock();
  }

  // Injects a clock that always returns 2016 Jan 11 15:10:24 in UTC.
  void InjectTestClock() {
    mpd_.InjectClockForTesting(
        std::make_unique<TestClock>("2016-01-11T15:10:24"));
  }
};

// Add one video check the output.
TEST_F(OnDemandMpdBuilderTest, Video) {
  MediaInfo video_media_info = GetTestMediaInfo(kFileNameVideoMediaInfo1);
  ASSERT_NO_FATAL_FAILURE(AddRepresentation(video_media_info));
  EXPECT_NO_FATAL_FAILURE(CheckMpd(kFileNameExpectedMpdOutputVideo1));
}

TEST_F(OnDemandMpdBuilderTest, TwoVideosWithDifferentResolutions) {
  MediaInfo media_info1 = GetTestMediaInfo(kFileNameVideoMediaInfo1);
  MediaInfo media_info2 = GetTestMediaInfo(kFileNameVideoMediaInfo2);
  // The order matters here to check against expected output.
  ASSERT_NO_FATAL_FAILURE(AddRepresentation(media_info1));
  ASSERT_NO_FATAL_FAILURE(AddRepresentation(media_info2));

  EXPECT_NO_FATAL_FAILURE(CheckMpd(kFileNameExpectedMpdOutputVideo1And2));
}

// Add both video and audio and check the output.
TEST_F(OnDemandMpdBuilderTest, VideoAndAudio) {
  MediaInfo video_media_info = GetTestMediaInfo(kFileNameVideoMediaInfo1);
  MediaInfo audio_media_info = GetTestMediaInfo(kFileNameAudioMediaInfo1);
  // The order matters here to check against expected output.
  ASSERT_NO_FATAL_FAILURE(AddRepresentation(video_media_info));
  ASSERT_NO_FATAL_FAILURE(AddRepresentation(audio_media_info));

  EXPECT_NO_FATAL_FAILURE(CheckMpd(kFileNameExpectedMpdOutputAudio1AndVideo1));
}

TEST_F(OnDemandMpdBuilderTest, CheckXmlTest) {
  const double kPeriod1StartTimeSeconds = 0.0;

  // Actual period duration is determined by the segments not by the period
  // start time above, which only provides an anchor point.
  const double kPeriod1SegmentStartSeconds = 0.2;
  const double kPeriod1SegmentDurationSeconds = 3.0;

  Period* period = mpd_.GetOrCreatePeriod(kPeriod1StartTimeSeconds);
  AddSegmentToPeriod(kPeriod1SegmentStartSeconds,
                     kPeriod1SegmentDurationSeconds, period);

  std::string mpd_doc;
  ASSERT_TRUE(mpd_.ToString(&mpd_doc));
  EXPECT_THAT(mpd_doc, HasSubstr("<Period id=\"0\">\n"));
  EXPECT_THAT(mpd_doc,
              HasSubstr("<SegmentBase indexRange=\"121-221\""
                        " timescale=\"1000\" presentationTimeOffset=\"200\">"));
}

TEST_F(OnDemandMpdBuilderTest, MultiplePeriodTest) {
  const double kPeriodStartTimeSeconds = 1.0;
  Period* period = mpd_.GetOrCreatePeriod(kPeriodStartTimeSeconds);
  ASSERT_TRUE(period);
  ASSERT_EQ(kPeriodStartTimeSeconds, period->start_time_in_seconds());

  const double kPeriodStartTimeSeconds2 = 1.1;
  Period* period2 = mpd_.GetOrCreatePeriod(kPeriodStartTimeSeconds2);
  ASSERT_TRUE(period2);
  // The old Period is re-used if they are closed to each other.
  ASSERT_EQ(period, period2);
  ASSERT_EQ(kPeriodStartTimeSeconds, period2->start_time_in_seconds());

  const double kPeriodStartTimeSeconds3 = 5.0;
  Period* period3 = mpd_.GetOrCreatePeriod(kPeriodStartTimeSeconds3);
  ASSERT_TRUE(period3);
  ASSERT_NE(period, period3);
  ASSERT_EQ(kPeriodStartTimeSeconds3, period3->start_time_in_seconds());
}

TEST_F(OnDemandMpdBuilderTest, MultiplePeriodCheckXmlTest) {
  const double kPeriod1StartTimeSeconds = 0.0;
  const double kPeriod2StartTimeSeconds = 3.1;
  const double kPeriod3StartTimeSeconds = 8.0;

  // Actual period duration is determined by the segments not by the period
  // start time above, which only provides an anchor point.
  const double kPeriod1SegmentStartSeconds = 0.2;
  const double kPeriod1SegmentDurationSeconds = 3.0;
  const double kPeriod2SegmentStartSeconds = 5.5;
  const double kPeriod2SegmentDurationSeconds = 10.5;
  const double kPeriod3SegmentStartSeconds = 1.5;
  const double kPeriod3SegmentDurationSeconds = 10.0;

  Period* period = mpd_.GetOrCreatePeriod(kPeriod1StartTimeSeconds);
  AddSegmentToPeriod(kPeriod1SegmentStartSeconds,
                     kPeriod1SegmentDurationSeconds, period);

  period = mpd_.GetOrCreatePeriod(kPeriod2StartTimeSeconds);
  AddSegmentToPeriod(kPeriod2SegmentStartSeconds,
                     kPeriod2SegmentDurationSeconds, period);

  period = mpd_.GetOrCreatePeriod(kPeriod3StartTimeSeconds);
  AddSegmentToPeriod(kPeriod3SegmentStartSeconds,
                     kPeriod3SegmentDurationSeconds, period);

  std::string mpd_doc;
  ASSERT_TRUE(mpd_.ToString(&mpd_doc));
  EXPECT_THAT(mpd_doc, HasSubstr("<Period id=\"0\" duration=\"PT3S\">\n"));
  EXPECT_THAT(mpd_doc,
              HasSubstr("<SegmentBase indexRange=\"121-221\""
                        " timescale=\"1000\" presentationTimeOffset=\"200\">"));
  EXPECT_THAT(mpd_doc, HasSubstr("<Period id=\"1\" duration=\"PT10.5S\">\n"));
  EXPECT_THAT(
      mpd_doc,
      HasSubstr("<SegmentBase indexRange=\"121-221\""
                " timescale=\"1000\" presentationTimeOffset=\"5500\">"));
  EXPECT_THAT(mpd_doc, HasSubstr("<Period id=\"2\" duration=\"PT10S\">\n"));
  EXPECT_THAT(
      mpd_doc,
      HasSubstr("<SegmentBase indexRange=\"121-221\""
                " timescale=\"1000\" presentationTimeOffset=\"1500\">"));
}

TEST_F(LiveMpdBuilderTest, MultiplePeriodCheckXmlTest) {
  const double kPeriod1StartTimeSeconds = 0.0;
  const double kPeriod2StartTimeSeconds = 3.1;
  const double kPeriod3StartTimeSeconds = 8.0;
  mpd_.GetOrCreatePeriod(kPeriod1StartTimeSeconds);
  mpd_.GetOrCreatePeriod(kPeriod2StartTimeSeconds);
  mpd_.GetOrCreatePeriod(kPeriod3StartTimeSeconds);

  std::string mpd_doc;
  ASSERT_TRUE(mpd_.ToString(&mpd_doc));
  EXPECT_THAT(mpd_doc, HasSubstr("  <Period id=\"0\" start=\"PT0S\"/>\n"
                                 "  <Period id=\"1\" start=\"PT3.1S\"/>\n"
                                 "  <Period id=\"2\" start=\"PT8S\"/>\n"));
}

// Check whether the attributes are set correctly for dynamic <MPD> element.
// This test must use ASSERT_EQ for comparison because XmlEqual() cannot
// handle namespaces correctly yet.
TEST_F(LiveMpdBuilderTest, DynamicCheckMpdAttributes) {
  static const char kExpectedOutput[] =
      "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
      "<!--Generated with https://github.com/shaka-project/shaka-packager"
      " version <tag>-<hash>-<test>-->\n"
      "<MPD xmlns=\"urn:mpeg:dash:schema:mpd:2011\""
      " xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\""
      " xsi:schemaLocation=\"urn:mpeg:dash:schema:mpd:2011 DASH-MPD.xsd\""
      " profiles=\"urn:mpeg:dash:profile:isoff-live:2011\""
      " minBufferTime=\"PT2S\""
      " type=\"dynamic\""
      " publishTime=\"2016-01-11T15:10:24Z\""
      " availabilityStartTime=\"2011-12-25T12:30:00\""
      " minimumUpdatePeriod=\"PT2S\">\n"
      "  <UTCTiming schemeIdUri=\"urn:mpeg:dash:utc:http-xsdate:2014\" "
      "value=\"http://foo.bar/my_body_is_the_current_date_and_time\"/>\n"
      "  <UTCTiming schemeIdUri=\"urn:mpeg:dash:utc:http-head:2014\" "
      "value=\"http://foo.bar/check_me_for_the_date_header\"/>\n"
      "</MPD>\n";

  std::string mpd_doc;
  mutable_mpd_options()->mpd_type = MpdType::kDynamic;
  mutable_mpd_options()->mpd_params.minimum_update_period = 2;
  mutable_mpd_options()->mpd_params.utc_timings = {
      {"urn:mpeg:dash:utc:http-xsdate:2014",
       "http://foo.bar/my_body_is_the_current_date_and_time"},
      {"urn:mpeg:dash:utc:http-head:2014",
       "http://foo.bar/check_me_for_the_date_header"}};
  ASSERT_TRUE(mpd_.ToString(&mpd_doc));
  ASSERT_EQ(kExpectedOutput, mpd_doc);
}

TEST_F(LiveMpdBuilderTest, StaticCheckMpdAttributes) {
  static const char kExpectedOutput[] =
      "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
      "<!--Generated with https://github.com/shaka-project/shaka-packager"
      " version <tag>-<hash>-<test>-->\n"
      "<MPD xmlns=\"urn:mpeg:dash:schema:mpd:2011\""
      " xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\""
      " xsi:schemaLocation=\"urn:mpeg:dash:schema:mpd:2011 DASH-MPD.xsd\""
      " profiles=\"urn:mpeg:dash:profile:isoff-live:2011\""
      " minBufferTime=\"PT2S\""
      " type=\"static\""
      " mediaPresentationDuration=\"PT0S\"/>\n";

  std::string mpd_doc;
  mutable_mpd_options()->mpd_type = MpdType::kStatic;

  // Ignored in static MPD.
  mutable_mpd_options()->mpd_params.minimum_update_period = 2;
  mutable_mpd_options()->mpd_params.utc_timings = {
      {"urn:mpeg:dash:utc:http-xsdate:2014",
       "http://foo.bar/my_body_is_the_current_date_and_time"},
      {"urn:mpeg:dash:utc:http-head:2014",
       "http://foo.bar/check_me_for_the_date_header"}};

  ASSERT_TRUE(mpd_.ToString(&mpd_doc));
  ASSERT_EQ(kExpectedOutput, mpd_doc);
}

namespace {
const char kMediaFile[] = "foo/bar/media.mp4";
const char kMediaFileBase[] = "media.mp4";
const char kInitSegment[] = "foo/bar/init.mp4";
const char kInitSegmentBase[] = "init.mp4";
const char kSegmentTemplate[] = "foo/bar/segment-$Number$.mp4";
const char kSegmentTemplateBase[] = "segment-$Number$.mp4";
const char kPathModifiedMpd[] = "foo/bar/media.mpd";
const char kPathNotModifiedMpd[] = "foo/baz/media.mpd";
}  // namespace

TEST(RelativePaths, PathsModified) {
  MediaInfo media_info;

  media_info.set_media_file_name(kMediaFile);
  media_info.set_init_segment_name(kInitSegment);
  media_info.set_segment_template(kSegmentTemplate);
  MpdBuilder::MakePathsRelativeToMpd(kPathModifiedMpd, &media_info);
  EXPECT_EQ(kMediaFileBase, media_info.media_file_url());
  EXPECT_EQ(kInitSegmentBase, media_info.init_segment_url());
  EXPECT_EQ(kSegmentTemplateBase, media_info.segment_template_url());
}

TEST(RelativePaths, PathsNotModified) {
  MediaInfo media_info;

  media_info.set_media_file_name(kMediaFile);
  media_info.set_init_segment_name(kInitSegment);
  media_info.set_segment_template(kSegmentTemplate);
  MpdBuilder::MakePathsRelativeToMpd(kPathNotModifiedMpd, &media_info);
  EXPECT_EQ(kMediaFile, media_info.media_file_url());
  EXPECT_EQ(kInitSegment, media_info.init_segment_url());
  EXPECT_EQ(kSegmentTemplate, media_info.segment_template_url());
}

}  // namespace shaka