trunks_factory_for_test.h revision 03d54dfbcbdd04384e8c0419b7c45282664a2c1a
1c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// Copyright 2014 The Chromium OS Authors. All rights reserved. 2c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// Use of this source code is governed by a BSD-style license that can be 3c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// found in the LICENSE file. 4c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 5c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn#ifndef TRUNKS_TRUNKS_FACTORY_FOR_TEST_H_ 6c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn#define TRUNKS_TRUNKS_FACTORY_FOR_TEST_H_ 7c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 8c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn#include "trunks/trunks_factory.h" 9c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 10c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn#include <string> 11c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 12c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn#include <base/macros.h> 13c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn#include <base/memory/scoped_ptr.h> 14c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn#include <chromeos/chromeos_export.h> 15c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 16c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnnamespace trunks { 17c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 18c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass AuthorizationDelegate; 19c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass MockTpm; 20c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass MockTpmState; 2103d54dfbcbdd04384e8c0419b7c45282664a2c1aDarren Krahnclass MockTpmUtility; 22a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahnclass NullAuthorizationDelegate; 23c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass Tpm; 24c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass TpmState; 25c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass TpmUtility; 26c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 27c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// A factory implementation for testing. Custom instances can be injected. If no 28c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// instance has been injected, a default mock instance will be used. Objects for 29c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// which ownership is passed to the caller are instantiated as forwarders which 30c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// simply forward calls to the current instance set for the class. 31c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// 32c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// Example usage: 33c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// TrunksFactoryForTest factory; 34c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// MockTpmState mock_tpm_state; 35c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// factory.set_tpm_state(mock_tpm_state); 36c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// // Set expectations on mock_tpm_state... 37c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass CHROMEOS_EXPORT TrunksFactoryForTest : public TrunksFactory { 38c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn public: 39c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn TrunksFactoryForTest(); 40c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn virtual ~TrunksFactoryForTest(); 41c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 42c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn // TrunksFactory methods. 43c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn Tpm* GetTpm() const override; 44c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn scoped_ptr<TpmState> GetTpmState() const override; 4503d54dfbcbdd04384e8c0419b7c45282664a2c1aDarren Krahn scoped_ptr<TpmUtility> GetTpmUtility() const override; 46c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn scoped_ptr<AuthorizationDelegate> GetPasswordAuthorization( 47c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn const std::string& password) const override; 48c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 49c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn // Mutators to inject custom mocks. 50c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn void set_tpm(Tpm* tpm) { 51c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn tpm_ = tpm; 52c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn } 53c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 54c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn void set_tpm_state(TpmState* tpm_state) { 55c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn tpm_state_ = tpm_state; 56c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn } 57c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 5803d54dfbcbdd04384e8c0419b7c45282664a2c1aDarren Krahn void set_tpm_utility(TpmUtility* tpm_utility) { 5903d54dfbcbdd04384e8c0419b7c45282664a2c1aDarren Krahn tpm_utility_ = tpm_utility; 6003d54dfbcbdd04384e8c0419b7c45282664a2c1aDarren Krahn } 6103d54dfbcbdd04384e8c0419b7c45282664a2c1aDarren Krahn 62a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahn void set_password_authorization_delegate(AuthorizationDelegate* delegate) { 63a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahn password_authorization_delegate_ = delegate; 64c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn } 65c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 66c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn private: 67c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn scoped_ptr<MockTpm> default_tpm_; 68c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn Tpm* tpm_; 69c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn scoped_ptr<MockTpmState> default_tpm_state_; 70c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn TpmState* tpm_state_; 7103d54dfbcbdd04384e8c0419b7c45282664a2c1aDarren Krahn scoped_ptr<MockTpmUtility> default_tpm_utility_; 7203d54dfbcbdd04384e8c0419b7c45282664a2c1aDarren Krahn TpmUtility* tpm_utility_; 73a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahn scoped_ptr<NullAuthorizationDelegate> default_authorization_delegate_; 74a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahn AuthorizationDelegate* password_authorization_delegate_; 75c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 76c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn DISALLOW_COPY_AND_ASSIGN(TrunksFactoryForTest); 77c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn}; 78c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 79c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn} // namespace trunks 80c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 81c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn#endif // TRUNKS_TRUNKS_FACTORY_FOR_TEST_H_ 82