1// 2// Copyright (C) 2015 The Android Open Source Project 3// 4// Licensed under the Apache License, Version 2.0 (the "License"); 5// you may not use this file except in compliance with the License. 6// You may obtain a copy of the License at 7// 8// http://www.apache.org/licenses/LICENSE-2.0 9// 10// Unless required by applicable law or agreed to in writing, software 11// distributed under the License is distributed on an "AS IS" BASIS, 12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13// See the License for the specific language governing permissions and 14// limitations under the License. 15// 16 17#ifndef TRUNKS_MOCK_HMAC_SESSION_H_ 18#define TRUNKS_MOCK_HMAC_SESSION_H_ 19 20#include <string> 21 22#include <gmock/gmock.h> 23 24#include "trunks/hmac_session.h" 25 26namespace trunks { 27 28class MockHmacSession : public HmacSession { 29 public: 30 MockHmacSession(); 31 ~MockHmacSession() override; 32 33 MOCK_METHOD0(GetDelegate, AuthorizationDelegate*()); 34 MOCK_METHOD3(StartBoundSession, 35 TPM_RC(TPMI_DH_ENTITY bind_entity, 36 const std::string& bind_authorization_value, 37 bool enable_encryption)); 38 MOCK_METHOD1(StartUnboundSession, TPM_RC(bool enable_encryption)); 39 MOCK_METHOD1(SetEntityAuthorizationValue, void(const std::string& value)); 40 MOCK_METHOD1(SetFutureAuthorizationValue, void(const std::string& value)); 41 42 private: 43 DISALLOW_COPY_AND_ASSIGN(MockHmacSession); 44}; 45 46} // namespace trunks 47 48#endif // TRUNKS_MOCK_HMAC_SESSION_H_ 49