Media Controls: Remove download button for infinite streams

third_party/WebKit/Source/core/html/shadow/MediaControlsTest.cpp

 // Copyright 2015 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/shadow/MediaControls.h"
 
+#include <limits>
+#include <memory>
 #include "core/HTMLNames.h"
 #include "core/css/StylePropertySet.h"
 #include "core/dom/Document.h"
 #include "core/dom/ElementTraversal.h"
 #include "core/dom/StyleEngine.h"
 #include "core/frame/Settings.h"
 #include "core/html/HTMLVideoElement.h"
 #include "core/loader/EmptyClients.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/heap/Handle.h"
 #include "platform/testing/UnitTestHelpers.h"
 #include "public/platform/WebMediaPlayer.h"
 #include "public/platform/WebSize.h"
 #include "public/platform/modules/remoteplayback/WebRemotePlaybackAvailability.h"
 #include "public/platform/modules/remoteplayback/WebRemotePlaybackClient.h"
 #include "testing/gtest/include/gtest/gtest.h"
-#include <memory>
 
 namespace blink {
 
 namespace {
 
 class MockVideoWebMediaPlayer : public WebMediaPlayer {
  public:
+  // WebMediaPlayer overrides:
   void load(LoadType, const WebMediaPlayerSource&, CORSMode) override{};
   void play() override{};
   void pause() override{};
   bool supportsSave() const override { return false; };
   void seek(double seconds) override{};
   void setRate(double) override{};
   void setVolume(double) override{};
   WebTimeRanges buffered() const override { return WebTimeRanges(); };
   WebTimeRanges seekable() const override { return WebTimeRanges(); };
   void setSinkId(const WebString& sinkId,
@@ skipping 120 matching lines @@
   void ensureLayout() {
     // Force a relayout, so that the controls know the width.  Otherwise,
     // they don't know if, for example, the cast button will fit.
     m_mediaControls->mediaElement().clientWidth();
   }
 
   void simulateHideMediaControlsTimerFired() {
     m_mediaControls->hideMediaControlsTimerFired(nullptr);
   }
 
+  void simulateLoadedMetadata() { m_mediaControls->onLoadedMetadata(); }
+
   MediaControls& mediaControls() { return *m_mediaControls; }
   Document& document() { return m_pageHolder->document(); }
 
  private:
   std::unique_ptr<DummyPageHolder> m_pageHolder;
   Persistent<MediaControls> m_mediaControls;
 };
 
 TEST_F(MediaControlsTest, HideAndShow) {
   mediaControls().mediaElement().setBooleanAttribute(HTMLNames::controlsAttr,
@@ skipping 127 matching lines @@
 
   mediaControls().show();
   mediaControls().toggleOverflowMenu();
   EXPECT_TRUE(isElementVisible(*overflowList));
 
   simulateHideMediaControlsTimerFired();
   EXPECT_TRUE(isElementVisible(*overflowList));
   EXPECT_TRUE(isElementVisible(*panel));
 }
 
+TEST_F(MediaControlsTest, DownloadButtonDisplayed) {
+  ensureLayout();
+
+  Element* downloadButton = getElementByShadowPseudoId(
+      mediaControls(), "-internal-media-controls-download-button");
+  ASSERT_NE(nullptr, downloadButton);
+
+  mediaControls().mediaElement().setSrc("https://example.com/foo.mp4");
+  testing::runPendingTasks();
+  simulateLoadedMetadata();
+
+  // Download button should normally be displayed.
+  EXPECT_TRUE(isElementVisible(*downloadButton));
+}
+
+TEST_F(MediaControlsTest, DownloadButtonNotDisplayedEmptyUrl) {
+  ensureLayout();
+
+  Element* downloadButton = getElementByShadowPseudoId(
+      mediaControls(), "-internal-media-controls-download-button");
+  ASSERT_NE(nullptr, downloadButton);
+
+  // Download button should not be displayed when URL is empty.
+  mediaControls().mediaElement().setSrc("");
+  testing::runPendingTasks();
+  simulateLoadedMetadata();
+  EXPECT_FALSE(isElementVisible(*downloadButton));
+}
+
+TEST_F(MediaControlsTest, DownloadButtonNotDisplayedInfiniteDuration) {
+  ensureLayout();
+
+  Element* downloadButton = getElementByShadowPseudoId(
+      mediaControls(), "-internal-media-controls-download-button");
+  ASSERT_NE(nullptr, downloadButton);
+
+  mediaControls().mediaElement().setSrc("https://example.com/foo.mp4");
+  testing::runPendingTasks();
+
+  // Download button should not be displayed when duration is infinite.
+  mediaControls().mediaElement().durationChanged(
+      std::numeric_limits<double>::infinity(), false /* requestSeek */);
+  simulateLoadedMetadata();
+  EXPECT_FALSE(isElementVisible(*downloadButton));
+}
+
+TEST_F(MediaControlsTest, DownloadButtonNotDisplayedHLS) {
+  ensureLayout();
+
+  Element* downloadButton = getElementByShadowPseudoId(
+      mediaControls(), "-internal-media-controls-download-button");
+  ASSERT_NE(nullptr, downloadButton);
+
+  // Download button should not be displayed for HLS streams.
+  mediaControls().mediaElement().setSrc("https://example.com/foo.m3u8");
+  testing::runPendingTasks();
+  simulateLoadedMetadata();
+  EXPECT_FALSE(isElementVisible(*downloadButton));
+}
+
 }  // namespace blink