# Copyright 2021 The Cobalt Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
""" Tests of suspend and resume during playing."""

import time
import unittest

import _env  # pylint: disable=unused-import
from cobalt.media_integration_tests.test_app import Features
from cobalt.media_integration_tests.test_case import TestCase


class SuspendResumeTest(TestCase):
  """
    Test cases for suspend and resume.
  """

  def run_test_with_url(self, url):
    app = self.CreateCobaltApp(url)
    with app:
      # Wait until the playback starts.
      app.WaitUntilPlayerStart()
      app.WaitUntilAdsEnd()
      # Let the playback play for 2 seconds.
      app.WaitUntilMediaTimeReached(app.CurrentMediaTime() + 2)
      # Suspend the app and wait the app enters background.
      app.Suspend()
      app.WaitUntilReachState(
          lambda _app: not _app.ApplicationState().is_visible)
      # Wait for 1 second before resume.
      time.sleep(1)
      # Resume the app and let it play for a few time.
      app.Resume()
      app.WaitUntilPlayerStart()
      app.WaitUntilAdsEnd()
      # Let the playback play for 2 seconds.
      app.WaitUntilMediaTimeReached(app.CurrentMediaTime() + 2)

  @unittest.skipIf(not TestCase.IsFeatureSupported(Features.SUSPEND_AND_RESUME),
                   'Suspend and resume is not supported on this platform.')
  def test_playback_h264(self):
    self.run_test_with_url('https://www.youtube.com/tv#/watch?v=RACW52qnJMI')

  @unittest.skipIf(not TestCase.IsFeatureSupported(Features.SUSPEND_AND_RESUME),
                   'Suspend and resume is not supported on this platform.')
  def test_encrypted_playback(self):
    self.run_test_with_url('https://www.youtube.com/tv#/watch?v=Vx5lkGS4w30')

  @unittest.skipIf(not TestCase.IsFeatureSupported(Features.SUSPEND_AND_RESUME),
                   'Suspend and resume is not supported on this platform.')
  def test_live_stream(self):
    self.run_test_with_url('https://www.youtube.com/tv#/watch?v=KI1XlTQrsa0')

  # Test for vp9 playback if supported.
  @unittest.skipIf(not TestCase.IsFeatureSupported(Features.SUSPEND_AND_RESUME),
                   'Suspend and resume is not supported on this platform.')
  def test_playback_vp9(self):
    self.run_test_with_url('https://www.youtube.com/tv#/watch?v=1La4QzGeaaQ')

  # Test for av1 playback if supported.
  @unittest.skipIf(not TestCase.IsFeatureSupported(Features.SUSPEND_AND_RESUME),
                   'Suspend and resume is not supported on this platform.')
  def test_playback_av1(self):
    self.run_test_with_url('https://www.youtube.com/tv#/watch?v=iXvy8ZeCs5M')

  # Test for vertical playback
  @unittest.skipIf(not TestCase.IsFeatureSupported(Features.SUSPEND_AND_RESUME),
                   'Suspend and resume is not supported on this platform.')
  def test_vertical_playback(self):
    self.run_test_with_url('https://www.youtube.com/tv#/watch?v=jNQXAC9IVRw')