trunks_factory_for_test.h revision a19238f46d4341489fd1d3140df1bb09bdbd8f01
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; 21a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahnclass NullAuthorizationDelegate; 22c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass Tpm; 23c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass TpmState; 24c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass TpmUtility; 25c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 26c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// A factory implementation for testing. Custom instances can be injected. If no 27c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// instance has been injected, a default mock instance will be used. Objects for 28c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// which ownership is passed to the caller are instantiated as forwarders which 29c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// simply forward calls to the current instance set for the class. 30c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// 31c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// Example usage: 32c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// TrunksFactoryForTest factory; 33c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// MockTpmState mock_tpm_state; 34c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// factory.set_tpm_state(mock_tpm_state); 35c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn// // Set expectations on mock_tpm_state... 36c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahnclass CHROMEOS_EXPORT TrunksFactoryForTest : public TrunksFactory { 37c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn public: 38c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn TrunksFactoryForTest(); 39c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn virtual ~TrunksFactoryForTest(); 40c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 41c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn // TrunksFactory methods. 42c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn Tpm* GetTpm() const override; 43c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn scoped_ptr<TpmState> GetTpmState() const override; 44c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn scoped_ptr<AuthorizationDelegate> GetPasswordAuthorization( 45c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn const std::string& password) const override; 46c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 47c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn // Mutators to inject custom mocks. 48c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn void set_tpm(Tpm* tpm) { 49c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn tpm_ = tpm; 50c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn } 51c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 52c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn void set_tpm_state(TpmState* tpm_state) { 53c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn tpm_state_ = tpm_state; 54c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn } 55c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 56a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahn void set_password_authorization_delegate(AuthorizationDelegate* delegate) { 57a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahn password_authorization_delegate_ = delegate; 58c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn } 59c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 60c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn private: 61c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn scoped_ptr<MockTpm> default_tpm_; 62c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn Tpm* tpm_; 63c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn scoped_ptr<MockTpmState> default_tpm_state_; 64c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn TpmState* tpm_state_; 65a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahn scoped_ptr<NullAuthorizationDelegate> default_authorization_delegate_; 66a19238f46d4341489fd1d3140df1bb09bdbd8f01Darren Krahn AuthorizationDelegate* password_authorization_delegate_; 67c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 68c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn DISALLOW_COPY_AND_ASSIGN(TrunksFactoryForTest); 69c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn}; 70c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 71c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn} // namespace trunks 72c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn 73c364caa9d091baae8eeb9144a6abf69e1fcabb39Darren Krahn#endif // TRUNKS_TRUNKS_FACTORY_FOR_TEST_H_ 74