third_party/WebKit/Source/core/page/NetworkStateNotifierTest.cpp

   1 /*
   2  * Copyright (c) 2014, Google Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are
   6  * met:
   7  *
   8  *     * Redistributions of source code must retain the above copyright
   9  * notice, this list of conditions and the following disclaimer.
  10  *     * Redistributions in binary form must reproduce the above
  11  * copyright notice, this list of conditions and the following disclaimer
  12  * in the documentation and/or other materials provided with the
  13  * distribution.
  14  *     * Neither the name of Google Inc. nor the names of its
  15  * contributors may be used to endorse or promote products derived from
  16  * this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31 #include "config.h"
  32 #include "core/page/NetworkStateNotifier.h"
  33
  34 #include "core/dom/Document.h"
  35 #include "platform/testing/UnitTestHelpers.h"
  36 #include "public/platform/Platform.h"
  37 #include "public/platform/WebConnectionType.h"
  38 #include "public/platform/WebThread.h"
  39 #include "wtf/Functional.h"
  40 #include <gtest/gtest.h>
  41
  42 namespace blink {
  43
  44 namespace {
  45     const double kNoneMaxBandwidthMbps = 0.0;
  46     const double kBluetoothMaxBandwidthMbps = 1.0;
  47     const double kEthernetMaxBandwidthMbps = 2.0;
  48 }
  49
  50 class StateObserver : public NetworkStateNotifier::NetworkStateObserver {
  51 public:
  52     StateObserver()
  53         : m_observedType(WebConnectionTypeNone)
  54         , m_observedMaxBandwidthMbps(0.0)
  55         , m_callbackCount(0)
  56     {
  57     }
  58
  59     virtual void connectionChange(WebConnectionType type, double maxBandwidthMbps)
  60     {
  61         m_observedType = type;
  62         m_observedMaxBandwidthMbps = maxBandwidthMbps;
  63         m_callbackCount += 1;
  64
  65         if (m_closure)
  66             (*m_closure)();
  67     }
  68
  69     WebConnectionType observedType() const
  70     {
  71         return m_observedType;
  72     }
  73
  74     double observedMaxBandwidth() const
  75     {
  76         return m_observedMaxBandwidthMbps;
  77     }
  78
  79     int callbackCount() const
  80     {
  81         return m_callbackCount;
  82     }
  83
  84     void setNotificationCallback(PassOwnPtr<Closure> closure)
  85     {
  86         m_closure = closure;
  87     }
  88
  89 private:
  90     OwnPtr<Closure> m_closure;
  91     WebConnectionType m_observedType;
  92     double m_observedMaxBandwidthMbps;
  93     int m_callbackCount;
  94 };
  95
  96 class NetworkStateNotifierTest : public ::testing::Test {
  97 public:
  98     NetworkStateNotifierTest()
  99         : m_document(Document::create())
 100         , m_document2(Document::create())
 101     {
 102     }
 103
 104     ExecutionContext* executionContext()
 105     {
 106         return m_document.get();
 107     }
 108
 109     ExecutionContext* executionContext2()
 110     {
 111         return m_document2.get();
 112     }
 113
 114 protected:
 115     void setConnection(WebConnectionType type, double maxBandwidthMbps)
 116     {
 117         m_notifier.setWebConnection(type, maxBandwidthMbps);
 118         testing::runPendingTasks();
 119     }
 120
 121     void addObserverOnNotification(StateObserver* observer, StateObserver* observerToAdd)
 122     {
 123         observer->setNotificationCallback(bind(&NetworkStateNotifier::addObserver, &m_notifier, observerToAdd, executionContext()));
 124     }
 125
 126     void removeObserverOnNotification(StateObserver* observer, StateObserver* observerToRemove)
 127     {
 128         observer->setNotificationCallback(bind(&NetworkStateNotifier::removeObserver, &m_notifier, observerToRemove, executionContext()));
 129     }
 130
 131     bool verifyObservations(const StateObserver& observer, WebConnectionType type, double maxBandwidthMbps)
 132     {
 133         EXPECT_EQ(observer.observedType(), type);
 134         EXPECT_EQ(observer.observedMaxBandwidth(), maxBandwidthMbps);
 135         return observer.observedType() == type && observer.observedMaxBandwidth() == maxBandwidthMbps;
 136     }
 137
 138     RefPtrWillBePersistent<Document> m_document;
 139     RefPtrWillBePersistent<Document> m_document2;
 140     NetworkStateNotifier m_notifier;
 141 };
 142
 143 TEST_F(NetworkStateNotifierTest, AddObserver)
 144 {
 145     StateObserver observer;
 146     m_notifier.addObserver(&observer, executionContext());
 147     EXPECT_TRUE(verifyObservations(observer, WebConnectionTypeNone, kNoneMaxBandwidthMbps));
 148
 149     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 150     EXPECT_TRUE(verifyObservations(observer, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 151     EXPECT_EQ(observer.callbackCount(), 1);
 152 }
 153
 154 TEST_F(NetworkStateNotifierTest, RemoveObserver)
 155 {
 156     StateObserver observer1, observer2;
 157     m_notifier.addObserver(&observer1, executionContext());
 158     m_notifier.removeObserver(&observer1, executionContext());
 159     m_notifier.addObserver(&observer2, executionContext());
 160
 161     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 162     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeNone, kNoneMaxBandwidthMbps));
 163     EXPECT_TRUE(verifyObservations(observer2, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 164 }
 165
 166 TEST_F(NetworkStateNotifierTest, RemoveSoleObserver)
 167 {
 168     StateObserver observer1;
 169     m_notifier.addObserver(&observer1, executionContext());
 170     m_notifier.removeObserver(&observer1, executionContext());
 171
 172     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 173     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeNone, kNoneMaxBandwidthMbps));
 174 }
 175
 176 TEST_F(NetworkStateNotifierTest, AddObserverWhileNotifying)
 177 {
 178     StateObserver observer1, observer2;
 179     m_notifier.addObserver(&observer1, executionContext());
 180     addObserverOnNotification(&observer1, &observer2);
 181
 182     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 183     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 184     EXPECT_TRUE(verifyObservations(observer2, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 185 }
 186
 187 TEST_F(NetworkStateNotifierTest, RemoveSoleObserverWhileNotifying)
 188 {
 189     StateObserver observer1;
 190     m_notifier.addObserver(&observer1, executionContext());
 191     removeObserverOnNotification(&observer1, &observer1);
 192
 193     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 194     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 195
 196     setConnection(WebConnectionTypeEthernet, kEthernetMaxBandwidthMbps);
 197     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 198 }
 199
 200 TEST_F(NetworkStateNotifierTest, RemoveCurrentObserverWhileNotifying)
 201 {
 202     StateObserver observer1, observer2;
 203     m_notifier.addObserver(&observer1, executionContext());
 204     m_notifier.addObserver(&observer2, executionContext());
 205     removeObserverOnNotification(&observer1, &observer1);
 206
 207     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 208     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 209     EXPECT_TRUE(verifyObservations(observer2, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 210
 211     setConnection(WebConnectionTypeEthernet, kEthernetMaxBandwidthMbps);
 212     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 213     EXPECT_TRUE(verifyObservations(observer2, WebConnectionTypeEthernet, kEthernetMaxBandwidthMbps));
 214 }
 215
 216 TEST_F(NetworkStateNotifierTest, RemovePastObserverWhileNotifying)
 217 {
 218     StateObserver observer1, observer2;
 219     m_notifier.addObserver(&observer1, executionContext());
 220     m_notifier.addObserver(&observer2, executionContext());
 221     removeObserverOnNotification(&observer2, &observer1);
 222
 223     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 224     EXPECT_EQ(observer1.observedType(), WebConnectionTypeBluetooth);
 225     EXPECT_EQ(observer2.observedType(), WebConnectionTypeBluetooth);
 226
 227     setConnection(WebConnectionTypeEthernet, kEthernetMaxBandwidthMbps);
 228     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 229     EXPECT_TRUE(verifyObservations(observer2, WebConnectionTypeEthernet, kEthernetMaxBandwidthMbps));
 230 }
 231
 232 TEST_F(NetworkStateNotifierTest, RemoveFutureObserverWhileNotifying)
 233 {
 234     StateObserver observer1, observer2, observer3;
 235     m_notifier.addObserver(&observer1, executionContext());
 236     m_notifier.addObserver(&observer2, executionContext());
 237     m_notifier.addObserver(&observer3, executionContext());
 238     removeObserverOnNotification(&observer1, &observer2);
 239
 240     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 241     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 242     EXPECT_TRUE(verifyObservations(observer2, WebConnectionTypeNone, kNoneMaxBandwidthMbps));
 243     EXPECT_TRUE(verifyObservations(observer3, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 244 }
 245
 246 TEST_F(NetworkStateNotifierTest, MultipleContextsAddObserver)
 247 {
 248     StateObserver observer1, observer2;
 249     m_notifier.addObserver(&observer1, executionContext());
 250     m_notifier.addObserver(&observer2, executionContext2());
 251
 252     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 253     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 254     EXPECT_TRUE(verifyObservations(observer2, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 255 }
 256
 257 TEST_F(NetworkStateNotifierTest, RemoveContext)
 258 {
 259     StateObserver observer1, observer2;
 260     m_notifier.addObserver(&observer1, executionContext());
 261     m_notifier.addObserver(&observer2, executionContext2());
 262     m_notifier.removeObserver(&observer2, executionContext2());
 263
 264     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 265     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps));
 266     EXPECT_TRUE(verifyObservations(observer2, WebConnectionTypeNone, kNoneMaxBandwidthMbps));
 267 }
 268
 269 TEST_F(NetworkStateNotifierTest, RemoveAllContexts)
 270 {
 271     StateObserver observer1, observer2;
 272     m_notifier.addObserver(&observer1, executionContext());
 273     m_notifier.addObserver(&observer2, executionContext2());
 274     m_notifier.removeObserver(&observer1, executionContext());
 275     m_notifier.removeObserver(&observer2, executionContext2());
 276
 277     setConnection(WebConnectionTypeBluetooth, kBluetoothMaxBandwidthMbps);
 278     EXPECT_TRUE(verifyObservations(observer1, WebConnectionTypeNone, kNoneMaxBandwidthMbps));
 279     EXPECT_TRUE(verifyObservations(observer2, WebConnectionTypeNone, kNoneMaxBandwidthMbps));
 280 }
 281
 282 } // namespace blink