| OLD | NEW |
| (Empty) | |
| 1 // Copyright 2016 The Chromium 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 "ash/common/system/chromeos/network/vpn_delegate.h" |
| 6 |
| 7 #include <algorithm> |
| 8 #include <vector> |
| 9 |
| 10 #include "base/macros.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" |
| 12 |
| 13 namespace ash { |
| 14 |
| 15 namespace { |
| 16 |
| 17 class TestVpnDelegate : public VPNDelegate { |
| 18 public: |
| 19 TestVpnDelegate() {} |
| 20 ~TestVpnDelegate() override {} |
| 21 |
| 22 // VPNDelegate: |
| 23 void ShowAddPage(const VPNProvider::Key& key) override {} |
| 24 |
| 25 private: |
| 26 DISALLOW_COPY_AND_ASSIGN(TestVpnDelegate); |
| 27 }; |
| 28 |
| 29 } // namespace |
| 30 |
| 31 using VpnDelegateTest = testing::Test; |
| 32 |
| 33 TEST_F(VpnDelegateTest, BuiltInProvider) { |
| 34 TestVpnDelegate delegate; |
| 35 |
| 36 // The VPN list should only contain the built-in provider. |
| 37 ASSERT_EQ(1u, delegate.vpn_providers().size()); |
| 38 VPNProvider provider = delegate.vpn_providers()[0]; |
| 39 EXPECT_FALSE(provider.key.third_party); |
| 40 EXPECT_TRUE(provider.key.extension_id.empty()); |
| 41 } |
| 42 |
| 43 TEST_F(VpnDelegateTest, ThirdPartyProviders) { |
| 44 TestVpnDelegate delegate; |
| 45 |
| 46 // The VPN list should only contain the built-in provider. |
| 47 EXPECT_EQ(1u, delegate.vpn_providers().size()); |
| 48 |
| 49 // Add some third party (extension-backed) providers. |
| 50 VPNProvider::Key key1("extension_id1"); |
| 51 VPNProvider provider1(key1, "name1"); |
| 52 VPNProvider::Key key2("extension_id2"); |
| 53 VPNProvider provider2(key2, "name2"); |
| 54 std::vector<VPNProvider> third_party_providers = {provider1, provider2}; |
| 55 delegate.SetThirdPartyVpnProviders(third_party_providers); |
| 56 |
| 57 // List contains the extension-backed providers. Order doesn't matter. |
| 58 std::vector<VPNProvider> providers = delegate.vpn_providers(); |
| 59 EXPECT_EQ(3u, providers.size()); |
| 60 EXPECT_EQ(1u, std::count(providers.begin(), providers.end(), provider1)); |
| 61 EXPECT_EQ(1u, std::count(providers.begin(), providers.end(), provider2)); |
| 62 } |
| 63 |
| 64 } // namespace ash |
| OLD | NEW |