Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "shill/wimax.h" |
| 6 | |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 7 | #include <string> |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 8 | |
Darin Petkov | 9893d9c | 2012-05-17 15:27:31 -0700 | [diff] [blame] | 9 | #include <base/stringprintf.h> |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 10 | #include <gtest/gtest.h> |
| 11 | |
| 12 | #include "shill/event_dispatcher.h" |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 13 | #include "shill/mock_manager.h" |
| 14 | #include "shill/mock_metrics.h" |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 15 | #include "shill/mock_wimax_device_proxy.h" |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 16 | #include "shill/mock_wimax_provider.h" |
Darin Petkov | 9893d9c | 2012-05-17 15:27:31 -0700 | [diff] [blame] | 17 | #include "shill/mock_wimax_service.h" |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 18 | #include "shill/nice_mock_control.h" |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 19 | #include "shill/proxy_factory.h" |
| 20 | |
| 21 | using std::string; |
Darin Petkov | 912f0de | 2012-05-16 14:12:14 +0200 | [diff] [blame] | 22 | using testing::_; |
Darin Petkov | 8ea0eaf | 2012-05-29 11:21:33 +0200 | [diff] [blame] | 23 | using testing::NiceMock; |
Darin Petkov | 9893d9c | 2012-05-17 15:27:31 -0700 | [diff] [blame] | 24 | using testing::Return; |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 25 | |
| 26 | namespace shill { |
| 27 | |
| 28 | namespace { |
| 29 | |
| 30 | const char kTestLinkName[] = "wm0"; |
Ben Chan | 4e64d2d | 2012-05-16 00:02:25 -0700 | [diff] [blame] | 31 | const char kTestAddress[] = "01:23:45:67:89:ab"; |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 32 | const int kTestInterfaceIndex = 5; |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 33 | const char kTestPath[] = "/org/chromium/WiMaxManager/Device/6"; |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 34 | |
| 35 | } // namespace |
| 36 | |
| 37 | class WiMaxTest : public testing::Test { |
| 38 | public: |
| 39 | WiMaxTest() |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 40 | : proxy_(new MockWiMaxDeviceProxy()), |
| 41 | proxy_factory_(this), |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 42 | manager_(&control_, &dispatcher_, &metrics_, NULL), |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 43 | device_(new WiMax(&control_, &dispatcher_, &metrics_, &manager_, |
| 44 | kTestLinkName, kTestAddress, kTestInterfaceIndex, |
| 45 | kTestPath)) {} |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 46 | |
| 47 | virtual ~WiMaxTest() {} |
| 48 | |
| 49 | protected: |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 50 | class TestProxyFactory : public ProxyFactory { |
| 51 | public: |
| 52 | explicit TestProxyFactory(WiMaxTest *test) : test_(test) {} |
| 53 | |
| 54 | virtual WiMaxDeviceProxyInterface *CreateWiMaxDeviceProxy( |
| 55 | const string &/*path*/) { |
| 56 | return test_->proxy_.release(); |
| 57 | } |
| 58 | |
| 59 | private: |
| 60 | WiMaxTest *test_; |
| 61 | |
| 62 | DISALLOW_COPY_AND_ASSIGN(TestProxyFactory); |
| 63 | }; |
| 64 | |
| 65 | virtual void SetUp() { |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 66 | device_->proxy_factory_ = &proxy_factory_; |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 67 | } |
| 68 | |
| 69 | virtual void TearDown() { |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 70 | device_->SelectService(NULL); |
| 71 | device_->pending_service_ = NULL; |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 72 | device_->proxy_factory_ = NULL; |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 73 | } |
| 74 | |
| 75 | scoped_ptr<MockWiMaxDeviceProxy> proxy_; |
| 76 | TestProxyFactory proxy_factory_; |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 77 | NiceMockControl control_; |
| 78 | EventDispatcher dispatcher_; |
| 79 | MockMetrics metrics_; |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 80 | MockManager manager_; |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 81 | WiMaxRefPtr device_; |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 82 | }; |
| 83 | |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 84 | TEST_F(WiMaxTest, Constructor) { |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 85 | EXPECT_EQ(kTestPath, device_->path()); |
| 86 | EXPECT_FALSE(device_->scanning()); |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 87 | } |
| 88 | |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 89 | TEST_F(WiMaxTest, TechnologyIs) { |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 90 | EXPECT_TRUE(device_->TechnologyIs(Technology::kWiMax)); |
| 91 | EXPECT_FALSE(device_->TechnologyIs(Technology::kEthernet)); |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 92 | } |
| 93 | |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 94 | TEST_F(WiMaxTest, StartStop) { |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 95 | EXPECT_FALSE(device_->proxy_.get()); |
Darin Petkov | 912f0de | 2012-05-16 14:12:14 +0200 | [diff] [blame] | 96 | EXPECT_CALL(*proxy_, Enable(_, _, _)); |
Darin Petkov | 9893d9c | 2012-05-17 15:27:31 -0700 | [diff] [blame] | 97 | EXPECT_CALL(*proxy_, set_networks_changed_callback(_)); |
Darin Petkov | 8ea0eaf | 2012-05-29 11:21:33 +0200 | [diff] [blame] | 98 | EXPECT_CALL(*proxy_, set_status_changed_callback(_)); |
Darin Petkov | 912f0de | 2012-05-16 14:12:14 +0200 | [diff] [blame] | 99 | EXPECT_CALL(*proxy_, Disable(_, _, _)); |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 100 | device_->Start(NULL, EnabledStateChangedCallback()); |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 101 | ASSERT_TRUE(device_->proxy_.get()); |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 102 | |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 103 | device_->networks_.insert("path"); |
| 104 | MockWiMaxProvider provider; |
| 105 | EXPECT_CALL(manager_, wimax_provider()).WillOnce(Return(&provider)); |
| 106 | EXPECT_CALL(provider, OnNetworksChanged()); |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 107 | device_->Stop(NULL, EnabledStateChangedCallback()); |
| 108 | EXPECT_TRUE(device_->networks_.empty()); |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 109 | } |
| 110 | |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 111 | TEST_F(WiMaxTest, OnServiceStopped) { |
Darin Petkov | 8ea0eaf | 2012-05-29 11:21:33 +0200 | [diff] [blame] | 112 | scoped_refptr<NiceMock<MockWiMaxService> > service0( |
| 113 | new NiceMock<MockWiMaxService>( |
| 114 | &control_, |
| 115 | reinterpret_cast<EventDispatcher *>(NULL), |
| 116 | &metrics_, |
| 117 | &manager_)); |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 118 | scoped_refptr<MockWiMaxService> service1( |
| 119 | new MockWiMaxService(&control_, NULL, &metrics_, &manager_)); |
| 120 | device_->SelectService(service0); |
| 121 | device_->pending_service_ = service1; |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 122 | |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 123 | device_->OnServiceStopped(NULL); |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 124 | EXPECT_TRUE(device_->selected_service()); |
| 125 | EXPECT_TRUE(device_->pending_service_); |
| 126 | |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 127 | device_->OnServiceStopped(service0); |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 128 | EXPECT_FALSE(device_->selected_service()); |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 129 | EXPECT_TRUE(device_->pending_service_); |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 130 | |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 131 | device_->OnServiceStopped(service1); |
| 132 | EXPECT_FALSE(device_->selected_service()); |
| 133 | EXPECT_FALSE(device_->pending_service_); |
Darin Petkov | b72b62e | 2012-05-15 16:55:36 +0200 | [diff] [blame] | 134 | } |
| 135 | |
Darin Petkov | 9893d9c | 2012-05-17 15:27:31 -0700 | [diff] [blame] | 136 | TEST_F(WiMaxTest, OnNetworksChanged) { |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 137 | MockWiMaxProvider provider; |
| 138 | EXPECT_CALL(manager_, wimax_provider()).WillOnce(Return(&provider)); |
| 139 | EXPECT_CALL(provider, OnNetworksChanged()); |
| 140 | device_->networks_.insert("foo"); |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 141 | RpcIdentifiers networks; |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 142 | networks.push_back("bar"); |
| 143 | networks.push_back("zoo"); |
| 144 | networks.push_back("bar"); |
Darin Petkov | d1cd797 | 2012-05-22 15:26:15 +0200 | [diff] [blame] | 145 | device_->OnNetworksChanged(networks); |
Darin Petkov | c63dcf0 | 2012-05-24 11:51:43 +0200 | [diff] [blame] | 146 | EXPECT_EQ(2, device_->networks_.size()); |
| 147 | EXPECT_TRUE(ContainsKey(device_->networks_, "bar")); |
| 148 | EXPECT_TRUE(ContainsKey(device_->networks_, "zoo")); |
Darin Petkov | 9893d9c | 2012-05-17 15:27:31 -0700 | [diff] [blame] | 149 | } |
| 150 | |
Darin Petkov | 8ea0eaf | 2012-05-29 11:21:33 +0200 | [diff] [blame] | 151 | TEST_F(WiMaxTest, OnConnectComplete) { |
| 152 | scoped_refptr<MockWiMaxService> service( |
| 153 | new MockWiMaxService(&control_, NULL, &metrics_, &manager_)); |
| 154 | device_->pending_service_ = service; |
| 155 | EXPECT_CALL(*service, SetState(_)).Times(0); |
| 156 | EXPECT_TRUE(device_->pending_service_); |
| 157 | EXPECT_CALL(*service, SetState(Service::kStateFailure)); |
| 158 | device_->OnConnectComplete(Error(Error::kOperationFailed)); |
| 159 | EXPECT_FALSE(device_->pending_service_); |
| 160 | } |
| 161 | |
Ben Chan | c07362b | 2012-05-12 10:54:11 -0700 | [diff] [blame] | 162 | } // namespace shill |