Convert ComponentManager into an interface and create a mock
This will help to mock out ComponentManager's functionality for unit
tests in the future.
Change-Id: Ie74c49c6b31b00b0c4d38bf0db715a62a9532bc7
Reviewed-on: https://weave-review.googlesource.com/1785
Reviewed-by: Vitaly Buka <vitalybuka@google.com>
diff --git a/src/mock_component_manager.h b/src/mock_component_manager.h
new file mode 100644
index 0000000..66b32d5
--- /dev/null
+++ b/src/mock_component_manager.h
@@ -0,0 +1,99 @@
+// Copyright 2015 The Weave 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 LIBWEAVE_SRC_MOCK_COMPONENT_MANAGER_H_
+#define LIBWEAVE_SRC_MOCK_COMPONENT_MANAGER_H_
+
+#include "src/component_manager.h"
+
+#include <gmock/gmock.h>
+
+namespace weave {
+
+class MockComponentManager : public ComponentManager {
+ public:
+ ~MockComponentManager() override {}
+ MOCK_METHOD2(LoadTraits, bool(const base::DictionaryValue& dict,
+ ErrorPtr* error));
+ MOCK_METHOD2(LoadTraits, bool(const std::string& json, ErrorPtr* error));
+ MOCK_METHOD1(AddTraitDefChangedCallback, void(const base::Closure& callback));
+ MOCK_METHOD4(AddComponent, bool(const std::string& path,
+ const std::string& name,
+ const std::vector<std::string>& traits,
+ ErrorPtr* error));
+ MOCK_METHOD4(AddComponentArrayItem,
+ bool(const std::string& path,
+ const std::string& name,
+ const std::vector<std::string>& traits,
+ ErrorPtr* error));
+ MOCK_METHOD1(AddComponentTreeChangedCallback,
+ void(const base::Closure& callback));
+ MOCK_METHOD1(MockAddCommand, void(CommandInstance* command_instance));
+ MOCK_METHOD4(AddCommand, bool(const base::DictionaryValue& command,
+ UserRole role,
+ std::string* id,
+ ErrorPtr* error));
+ MOCK_METHOD1(FindCommand, CommandInstance*(const std::string& id));
+ MOCK_METHOD1(AddCommandAddedCallback,
+ void(const CommandQueue::CommandCallback& callback));
+ MOCK_METHOD1(AddCommandRemovedCallback,
+ void(const CommandQueue::CommandCallback& callback));
+ MOCK_METHOD3(AddCommandHandler,
+ void(const std::string& component_path,
+ const std::string& command_name,
+ const Device::CommandHandlerCallback& callback));
+ MOCK_CONST_METHOD2(FindComponent,
+ const base::DictionaryValue*(const std::string& path,
+ ErrorPtr* error));
+ MOCK_CONST_METHOD1(FindTraitDefinition,
+ const base::DictionaryValue*(const std::string& name));
+ MOCK_CONST_METHOD1(
+ FindCommandDefinition,
+ const base::DictionaryValue*(const std::string& command_name));
+ MOCK_CONST_METHOD3(GetMinimalRole, bool(const std::string& command_name,
+ UserRole* minimal_role,
+ ErrorPtr* error));
+ MOCK_CONST_METHOD0(GetTraits, const base::DictionaryValue&());
+ MOCK_CONST_METHOD0(GetComponents, const base::DictionaryValue&());
+ MOCK_METHOD3(SetStateProperties, bool(const std::string& component_path,
+ const base::DictionaryValue& dict,
+ ErrorPtr* error));
+ MOCK_METHOD3(SetStatePropertiesFromJson,
+ bool(const std::string& component_path,
+ const std::string& json,
+ ErrorPtr* error));
+ MOCK_CONST_METHOD3(GetStateProperty,
+ const base::Value*(const std::string& component_path,
+ const std::string& name,
+ ErrorPtr* error));
+ MOCK_METHOD4(SetStateProperty, bool(const std::string& component_path,
+ const std::string& name,
+ const base::Value& value,
+ ErrorPtr* error));
+ MOCK_METHOD1(AddStateChangedCallback, void(const base::Closure& callback));
+ MOCK_METHOD0(MockGetAndClearRecordedStateChanges, StateSnapshot&());
+ MOCK_METHOD1(NotifyStateUpdatedOnServer, void(UpdateID id));
+ MOCK_CONST_METHOD0(GetLastStateChangeId, UpdateID());
+ MOCK_METHOD1(MockAddServerStateUpdatedCallback,
+ base::CallbackList<void(UpdateID)>::Subscription*(
+ const base::Callback<void(UpdateID)>& callback));
+ MOCK_CONST_METHOD1(FindComponentWithTrait,
+ std::string(const std::string& trait));
+
+ private:
+ void AddCommand(std::unique_ptr<CommandInstance> command_instance) override {
+ MockAddCommand(command_instance.get());
+ }
+ StateSnapshot GetAndClearRecordedStateChanges() override {
+ return std::move(MockGetAndClearRecordedStateChanges());
+ }
+ Token AddServerStateUpdatedCallback(
+ const base::Callback<void(UpdateID)>& callback) override {
+ return Token{MockAddServerStateUpdatedCallback(callback)};
+ }
+};
+
+} // namespace weave
+
+#endif // LIBWEAVE_SRC_COMPONENT_MANAGER_H_
