[DevTools] Network.emulateNetworkConditions now affects NetworkStateNotifier.

third_party/WebKit/Source/core/page/NetworkStateNotifier.h

 /*
  * Copyright (C) 2008 Apple Inc. All Rights Reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
@@ skipping 30 matching lines @@
 class CORE_EXPORT NetworkStateNotifier {
     WTF_MAKE_NONCOPYABLE(NetworkStateNotifier); USING_FAST_MALLOC(NetworkStateNotifier);
 public:
     class NetworkStateObserver {
     public:
         // Will be called on the thread of the context passed in addObserver.
         virtual void connectionChange(WebConnectionType, double maxBandwidthMbps) = 0;
     };
 
     NetworkStateNotifier()
-        : m_initialized(false)
+        : m_onLineInitialized(false)
         , m_isOnLine(true)
+        , m_connectionInitialized(false)
         , m_type(WebConnectionTypeOther)
         , m_maxBandwidthMbps(kInvalidMaxBandwidth)
-        , m_testUpdatesOnly(false)
+        , m_hasOverride(false)
+        , m_overrideOnLine(true)
+        , m_overrideType(WebConnectionTypeOther)
+        , m_overrideMaxBandwidthMbps(kInvalidMaxBandwidth)
     {
     }
 
     // Can be called on any thread.
     bool onLine() const
     {
         MutexLocker locker(m_mutex);
-        ASSERT(m_initialized);
-        return m_isOnLine;
+        DCHECK(m_hasOverride || m_onLineInitialized);
+        return m_hasOverride ? m_overrideOnLine : m_isOnLine;
     }
 
     void setOnLine(bool);
 
     // Can be called on any thread.
     WebConnectionType connectionType() const
     {
         MutexLocker locker(m_mutex);
-        ASSERT(m_initialized);
-        return m_type;
+        DCHECK(m_hasOverride || m_connectionInitialized);
+        return m_hasOverride ? m_overrideType : m_type;
     }
 
     // Can be called on any thread.
     bool isCellularConnectionType() const
     {
         switch (connectionType()) {
         case WebConnectionTypeCellular2G:
         case WebConnectionTypeCellular3G:
         case WebConnectionTypeCellular4G:
             return true;
         case WebConnectionTypeBluetooth:
         case WebConnectionTypeEthernet:
         case WebConnectionTypeWifi:
         case WebConnectionTypeWimax:
         case WebConnectionTypeOther:
         case WebConnectionTypeNone:
         case WebConnectionTypeUnknown:
             return false;
         }
         ASSERT_NOT_REACHED();
         return false;
     }
 
     // Can be called on any thread.
     double maxBandwidth() const
     {
         MutexLocker locker(m_mutex);
-        ASSERT(m_initialized);
-        return m_maxBandwidthMbps;
+        DCHECK(m_hasOverride || m_connectionInitialized);
+        return m_hasOverride ? m_overrideMaxBandwidthMbps : m_maxBandwidthMbps;
     }
 
     void setWebConnection(WebConnectionType, double maxBandwidthMbps);
 
+    // When called, successive setWebConnectionType/setOnLine calls are ignored,
+    // and supplied overridden values are used instead.
+    // This is used for layout tests (see crbug.com/377736) and inspector emulation.
+    //
+    // Since this class is a singleton, tests must clear override when completed to
+    // avoid indeterminate state across the test harness. When switching in or out of test
+    // mode, all state will be reset to default values.
+    void setOverride(bool onLine, WebConnectionType, double maxBandwidthMbps);
+    void clearOverride();
+
     // Must be called on the context's thread. An added observer must be removed
     // before its ExecutionContext is deleted. It's possible for an observer to
     // be called twice for the same event if it is first removed and then added
     // during notification.
     void addObserver(NetworkStateObserver*, ExecutionContext*);
     void removeObserver(NetworkStateObserver*, ExecutionContext*);
 
-    // The following functions are for testing purposes.
-
-    // When true, setWebConnectionType calls are ignored and only setWebConnectionTypeForTest
-    // can update the connection type. This is used for layout tests (see crbug.com/377736).
-    //
-    // Since this class is a singleton, tests must call this with false when completed to
-    // avoid indeterminate state across the test harness. When switching in or out of test
-    // mode, all state will be reset to default values.
-    void setTestUpdatesOnly(bool);
-    // Tests should call this as it will change the type regardless of the value of m_testUpdatesOnly.
-    void setWebConnectionForTest(WebConnectionType, double maxBandwidthMbps);
-
 private:
     struct ObserverList {
         ObserverList()
             : iterating(false)
         {
         }
         bool iterating;
         Vector<NetworkStateObserver*> observers;
         Vector<size_t> zeroedObservers; // Indices in observers that are 0.
     };
 
     const int kInvalidMaxBandwidth = -1;
 
-    void setWebConnectionImpl(WebConnectionType, double maxBandwidthMbps);
-    void setMaxBandwidthImpl(double maxBandwidthMbps);
-
     // The ObserverListMap is cross-thread accessed, adding/removing Observers running
     // within an ExecutionContext. Kept off-heap to ease cross-thread allocation and use;
     // the observers are (already) responsible for explicitly unregistering while finalizing.
     using ObserverListMap = HashMap<UntracedMember<ExecutionContext>, std::unique_ptr<ObserverList>>;
 
+    void notifyObservers();
     void notifyObserversOfConnectionChangeOnContext(WebConnectionType, double maxBandwidthMbps, ExecutionContext*);
 
     ObserverList* lockAndFindObserverList(ExecutionContext*);
 
     // Removed observers are nulled out in the list in case the list is being
     // iterated over. Once done iterating, call this to clean up nulled
     // observers.
     void collectZeroedObservers(ObserverList*, ExecutionContext*);
 
     mutable Mutex m_mutex;
-    bool m_initialized;
+    bool m_onLineInitialized;
     bool m_isOnLine;
+    bool m_connectionInitialized;
     WebConnectionType m_type;
     double m_maxBandwidthMbps;
     ObserverListMap m_observers;
-    bool m_testUpdatesOnly;
+    bool m_hasOverride;
+    bool m_overrideOnLine;
+    WebConnectionType m_overrideType;
+    double m_overrideMaxBandwidthMbps;
 };
 
 CORE_EXPORT NetworkStateNotifier& networkStateNotifier();
 
 } // namespace blink
 
 #endif // NetworkStateNotifier_h