Support DataSaver for SharedWorker

third_party/WebKit/Source/core/html/HTMLMediaElementTest.cpp

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "core/html/HTMLMediaElement.h"
 
 #include "core/frame/Settings.h"
 #include "core/html/HTMLAudioElement.h"
 #include "core/html/HTMLVideoElement.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/network/NetworkStateNotifier.h"
+#include "public/web/WebDataSaverFlag.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
 
 enum class TestParam { kAudio, kVideo };
 
 class HTMLMediaElementTest : public ::testing::TestWithParam<TestParam> {
  protected:
   void SetUp() {
     dummy_page_holder_ = DummyPageHolder::Create();
@@ skipping 63 matching lines @@
     case TestURLScheme::kBlob:
       return "blob:http://example.com/00000000-0000-0000-0000-000000000000";
     default:
       NOTREACHED();
   }
   return g_empty_string;
 }
 
 TEST_P(HTMLMediaElementTest, preloadType) {
   struct TestData {
-    bool data_saver_enabled;
+    WebDataSaverFlag data_saver_flag;
     bool force_preload_none_for_media_elements;
     bool is_cellular;
     TestURLScheme src_scheme;
     AtomicString preload_to_set;
     AtomicString preload_expected;
   } test_data[] = {
       // Tests for conditions in which preload type should be overriden to
       // "none".
-      {false, true, false, TestURLScheme::kHttp, "auto", "none"},
-      {true, true, false, TestURLScheme::kHttps, "auto", "none"},
-      {true, true, false, TestURLScheme::kFtp, "metadata", "none"},
-      {false, false, false, TestURLScheme::kHttps, "auto", "auto"},
-      {false, true, false, TestURLScheme::kFile, "auto", "auto"},
-      {false, true, false, TestURLScheme::kData, "metadata", "metadata"},
-      {false, true, false, TestURLScheme::kBlob, "auto", "auto"},
-      {false, true, false, TestURLScheme::kFile, "none", "none"},
+      {WebDataSaverFlag::kDisabled, true, false, TestURLScheme::kHttp, "auto",
+       "none"},
+      {WebDataSaverFlag::kEnabled, true, false, TestURLScheme::kHttps, "auto",
+       "none"},
+      {WebDataSaverFlag::kEnabled, true, false, TestURLScheme::kFtp, "metadata",
+       "none"},
+      {WebDataSaverFlag::kDisabled, false, false, TestURLScheme::kHttps, "auto",
+       "auto"},
+      {WebDataSaverFlag::kDisabled, true, false, TestURLScheme::kFile, "auto",
+       "auto"},
+      {WebDataSaverFlag::kDisabled, true, false, TestURLScheme::kData,
+       "metadata", "metadata"},
+      {WebDataSaverFlag::kDisabled, true, false, TestURLScheme::kBlob, "auto",
+       "auto"},
+      {WebDataSaverFlag::kDisabled, true, false, TestURLScheme::kFile, "none",
+       "none"},
       // Tests for conditions in which preload type should be overriden to
       // "metadata".
-      {false, false, true, TestURLScheme::kHttp, "auto", "metadata"},
-      {false, false, true, TestURLScheme::kHttp, "scheme", "metadata"},
-      {false, false, true, TestURLScheme::kHttp, "none", "none"},
+      {WebDataSaverFlag::kDisabled, false, true, TestURLScheme::kHttp, "auto",
+       "metadata"},
+      {WebDataSaverFlag::kDisabled, false, true, TestURLScheme::kHttp, "scheme",
+       "metadata"},
+      {WebDataSaverFlag::kDisabled, false, true, TestURLScheme::kHttp, "none",
+       "none"},
       // Tests that the preload is overriden to "auto"
-      {false, false, false, TestURLScheme::kHttp, "foo", "auto"},
+      {WebDataSaverFlag::kDisabled, false, false, TestURLScheme::kHttp, "foo",
+       "auto"},
   };
 
   int index = 0;
   for (const auto& data : test_data) {
-    Media()->GetDocument().GetSettings()->SetDataSaverEnabled(
-        data.data_saver_enabled);
+    Media()->GetDocument().GetSettings()->SetDataSaverFlag(
+        data.data_saver_flag);
     Media()->GetDocument().GetSettings()->SetForcePreloadNoneForMediaElements(
         data.force_preload_none_for_media_elements);
     if (data.is_cellular) {
       GetNetworkStateNotifier().SetOverride(
           true, WebConnectionType::kWebConnectionTypeCellular3G, 2.0);
     } else {
       GetNetworkStateNotifier().ClearOverride();
     }
     SetCurrentSrc(SrcSchemeToURL(data.src_scheme));
     Media()->setPreload(data.preload_to_set);
 
     EXPECT_EQ(data.preload_expected, Media()->preload())
         << "preload type differs at index" << index;
     ++index;
   }
 }
 
 }  // namespace blink