//
// Copyright 2013 Google Inc. All Rights Reserved.
//

#ifndef WV_DRM_PLUGIN_MOCK_CDM_H_
#define WV_DRM_PLUGIN_MOCK_CDM_H_

#include <string>
#include <vector>

#include "gmock/gmock.h"
#include "wv_cdm_types.h"
#include "wv_content_decryption_module.h"


namespace wvcdm {

class MockCDM : public WvContentDecryptionModule {
 public:
  MOCK_METHOD2(OpenSession, CdmResponseType(const CdmKeySystem&,
                                            CdmSessionId*));

  MOCK_METHOD1(CloseSession, CdmResponseType(CdmSessionId&));

  MOCK_METHOD5(GenerateKeyRequest, CdmResponseType(const CdmSessionId&,
                                                   const CdmInitData&,
                                                   const CdmLicenseType,
                                                   CdmAppParameterMap&,
                                                   CdmKeyMessage*));

  MOCK_METHOD2(AddKey, CdmResponseType(const CdmSessionId&,
                                       const CdmKeyResponse&));

  MOCK_METHOD1(CancelKeyRequest, CdmResponseType(const CdmSessionId&));

  MOCK_METHOD2(QueryKeyStatus, CdmResponseType(const CdmSessionId&,
                                               CdmQueryMap*));

  MOCK_METHOD2(GetProvisioningRequest, CdmResponseType(CdmProvisioningRequest*,
                                                       std::string*));

  MOCK_METHOD1(HandleProvisioningResponse,
      CdmResponseType(CdmProvisioningResponse&));

  MOCK_METHOD1(GetSecureStops, CdmResponseType(CdmSecureStops*));

  MOCK_METHOD1(ReleaseSecureStops,
      CdmResponseType(const CdmSecureStopReleaseMessage&));
};

} // namespace wvcdm

#endif // WV_DRM_PLUGIN_MOCK_CDM_H_