Home | History | Annotate | Download | only in functional 
      1 #!/usr/bin/env python
      2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
      3 # Use of this source code is governed by a BSD-style license that can be
      4 # found in the LICENSE file.
      5 
      6 import logging
      7 import os
      8 import time
      9 
     10 import pyauto_functional
     11 import pyauto
     12 import test_utils
     13 
     14 
     15 class NetflixTestHelper():
     16   """Helper functions for Netflix tests.
     17      
     18   For sample usage, look at class NetflixTest.
     19   """
     20 
     21   # Netflix player states.
     22   IS_GUEST_MODE_ERROR = '0'
     23   IS_PLAYING = '4'
     24 
     25   TITLE_HOMEPAGE = 'http://movies.netflix.com/WiHome'
     26   SIGNOUT_PAGE = 'https://account.netflix.com/Logout'
     27   # 30 Rock.
     28   VIDEO_URL = 'https://movies.netflix.com/WiPlayer?movieid=70136124'
     29   ALT_VIDEO_URL = 'https://movies.netflix.com/WiPlayer?movieid=70133713'
     30   _pyauto = None
     31 
     32   def __init__(self, pyauto):
     33     self._pyauto = pyauto
     34 
     35   def _IsNetflixPluginEnabled(self):
     36     """Determine Netflix plugin availability and its state."""
     37     return [x for x in self._pyauto.GetPluginsInfo().Plugins() \
     38             if x['name'] == 'Netflix' and x['enabled']]
     39 
     40   def _LoginToNetflix(self):
     41     """Login to Netflix."""
     42     credentials = self._pyauto.GetPrivateInfo()['test_netflix_acct']
     43     board_name = self._pyauto.ChromeOSBoard()
     44     assert credentials.get(board_name), \
     45            'No netflix credentials for %s.' % board_name
     46     self._pyauto.NavigateToURL(credentials['login_url'])
     47     login_js = """
     48         document.getElementById('email').value='%s';
     49         document.getElementById('password').value='%s';
     50         window.domAutomationController.send('ok');
     51     """ % (credentials[board_name], credentials['password'])
     52     self._pyauto.assertEqual(self._pyauto.ExecuteJavascript(login_js), 'ok',
     53         msg='Failed to set login credentials.')
     54     self._pyauto.assertTrue(self._pyauto.SubmitForm('login-form'),
     55         msg='Login to Netflix failed. We think this is an authetication '
     56             'problem from the Netflix side. Sometimes we also see this while '
     57             'login in manually.')
     58     
     59   def _GetVideoDroppedFrames(self, tab_index=0, windex=0):
     60     """Returns total Netflix video dropped frames."""
     61     js = """
     62         var frames = nrdp.video.droppedFrames; 
     63         window.domAutomationController.send(frames + '');
     64     """
     65     return int(self._pyauto.ExecuteJavascript(js, tab_index=tab_index,
     66                                               windex=windex))
     67 
     68   def _GetVideoFrames(self, tab_index=0, windex=0):
     69     """Returns Netflix video total frames."""
     70     js = """
     71         var frames = nrdp.video.totalFrames; 
     72         window.domAutomationController.send(frames + '');
     73     """
     74     return int(self._pyauto.ExecuteJavascript(js, tab_index=tab_index,
     75                                               windex=windex))
     76 
     77   def _HandleInfobars(self, err_msg):
     78     """Manage infobars that come up during the test."""
     79     def _HandleNetflixInfobar():
     80       tab_info = self._pyauto.GetBrowserInfo()['windows'][0]['tabs'][0]
     81       infobars = tab_info['infobars']
     82       index = 0
     83       for infobar in infobars:
     84          if 'netflix' in infobar['text']:
     85            # After storage infobar pops up, clicking the Ok button immediately
     86            # returns the Storage error on faster machines like Stumpy/Lumpy so
     87            # adding a delay of 1 second here.
     88            time.sleep(1)
     89            self._pyauto.PerformActionOnInfobar('accept', infobar_index=index)
     90            return True
     91          index = index + 1
     92       return False
     93     self._pyauto.assertTrue(self._pyauto.WaitUntil(_HandleNetflixInfobar),
     94                             msg=err_msg)
     95  
     96   def CurrentPlaybackTime(self):
     97     """Returns the current playback time in seconds."""
     98     time = self._pyauto.ExecuteJavascript("""
     99         time = nrdp.video.currentTime;
    100         window.domAutomationController.send(time + '');
    101     """)
    102     return int(float(time))
    103 
    104   def SignOut(self):
    105     """Sign out from Netflix Login."""
    106     self._pyauto.NavigateToURL(self.SIGNOUT_PAGE)
    107 
    108   def LoginAndStartPlaying(self):
    109     """Login and start playing the video."""
    110     self._pyauto.assertTrue(self._pyauto._IsNetflixPluginEnabled(), 
    111                             msg='Netflix plugin is disabled or not available.')
    112     self._pyauto._LoginToNetflix()
    113     self._pyauto.assertTrue(self._pyauto.WaitUntil(
    114         lambda: self._pyauto.GetActiveTabURL().spec(),
    115         expect_retval=self.TITLE_HOMEPAGE),
    116         msg='Login to Netflix failed.')
    117     self._pyauto.NavigateToURL(self.VIDEO_URL)
    118     self._pyauto._HandleInfobars(err_msg='Netflix infobar did not show up')
    119 
    120   def CheckNetflixPlaying(self, expected_result, error_msg):
    121     """Check if Netflix is playing the video or not.
    122 
    123     Args:
    124       expected_result: expected return value from Netflix player.
    125       error_msg: If expected value isn't matching, error message to throw.
    126     """
    127     self._pyauto.assertTrue(self._pyauto.WaitUntil(
    128         lambda: self._pyauto.ExecuteJavascript("""
    129             if (typeof nrdp == 'undefined') {
    130               window.domAutomationController.send('not ready');
    131             }
    132             player_status = nrdp.video.readyState;
    133             window.domAutomationController.send(player_status + '');
    134         """), expect_retval=expected_result),
    135         msg=error_msg)
    136 
    137 
    138 class NetflixTest(pyauto.PyUITest, NetflixTestHelper):
    139   """Test case for Netflix player."""
    140 
    141   def __init__(self, methodName='runTest', **kwargs):
    142     pyauto.PyUITest.__init__(self, methodName, **kwargs)
    143     NetflixTestHelper.__init__(self, self)
    144 
    145   def ShouldAutoLogin(self):
    146     return False
    147 
    148   def _Login(self):
    149     """Perform login"""
    150     credentials = self.GetPrivateInfo()['test_google_account']
    151     self.Login(credentials['username'], credentials['password'])
    152     logging.info('Logged in as %s' % credentials['username'])
    153     login_info = self.GetLoginInfo()
    154     self.assertTrue(login_info['is_logged_in'], msg='Login failed.')
    155     self.assertFalse(login_info['is_guest'],
    156                      msg='Should not be logged in as guest.')
    157 
    158   def setUp(self):
    159     assert os.geteuid() == 0, 'Run test as root since we might need to logout'
    160     pyauto.PyUITest.setUp(self)
    161     if self.GetLoginInfo()['is_logged_in']:
    162       self.Logout()
    163     self._Login()
    164 
    165   def tearDown(self):
    166     self.SignOut()
    167     pyauto.PyUITest.tearDown(self)
    168 
    169   def testPlayerLoadsAndPlays(self):
    170     """Test that Netflix player loads and plays the title."""
    171     self.LoginAndStartPlaying()
    172     self._HandleInfobars(err_msg='Netflix plugin access infobar did not show up')
    173     self.CheckNetflixPlaying(self.IS_PLAYING,
    174                               'Player did not start playing the title.')
    175 
    176   def testMultiplePlayback(self):
    177     """Test that playing two titles, Netflix returns multiple play error."""
    178     self.LoginAndStartPlaying()
    179     self._HandleInfobars(err_msg='Netflix plugin access infobar did not show up')
    180     self.CheckNetflixPlaying(self.IS_PLAYING,
    181                               'Player did not start playing the title.')
    182     self.AppendTab(self.ALT_VIDEO_URL)
    183     self.assertTrue('Multiple Play Error' in self.GetTabContents(),
    184                     msg='Multiple Play Error is not found on the page.')
    185 
    186   def testPlaying(self):
    187     """Test that title playing progresses."""
    188     self.LoginAndStartPlaying()
    189     self._HandleInfobars(err_msg='Netflix plugin access infobar did not show up')
    190     self.CheckNetflixPlaying(self.IS_PLAYING,
    191                               'Player did not start playing the title.')
    192     title_length =  self.ExecuteJavascript("""
    193         time = nrdp.video.duration;
    194         window.domAutomationController.send(time + '');
    195     """)
    196     title_length = int(float(title_length))
    197     prev_time = 0
    198     current_time = 0
    199     count = 0
    200     while current_time < title_length:
    201       # We want to test playing only for ten seconds.
    202       count = count + 1
    203       if count == 10:
    204         break
    205       current_time = self.CurrentPlaybackTime()
    206       self.assertTrue(prev_time <= current_time,
    207           msg='Prev playing time %s is greater than current time %s.'
    208           % (prev_time, current_time))
    209       prev_time = current_time
    210       # play video for some time 
    211       time.sleep(1)
    212     # In case player doesn't start playing at all, above while loop may
    213     # still pass. So re-verifying and assuming that player did play something
    214     # during last 10 seconds.
    215     self.assertTrue(current_time > 0,
    216                     msg='Netflix player did not start playing.')
    217 
    218 
    219 class NetflixGuestModeTest(pyauto.PyUITest, NetflixTestHelper):
    220   """Netflix in guest mode."""
    221 
    222   def __init__(self, methodName='runTest', **kwargs):
    223     pyauto.PyUITest.__init__(self, methodName, **kwargs)
    224     NetflixTestHelper.__init__(self, self)
    225 
    226   def setUp(self):
    227     assert os.geteuid() == 0, 'Run test as root since we might need to logout'
    228     pyauto.PyUITest.setUp(self)
    229     if self.GetLoginInfo()['is_logged_in']:
    230       self.Logout()
    231     self.LoginAsGuest()
    232     login_info = self.GetLoginInfo()
    233     self.assertTrue(login_info['is_logged_in'], msg='Not logged in at all.')
    234     self.assertTrue(login_info['is_guest'], msg='Not logged in as guest.')
    235 
    236   def ShouldAutoLogin(self):
    237     return False
    238 
    239   def tearDown(self):
    240     self.AppendTab(self.SIGNOUT_PAGE)
    241     self.Logout()
    242     pyauto.PyUITest.tearDown(self)
    243 
    244   def testGuestMode(self):
    245     """Test that Netflix doesn't play in guest mode login."""
    246     self.LoginAndStartPlaying()
    247     self.CheckNetflixPlaying(
    248         self.IS_GUEST_MODE_ERROR,
    249         'Netflix player did not return a Guest mode error.')
    250     # crosbug.com/p/14009
    251     self.assertTrue('Netflix Video Player Unavailable' in self.GetTabContents(),
    252                     msg='Guest Mode error is not found on the page.')
    253     
    254 
    255 if __name__ == '__main__':
    256   pyauto_functional.Main()
    257