| // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef CHAPS_OBJECT_STORE_MOCK_H_ |
| #define CHAPS_OBJECT_STORE_MOCK_H_ |
| |
| #include "chaps/object_store.h" |
| |
| #include <map> |
| #include <string> |
| |
| #include <gmock/gmock.h> |
| |
| namespace chaps { |
| |
| class ObjectStoreMock : public ObjectStore { |
| public: |
| ObjectStoreMock(); |
| ~ObjectStoreMock() override; |
| MOCK_METHOD2(GetInternalBlob, bool(int blob_id, std::string* blob)); |
| MOCK_METHOD2(SetInternalBlob, bool(int blob_id, const std::string& blob)); |
| MOCK_METHOD1(SetEncryptionKey, bool(const brillo::SecureBlob& key)); |
| MOCK_METHOD2(InsertObjectBlob, bool(const ObjectBlob& blob, int* blob_id)); |
| MOCK_METHOD1(DeleteObjectBlob, bool(int blob_id)); |
| MOCK_METHOD0(DeleteAllObjectBlobs, bool()); |
| MOCK_METHOD2(UpdateObjectBlob, bool(int blob_id, const ObjectBlob& blob)); |
| MOCK_METHOD1(LoadPublicObjectBlobs, bool(std::map<int, ObjectBlob>* blobs)); |
| MOCK_METHOD1(LoadPrivateObjectBlobs, bool(std::map<int, ObjectBlob>* blobs)); |
| }; |
| |
| } // namespace chaps |
| |
| #endif // CHAPS_OBJECT_STORE_MOCK_H_ |