// Copyright 2015 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

#ifndef MPD_BASE_MOCK_MPD_NOTIFIER_H_
#define MPD_BASE_MOCK_MPD_NOTIFIER_H_

#include <gmock/gmock.h>

#include <packager/mpd/base/content_protection_element.h>
#include <packager/mpd/base/media_info.pb.h>
#include <packager/mpd/base/mpd_notifier.h>

namespace shaka {

class MockMpdNotifier : public MpdNotifier {
 public:
  explicit MockMpdNotifier(const MpdOptions& mpd_options);
  virtual ~MockMpdNotifier();

  MOCK_METHOD0(Init, bool());
  MOCK_METHOD2(NotifyNewContainer,
               bool(const MediaInfo& media_info, uint32_t* container_id));
  MOCK_METHOD2(NotifySampleDuration,
               bool(uint32_t container_id, int32_t sample_duration));
  MOCK_METHOD5(NotifyNewSegment,
               bool(uint32_t container_id,
                    int64_t start_time,
                    int64_t duration,
                    uint64_t size,
                    int64_t segment_number));
  MOCK_METHOD3(NotifyCompletedSegment,
               bool(uint32_t container_id, int64_t duration, uint64_t size));
  MOCK_METHOD1(NotifyAvailabilityTimeOffset, bool(uint32_t container_id));
  MOCK_METHOD1(NotifySegmentDuration, bool(uint32_t container_id));
  MOCK_METHOD2(NotifyCueEvent, bool(uint32_t container_id, int64_t timestamp));
  MOCK_METHOD4(NotifyEncryptionUpdate,
               bool(uint32_t container_id,
                    const std::string& drm_uuid,
                    const std::vector<uint8_t>& new_key_id,
                    const std::vector<uint8_t>& new_pssh));
  MOCK_METHOD2(NotifyMediaInfoUpdate,
               bool(uint32_t container_id, const MediaInfo& media_info));
  MOCK_METHOD0(Flush, bool());
};

}  // namespace shaka

#endif  // MPD_BASE_MOCK_MPD_NOTIFIER_H_