mock_media_playlist.h

// Copyright 2016 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 PACKAGER_HLS_BASE_MOCK_MEDIA_PLAYLIST_H_
#define PACKAGER_HLS_BASE_MOCK_MEDIA_PLAYLIST_H_
 
#include <cstdint>
#include <filesystem>
#include <string>
 
#include <gmock/gmock.h>
 
#include <packager/hls/base/media_playlist.h>
#include <packager/mpd/base/media_info.pb.h>
 
namespace shaka {
namespace hls {
 
class MockMediaPlaylist : public MediaPlaylist {
 public:
  // The actual parameters to MediaPlaylist() (parent) constructor doesn't
  // matter because the return value can be mocked.
  MockMediaPlaylist(const std::string& file_name,
                    const std::string& name,
                    const std::string& group_id);
  ~MockMediaPlaylist() override;
 
  MOCK_METHOD1(SetMediaInfo, bool(const MediaInfo& media_info));
  MOCK_METHOD5(AddSegment,
               void(const std::string& file_name,
                    int64_t start_time,
                    int64_t duration,
                    uint64_t start_byte_offset,
                    uint64_t size));
  MOCK_METHOD3(AddKeyFrame,
               void(int64_t timestamp,
                    uint64_t start_byte_offset,
                    uint64_t size));
  MOCK_METHOD6(AddEncryptionInfo,
               void(EncryptionMethod method,
                    const std::string& url,
                    const std::string& key_id,
                    const std::string& iv,
                    const std::string& key_format,
                    const std::string& key_format_versions));
  MOCK_METHOD0(AddPlacementOpportunity, void());
  MOCK_METHOD3(WriteToFile,
               bool(const std::filesystem::path& file_path,
                    bool event_to_vod_on_end_of_stream,
                    bool end_stream));
  MOCK_CONST_METHOD0(MaxBitrate, uint64_t());
  MOCK_CONST_METHOD0(AvgBitrate, uint64_t());
  MOCK_CONST_METHOD0(GetLongestSegmentDuration, double());
  MOCK_METHOD1(SetTargetDuration, void(int32_t target_duration));
  MOCK_CONST_METHOD0(GetNumChannels, int());
  MOCK_CONST_METHOD0(GetEC3JocComplexity, int());
  MOCK_CONST_METHOD0(GetAC4ImsFlag, bool());
  MOCK_CONST_METHOD0(GetAC4CbiFlag, bool());
  MOCK_CONST_METHOD2(GetDisplayResolution,
                     bool(uint32_t* width, uint32_t* height));
  MOCK_CONST_METHOD0(GetFrameRate, double());
};
 
}  // namespace hls
}  // namespace shaka
 
#endif  // PACKAGER_HLS_BASE_MOCK_MEDIA_PLAYLIST_H_