| Index: chromeos/network/firewall_hole_unittest.cc
|
| diff --git a/chromeos/network/firewall_hole_unittest.cc b/chromeos/network/firewall_hole_unittest.cc
|
| index ea5f5ec300ce58a00e9e01470564059e3d9e41af..3032d5b81c0ba0bf5e39509a35901cc18e85ce19 100644
|
| --- a/chromeos/network/firewall_hole_unittest.cc
|
| +++ b/chromeos/network/firewall_hole_unittest.cc
|
| @@ -5,7 +5,7 @@
|
| #include "base/bind.h"
|
| #include "base/run_loop.h"
|
| #include "chromeos/dbus/dbus_thread_manager.h"
|
| -#include "chromeos/dbus/fake_permission_broker_client.h"
|
| +#include "chromeos/dbus/mock_permission_broker_client.h"
|
| #include "chromeos/network/firewall_hole.h"
|
| #include "dbus/file_descriptor.h"
|
| #include "testing/gmock/include/gmock/gmock.h"
|
| @@ -13,6 +13,7 @@
|
|
|
| using chromeos::DBusThreadManager;
|
| using chromeos::FirewallHole;
|
| +using chromeos::MockPermissionBrokerClient;
|
| using testing::_;
|
|
|
| namespace {
|
| @@ -23,36 +24,6 @@ ACTION_TEMPLATE(InvokeCallback,
|
| ::std::tr1::get<k>(args).Run(p1);
|
| }
|
|
|
| -class MockPermissionsBrokerClient : public chromeos::PermissionBrokerClient {
|
| - public:
|
| - MockPermissionsBrokerClient() {}
|
| - ~MockPermissionsBrokerClient() override {}
|
| -
|
| - MOCK_METHOD1(Init, void(dbus::Bus* bus));
|
| - MOCK_METHOD3(RequestPathAccess,
|
| - void(const std::string& path,
|
| - int interface_id,
|
| - const ResultCallback& callback));
|
| - MOCK_METHOD4(RequestTcpPortAccess,
|
| - void(uint16 port,
|
| - const std::string& interface,
|
| - const dbus::FileDescriptor& lifeline_fd,
|
| - const ResultCallback& callback));
|
| - MOCK_METHOD4(RequestUdpPortAccess,
|
| - void(uint16 port,
|
| - const std::string& interface,
|
| - const dbus::FileDescriptor& lifeline_fd,
|
| - const ResultCallback& callback));
|
| - MOCK_METHOD3(ReleaseTcpPort,
|
| - void(uint16 port,
|
| - const std::string& interface,
|
| - const ResultCallback& callback));
|
| - MOCK_METHOD3(ReleaseUdpPort,
|
| - void(uint16 port,
|
| - const std::string& interface,
|
| - const ResultCallback& callback));
|
| -};
|
| -
|
| } // namespace
|
|
|
| class FirewallHoleTest : public testing::Test {
|
| @@ -61,9 +32,9 @@ class FirewallHoleTest : public testing::Test {
|
| ~FirewallHoleTest() override {}
|
|
|
| void SetUp() override {
|
| - mock_permissions_broker_client_ = new MockPermissionsBrokerClient();
|
| + mock_permission_broker_client_ = new MockPermissionBrokerClient();
|
| DBusThreadManager::GetSetterForTesting()->SetPermissionBrokerClient(
|
| - make_scoped_ptr(mock_permissions_broker_client_));
|
| + make_scoped_ptr(mock_permission_broker_client_));
|
| }
|
|
|
| void TearDown() override { DBusThreadManager::Shutdown(); }
|
| @@ -83,14 +54,14 @@ class FirewallHoleTest : public testing::Test {
|
|
|
| protected:
|
| base::RunLoop run_loop_;
|
| - MockPermissionsBrokerClient* mock_permissions_broker_client_ = nullptr;
|
| + MockPermissionBrokerClient* mock_permission_broker_client_ = nullptr;
|
| };
|
|
|
| TEST_F(FirewallHoleTest, GrantTcpPortAccess) {
|
| - EXPECT_CALL(*mock_permissions_broker_client_,
|
| + EXPECT_CALL(*mock_permission_broker_client_,
|
| RequestTcpPortAccess(1234, "foo0", _, _))
|
| .WillOnce(InvokeCallback<3>(true));
|
| - EXPECT_CALL(*mock_permissions_broker_client_, ReleaseTcpPort(1234, "foo0", _))
|
| + EXPECT_CALL(*mock_permission_broker_client_, ReleaseTcpPort(1234, "foo0", _))
|
| .WillOnce(InvokeCallback<2>(true));
|
|
|
| FirewallHole::Open(
|
| @@ -100,7 +71,7 @@ TEST_F(FirewallHoleTest, GrantTcpPortAccess) {
|
| }
|
|
|
| TEST_F(FirewallHoleTest, DenyTcpPortAccess) {
|
| - EXPECT_CALL(*mock_permissions_broker_client_,
|
| + EXPECT_CALL(*mock_permission_broker_client_,
|
| RequestTcpPortAccess(1234, "foo0", _, _))
|
| .WillOnce(InvokeCallback<3>(false));
|
|
|
| @@ -111,10 +82,10 @@ TEST_F(FirewallHoleTest, DenyTcpPortAccess) {
|
| }
|
|
|
| TEST_F(FirewallHoleTest, GrantUdpPortAccess) {
|
| - EXPECT_CALL(*mock_permissions_broker_client_,
|
| + EXPECT_CALL(*mock_permission_broker_client_,
|
| RequestUdpPortAccess(1234, "foo0", _, _))
|
| .WillOnce(InvokeCallback<3>(true));
|
| - EXPECT_CALL(*mock_permissions_broker_client_, ReleaseUdpPort(1234, "foo0", _))
|
| + EXPECT_CALL(*mock_permission_broker_client_, ReleaseUdpPort(1234, "foo0", _))
|
| .WillOnce(InvokeCallback<2>(true));
|
|
|
| FirewallHole::Open(
|
| @@ -124,7 +95,7 @@ TEST_F(FirewallHoleTest, GrantUdpPortAccess) {
|
| }
|
|
|
| TEST_F(FirewallHoleTest, DenyUdpPortAccess) {
|
| - EXPECT_CALL(*mock_permissions_broker_client_,
|
| + EXPECT_CALL(*mock_permission_broker_client_,
|
| RequestUdpPortAccess(1234, "foo0", _, _))
|
| .WillOnce(InvokeCallback<3>(false));
|
|
|
|
|