1// 2// Copyright (C) 2014 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_TRUNKS_FACTORY_FOR_TEST_H_ 18#define TRUNKS_TRUNKS_FACTORY_FOR_TEST_H_ 19 20#include "trunks/trunks_factory.h" 21 22#include <memory> 23#include <string> 24 25#include <base/macros.h> 26 27#include "trunks/password_authorization_delegate.h" 28#include "trunks/trunks_export.h" 29 30namespace trunks { 31 32class AuthorizationDelegate; 33class MockBlobParser; 34class MockHmacSession; 35class MockPolicySession; 36class MockSessionManager; 37class MockTpm; 38class MockTpmState; 39class MockTpmUtility; 40class HmacSession; 41class PasswordAuthorizationDelegate; 42class PolicySession; 43class SessionManager; 44class Tpm; 45class TpmState; 46class TpmUtility; 47 48// A factory implementation for testing. Custom instances can be injected. If no 49// instance has been injected, a default mock instance will be used. Objects for 50// which ownership is passed to the caller are instantiated as forwarders which 51// simply forward calls to the current instance set for the class. 52// 53// Example usage: 54// TrunksFactoryForTest factory; 55// MockTpmState mock_tpm_state; 56// factory.set_tpm_state(mock_tpm_state); 57// // Set expectations on mock_tpm_state... 58class TRUNKS_EXPORT TrunksFactoryForTest : public TrunksFactory { 59 public: 60 TrunksFactoryForTest(); 61 ~TrunksFactoryForTest() override; 62 63 // TrunksFactory methods. 64 Tpm* GetTpm() const override; 65 std::unique_ptr<TpmState> GetTpmState() const override; 66 std::unique_ptr<TpmUtility> GetTpmUtility() const override; 67 std::unique_ptr<AuthorizationDelegate> GetPasswordAuthorization( 68 const std::string& password) const override; 69 std::unique_ptr<SessionManager> GetSessionManager() const override; 70 std::unique_ptr<HmacSession> GetHmacSession() const override; 71 std::unique_ptr<PolicySession> GetPolicySession() const override; 72 std::unique_ptr<PolicySession> GetTrialSession() const override; 73 std::unique_ptr<BlobParser> GetBlobParser() const override; 74 75 // Mutators to inject custom mocks. 76 void set_tpm(Tpm* tpm) { tpm_ = tpm; } 77 78 void set_tpm_state(TpmState* tpm_state) { tpm_state_ = tpm_state; } 79 80 void set_tpm_utility(TpmUtility* tpm_utility) { tpm_utility_ = tpm_utility; } 81 82 void set_password_authorization_delegate(AuthorizationDelegate* delegate) { 83 password_authorization_delegate_ = delegate; 84 } 85 86 void set_session_manager(SessionManager* session_manager) { 87 session_manager_ = session_manager; 88 } 89 90 void set_hmac_session(HmacSession* hmac_session) { 91 hmac_session_ = hmac_session; 92 } 93 94 void set_policy_session(PolicySession* policy_session) { 95 policy_session_ = policy_session; 96 } 97 98 void set_trial_session(PolicySession* trial_session) { 99 trial_session_ = trial_session; 100 } 101 102 void set_blob_parser(BlobParser* blob_parser) { blob_parser_ = blob_parser; } 103 104 private: 105 std::unique_ptr<MockTpm> default_tpm_; 106 Tpm* tpm_; 107 std::unique_ptr<MockTpmState> default_tpm_state_; 108 TpmState* tpm_state_; 109 std::unique_ptr<MockTpmUtility> default_tpm_utility_; 110 TpmUtility* tpm_utility_; 111 std::unique_ptr<PasswordAuthorizationDelegate> 112 default_authorization_delegate_; 113 AuthorizationDelegate* password_authorization_delegate_; 114 std::unique_ptr<MockSessionManager> default_session_manager_; 115 SessionManager* session_manager_; 116 std::unique_ptr<MockHmacSession> default_hmac_session_; 117 HmacSession* hmac_session_; 118 std::unique_ptr<MockPolicySession> default_policy_session_; 119 PolicySession* policy_session_; 120 std::unique_ptr<MockPolicySession> default_trial_session_; 121 PolicySession* trial_session_; 122 std::unique_ptr<MockBlobParser> default_blob_parser_; 123 BlobParser* blob_parser_; 124 125 DISALLOW_COPY_AND_ASSIGN(TrunksFactoryForTest); 126}; 127 128} // namespace trunks 129 130#endif // TRUNKS_TRUNKS_FACTORY_FOR_TEST_H_ 131