p2p: Add P2PManager mock
This mock is useful when testing code with expectcations on P2PManager
interface. All calls are delegated to an underlying FakeP2PManager.
BUG=chromium:260426
TEST=Unit tests pass
Change-Id: Iff8088a634a672fa202127630673bf26e2b93153
Reviewed-on: https://chromium-review.googlesource.com/64826
Reviewed-by: David Zeuthen <zeuthen@chromium.org>
Commit-Queue: David Zeuthen <zeuthen@chromium.org>
Tested-by: David Zeuthen <zeuthen@chromium.org>
diff --git a/mock_p2p_manager.h b/mock_p2p_manager.h
new file mode 100644
index 0000000..ff633f8
--- /dev/null
+++ b/mock_p2p_manager.h
@@ -0,0 +1,91 @@
+// Copyright (c) 2013 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 CHROMEOS_PLATFORM_UPDATE_ENGINE_MOCK_P2P_MANAGER_H__
+#define CHROMEOS_PLATFORM_UPDATE_ENGINE_MOCK_P2P_MANAGER_H__
+
+#include "fake_p2p_manager.h"
+
+#include <gmock/gmock.h>
+
+namespace chromeos_update_engine {
+
+// A mocked, fake implementation of P2PManager.
+class MockP2PManager : public P2PManager {
+public:
+ MockP2PManager() {
+ // Delegate all calls to the fake instance
+ ON_CALL(*this, IsP2PEnabled())
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::IsP2PEnabled));
+ ON_CALL(*this, EnsureP2PRunning())
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::EnsureP2PRunning));
+ ON_CALL(*this, EnsureP2PNotRunning())
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::EnsureP2PNotRunning));
+ ON_CALL(*this, PerformHousekeeping())
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::PerformHousekeeping));
+ ON_CALL(*this, LookupUrlForFile(testing::_, testing::_, testing::_,
+ testing::_))
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::LookupUrlForFile));
+ ON_CALL(*this, FileShare(testing::_, testing::_))
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::FileShare));
+ ON_CALL(*this, FileGetPath(testing::_))
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::FileGetPath));
+ ON_CALL(*this, FileGetSize(testing::_))
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::FileGetSize));
+ ON_CALL(*this, FileGetExpectedSize(testing::_))
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::FileGetExpectedSize));
+ ON_CALL(*this, FileGetVisible(testing::_, testing::_))
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::FileGetVisible));
+ ON_CALL(*this, FileMakeVisible(testing::_))
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::FileMakeVisible));
+ ON_CALL(*this, CountSharedFiles())
+ .WillByDefault(testing::Invoke(&fake_,
+ &FakeP2PManager::CountSharedFiles));
+ }
+
+ virtual ~MockP2PManager() {}
+
+ // P2PManager overrides.
+ MOCK_METHOD0(IsP2PEnabled, bool());
+ MOCK_METHOD0(EnsureP2PRunning, bool());
+ MOCK_METHOD0(EnsureP2PNotRunning, bool());
+ MOCK_METHOD0(PerformHousekeeping, bool());
+ MOCK_METHOD4(LookupUrlForFile, void(const std::string&,
+ size_t,
+ base::TimeDelta,
+ LookupCallback));
+ MOCK_METHOD2(FileShare, bool(const std::string&, size_t));
+ MOCK_METHOD1(FileGetPath, base::FilePath(const std::string&));
+ MOCK_METHOD1(FileGetSize, ssize_t(const std::string&));
+ MOCK_METHOD1(FileGetExpectedSize, ssize_t(const std::string&));
+ MOCK_METHOD2(FileGetVisible, bool(const std::string&, bool*));
+ MOCK_METHOD1(FileMakeVisible, bool(const std::string&));
+ MOCK_METHOD0(CountSharedFiles, int());
+
+ // Returns a reference to the underlying FakeP2PManager.
+ FakeP2PManager& fake() {
+ return fake_;
+ }
+
+private:
+ // The underlying FakeP2PManager.
+ FakeP2PManager fake_;
+
+ DISALLOW_COPY_AND_ASSIGN(MockP2PManager);
+};
+
+} // namespace chromeos_update_engine
+
+#endif // CHROMEOS_PLATFORM_UPDATE_ENGINE_MOCK_P2P_MANAGER_H__