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 ATTESTATION_SERVER_MOCK_KEY_STORE_H_ 18 #define ATTESTATION_SERVER_MOCK_KEY_STORE_H_ 19 20 #include "attestation/server/key_store.h" 21 22 #include <string> 23 24 #include <base/macros.h> 25 #include <gmock/gmock.h> 26 27 namespace attestation { 28 29 class MockKeyStore : public KeyStore { 30 public: 31 MockKeyStore(); 32 virtual ~MockKeyStore(); 33 34 MOCK_METHOD3(Read, bool(const std::string& username, 35 const std::string& name, 36 std::string* key_data)); 37 MOCK_METHOD3(Write, bool(const std::string& username, 38 const std::string& name, 39 const std::string& key_data)); 40 MOCK_METHOD2(Delete, bool(const std::string& username, 41 const std::string& name)); 42 MOCK_METHOD2(DeleteByPrefix, bool(const std::string& username, 43 const std::string& key_prefix)); 44 MOCK_METHOD7(Register, bool(const std::string& username, 45 const std::string& label, 46 KeyType key_type, 47 KeyUsage key_usage, 48 const std::string& private_key_blob, 49 const std::string& public_key_der, 50 const std::string& certificate)); 51 MOCK_METHOD2(RegisterCertificate, bool(const std::string& username, 52 const std::string& certificate)); 53 54 private: 55 DISALLOW_COPY_AND_ASSIGN(MockKeyStore); 56 }; 57 58 } // namespace attestation 59 60 #endif // ATTESTATION_SERVER_MOCK_KEY_STORE_H_ 61
