Peter Qiu | c0beca5 | 2015-09-03 11:25:46 -0700 | [diff] [blame] | 1 | // |
| 2 | // Copyright (C) 2013 The Android Open Source Project |
| 3 | // |
| 4 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | // you may not use this file except in compliance with the License. |
| 6 | // You may obtain a copy of the License at |
| 7 | // |
| 8 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | // |
| 10 | // Unless required by applicable law or agreed to in writing, software |
| 11 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | // See the License for the specific language governing permissions and |
| 14 | // limitations under the License. |
| 15 | // |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 16 | |
Ben Chan | 8760251 | 2014-11-07 20:50:05 -0800 | [diff] [blame] | 17 | #include "shill/ethernet/ethernet_eap_provider.h" |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 18 | |
| 19 | #include <base/bind.h> |
| 20 | #include <gmock/gmock.h> |
| 21 | #include <gtest/gtest.h> |
| 22 | |
Ben Chan | 8760251 | 2014-11-07 20:50:05 -0800 | [diff] [blame] | 23 | #include "shill/ethernet/mock_ethernet.h" |
Paul Stewart | bc14fb7 | 2013-07-30 08:21:58 -0700 | [diff] [blame] | 24 | #include "shill/key_value_store.h" |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 25 | #include "shill/mock_control.h" |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 26 | #include "shill/mock_event_dispatcher.h" |
| 27 | #include "shill/mock_manager.h" |
| 28 | #include "shill/mock_metrics.h" |
| 29 | |
| 30 | using testing::_; |
| 31 | using testing::Mock; |
| 32 | using testing::SaveArg; |
| 33 | |
| 34 | namespace shill { |
| 35 | |
| 36 | class EthernetEapProviderTest : public testing::Test { |
| 37 | public: |
| 38 | EthernetEapProviderTest() |
| 39 | : metrics_(&dispatcher_), |
mukesh agrawal | bad1c10 | 2015-09-18 16:15:08 -0700 | [diff] [blame] | 40 | manager_(&control_, &dispatcher_, &metrics_), |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 41 | provider_(&control_, &dispatcher_, &metrics_, &manager_) {} |
| 42 | virtual ~EthernetEapProviderTest() {} |
| 43 | |
| 44 | MOCK_METHOD0(Callback0, void()); |
| 45 | MOCK_METHOD0(Callback1, void()); |
| 46 | |
| 47 | protected: |
Paul Stewart | 7e779d8 | 2015-06-16 13:13:10 -0700 | [diff] [blame] | 48 | const EthernetEapProvider::CallbackMap& CallbackMap() { |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 49 | return provider_.callback_map_; |
| 50 | } |
| 51 | |
| 52 | MockControl control_; |
| 53 | MockEventDispatcher dispatcher_; |
| 54 | MockMetrics metrics_; |
| 55 | MockManager manager_; |
| 56 | EthernetEapProvider provider_; |
| 57 | }; |
| 58 | |
| 59 | TEST_F(EthernetEapProviderTest, Construct) { |
| 60 | EXPECT_EQ(ServiceRefPtr(), provider_.service()); |
| 61 | EXPECT_TRUE(CallbackMap().empty()); |
| 62 | } |
| 63 | |
| 64 | TEST_F(EthernetEapProviderTest, StartAndStop) { |
| 65 | ServiceRefPtr service; |
| 66 | EXPECT_CALL(manager_, RegisterService(_)).WillOnce(SaveArg<0>(&service)); |
| 67 | provider_.Start(); |
| 68 | EXPECT_NE(ServiceRefPtr(), provider_.service()); |
| 69 | EXPECT_EQ(service, provider_.service()); |
| 70 | |
| 71 | EXPECT_CALL(manager_, DeregisterService(service)); |
| 72 | provider_.Stop(); |
| 73 | EXPECT_EQ(service, provider_.service()); |
| 74 | |
| 75 | // Provider re-uses the same service on restart. |
| 76 | EXPECT_CALL(manager_, RegisterService(service)); |
| 77 | provider_.Start(); |
| 78 | Mock::VerifyAndClearExpectations(&manager_); |
| 79 | } |
| 80 | |
| 81 | TEST_F(EthernetEapProviderTest, CredentialChangeCallback) { |
| 82 | EXPECT_CALL(*this, Callback0()).Times(0); |
| 83 | EXPECT_CALL(*this, Callback1()).Times(0); |
| 84 | provider_.OnCredentialsChanged(); |
| 85 | |
| 86 | scoped_refptr<MockEthernet> device0 = new MockEthernet(&control_, |
| 87 | &dispatcher_, |
| 88 | &metrics_, |
| 89 | &manager_, |
| 90 | "eth0", |
| 91 | "addr0", |
| 92 | 0); |
| 93 | EthernetEapProvider::CredentialChangeCallback callback0 = |
| 94 | base::Bind(&EthernetEapProviderTest::Callback0, |
| 95 | base::Unretained(this)); |
| 96 | |
Alex Vakulenko | 0951ccb | 2014-12-10 12:52:31 -0800 | [diff] [blame] | 97 | provider_.SetCredentialChangeCallback(device0.get(), callback0); |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 98 | EXPECT_CALL(*this, Callback0()); |
| 99 | EXPECT_CALL(*this, Callback1()).Times(0); |
| 100 | provider_.OnCredentialsChanged(); |
| 101 | |
| 102 | scoped_refptr<MockEthernet> device1 = new MockEthernet(&control_, |
| 103 | &dispatcher_, |
| 104 | &metrics_, |
| 105 | &manager_, |
| 106 | "eth1", |
| 107 | "addr1", |
| 108 | 1); |
| 109 | EthernetEapProvider::CredentialChangeCallback callback1 = |
| 110 | base::Bind(&EthernetEapProviderTest::Callback1, |
| 111 | base::Unretained(this)); |
| 112 | |
Alex Vakulenko | 0951ccb | 2014-12-10 12:52:31 -0800 | [diff] [blame] | 113 | provider_.SetCredentialChangeCallback(device1.get(), callback1); |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 114 | EXPECT_CALL(*this, Callback0()); |
| 115 | EXPECT_CALL(*this, Callback1()); |
| 116 | provider_.OnCredentialsChanged(); |
| 117 | |
Alex Vakulenko | 0951ccb | 2014-12-10 12:52:31 -0800 | [diff] [blame] | 118 | provider_.SetCredentialChangeCallback(device1.get(), callback0); |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 119 | EXPECT_CALL(*this, Callback0()).Times(2); |
| 120 | EXPECT_CALL(*this, Callback1()).Times(0); |
| 121 | provider_.OnCredentialsChanged(); |
| 122 | |
Alex Vakulenko | 0951ccb | 2014-12-10 12:52:31 -0800 | [diff] [blame] | 123 | provider_.ClearCredentialChangeCallback(device0.get()); |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 124 | EXPECT_CALL(*this, Callback0()); |
| 125 | EXPECT_CALL(*this, Callback1()).Times(0); |
| 126 | provider_.OnCredentialsChanged(); |
| 127 | |
Alex Vakulenko | 0951ccb | 2014-12-10 12:52:31 -0800 | [diff] [blame] | 128 | provider_.ClearCredentialChangeCallback(device1.get()); |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 129 | EXPECT_CALL(*this, Callback0()).Times(0); |
| 130 | EXPECT_CALL(*this, Callback1()).Times(0); |
| 131 | provider_.OnCredentialsChanged(); |
| 132 | } |
| 133 | |
Paul Stewart | bc14fb7 | 2013-07-30 08:21:58 -0700 | [diff] [blame] | 134 | TEST_F(EthernetEapProviderTest, ServiceConstructors) { |
| 135 | ServiceRefPtr service; |
| 136 | EXPECT_CALL(manager_, RegisterService(_)).WillOnce(SaveArg<0>(&service)); |
| 137 | provider_.Start(); |
| 138 | KeyValueStore args; |
Ben Chan | 783f00b | 2013-09-20 13:50:34 -0700 | [diff] [blame] | 139 | args.SetString(kTypeProperty, kTypeEthernetEap); |
Paul Stewart | bc14fb7 | 2013-07-30 08:21:58 -0700 | [diff] [blame] | 140 | { |
| 141 | Error error; |
| 142 | EXPECT_EQ(service, provider_.GetService(args, &error)); |
| 143 | EXPECT_TRUE(error.IsSuccess()); |
| 144 | } |
| 145 | { |
| 146 | Error error; |
| 147 | EXPECT_EQ(service, provider_.FindSimilarService(args, &error)); |
| 148 | EXPECT_TRUE(error.IsSuccess()); |
| 149 | } |
| 150 | { |
| 151 | Error error; |
| 152 | Mock::VerifyAndClearExpectations(&manager_); |
| 153 | EXPECT_CALL(manager_, RegisterService(_)).Times(0); |
| 154 | ServiceRefPtr temp_service = provider_.CreateTemporaryService(args, &error); |
| 155 | EXPECT_TRUE(error.IsSuccess()); |
| 156 | // Returned service should be non-NULL but not the provider's own service. |
| 157 | EXPECT_NE(ServiceRefPtr(), temp_service); |
| 158 | EXPECT_NE(service, temp_service); |
| 159 | } |
| 160 | } |
| 161 | |
Paul Stewart | 35eff13 | 2013-04-12 12:08:40 -0700 | [diff] [blame] | 162 | } // namespace shill |