Add pyauto tests for the New Tab page.

chrome/test/functional/ntp.py

 #!/usr/bin/python
 # Copyright (c) 2010 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.
 
 import os
 
 import pyauto_functional  # Must be imported before pyauto
 import pyauto
+import test_utils
 
 
 class NTPTest(pyauto.PyUITest):
   """Test of the NTP."""
 
   def Debug(self):
     """Test method for experimentation.
 
     This method is not run automatically.
     """
     while True:
       raw_input('Interact with the browser and hit <enter> to dump NTP info...')
       print '*' * 20
       import pprint
       pp = pprint.PrettyPrinter(indent=2)
       pp.pprint(self._GetNTPInfo())
 
   def __init__(self, methodName='runTest'):
     super(NTPTest, self).__init__(methodName)
 
     # Create some dummy file urls we can use in the tests.
     filenames = ['title1.html', 'title2.html']
     titles = [u'', u'Title Of Awesomeness']
     urls = map(lambda name: self.GetFileURLForDataPath(name), filenames)
     self.PAGES = map(lambda url, title: {'url': url, 'title': title},
                      urls, titles)
 
+  def _NTPContainsThumbnail(self, check_thumbnail):

[Nirnimesh, 2010/11/24 09:58:23]

docstring please

[kkania, 2010/11/24 17:40:35]

Done.

+    for thumbnail in self.GetNTPThumbnails():
+      if check_thumbnail['url'] == thumbnail['url']:
+        return True
+    return False
+
   def testFreshProfile(self):
     """Tests that the NTP with a fresh profile is correct"""
     thumbnails = self.GetNTPThumbnails()
     default_sites = self.GetNTPDefaultSites()
     self.assertEqual(len(default_sites), len(thumbnails))
     for thumbnail, default_site in zip(thumbnails, default_sites):
       self.assertEqual(thumbnail['url'], default_site)
     self.assertEqual(0, len(self.GetNTPRecentlyClosed()))
 
   def testRemoveDefaultThumbnails(self):
     """Tests that the default thumbnails can be removed"""
     self.RemoveNTPDefaultThumbnails()
     self.assertFalse(self.GetNTPThumbnails())
     self.RestoreAllNTPThumbnails()
     self.assertEqual(len(self.GetNTPDefaultSites()),
                      len(self.GetNTPThumbnails()))
     self.RemoveNTPDefaultThumbnails()
     self.assertFalse(self.GetNTPThumbnails())
 
   def testOneMostVisitedSite(self):
     """Tests that a site is added to the most visited sites"""
     self.RemoveNTPDefaultThumbnails()
     self.NavigateToURL(self.PAGES[1]['url'])
-    self.assertEqual(self.PAGES[1]['url'], self.GetNTPThumbnails()[0]['url'])
-    self.assertEqual(self.PAGES[1]['title'],
-                     self.GetNTPThumbnails()[0]['title'])
-
-  def testOneRecentlyClosedTab(self):
-    """Tests that closing a tab populates the recently closed tabs list"""
-    self.RemoveNTPDefaultThumbnails()
-    self.AppendTab(pyauto.GURL(self.PAGES[1]['url']))
-    self.GetBrowserWindow(0).GetTab(1).Close(True)
-    self.assertEqual(self.PAGES[1]['url'],
-                     self.GetNTPRecentlyClosed()[0]['url'])
-    self.assertEqual(self.PAGES[1]['title'],
-                     self.GetNTPRecentlyClosed()[0]['title'])
+    thumbnail = self.GetNTPThumbnails()[0]
+    self.assertEqual(self.PAGES[1]['url'], thumbnail['url'])
+    self.assertEqual(self.PAGES[1]['title'], thumbnail['title'])
+    self.assertFalse(thumbnail['is_pinned'])
 
   def testMoveThumbnailBasic(self):
     """Tests moving a thumbnail to a different index"""
     self.RemoveNTPDefaultThumbnails()
     self.NavigateToURL(self.PAGES[0]['url'])
     self.NavigateToURL(self.PAGES[1]['url'])
     thumbnails = self.GetNTPThumbnails()
     self.MoveNTPThumbnail(thumbnails[0], 1)
     self.assertTrue(self.IsNTPThumbnailPinned(thumbnails[0]))
     self.assertFalse(self.IsNTPThumbnailPinned(thumbnails[1]))
     self.assertEqual(self.PAGES[0]['url'], self.GetNTPThumbnails()[1]['url'])
     self.assertEqual(1, self.GetNTPThumbnailIndex(thumbnails[0]))
 
   def testPinningThumbnailBasic(self):
     """Tests that we can pin/unpin a thumbnail"""
     self.RemoveNTPDefaultThumbnails()
     self.NavigateToURL(self.PAGES[0]['url'])
     thumbnail1 = self.GetNTPThumbnails()[0]
     self.assertFalse(self.IsNTPThumbnailPinned(thumbnail1))
     self.PinNTPThumbnail(thumbnail1)
     self.assertTrue(self.IsNTPThumbnailPinned(thumbnail1))
     self.UnpinNTPThumbnail(thumbnail1)
     self.assertFalse(self.IsNTPThumbnailPinned(thumbnail1))
 
+  def testRemoveThumbnail(self):
+    """Tests removing a thumbnail works"""
+    self.RemoveNTPDefaultThumbnails()
+    for i in range(len(self.PAGES)):

[Nirnimesh, 2010/11/24 09:58:23]

You don't really need |i|

for page in self.PAGES:
  self.AppendTab(pyauto.GURL(page['url']))
  ...

[kkania, 2010/11/24 17:40:35]

Done.

+      self.AppendTab(pyauto.GURL(self.PAGES[i]['url']))
+      self.assertTrue(self._NTPContainsThumbnail(self.PAGES[i]))
+
+    thumbnails = self.GetNTPThumbnails()
+    for i in range(len(thumbnails)):

[Nirnimesh, 2010/11/24 09:58:23]

ditto

[kkania, 2010/11/24 17:40:35]

Done.

+      self.assertEquals(thumbnails[i], self.GetNTPThumbnails()[0])

[Nirnimesh, 2010/11/24 09:58:23]

this line looks like a tautology
Did you mean to use PAGES?

[kkania, 2010/11/24 17:40:35]

I wanted to make sure that the thumbnails shift to the front when the first
thumbnail is removed

+      self.RemoveNTPThumbnail(thumbnails[i])
+      self.assertFalse(self._NTPContainsThumbnail(self.PAGES[i]))
+    self.assertFalse(self.GetNTPThumbnails())
+
+  def testIncognitoNotAppearInMostVisited(self):
+    """Tests that visiting a page in incognito mode does cause it to appear in
+    the Most Visited section"""
+    self.RemoveNTPDefaultThumbnails()
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
+    self.NavigateToURL(self.PAGES[0]['url'], 1, 0)
+    self.assertEquals([], self.GetNTPThumbnails())

[Nirnimesh, 2010/11/24 09:58:23]

assertFalse

[kkania, 2010/11/24 17:40:35]

I was wondering about this. assertEquals([], ...) has the nice property that
when it fails it will spit out what the thumbnails were, which is nice for
debugging. I guess it is not worth it though. Done.

+
+  def testRestoreOncePinnedThumbnail(self):
+    """Tests that after restoring a once pinned thumbnail, the thumbnail is
+    not pinned"""
+    self.RemoveNTPDefaultThumbnails()
+    self.NavigateToURL(self.PAGES[0]['url'])
+    thumbnail1 = self.GetNTPThumbnails()[0]
+    self.PinNTPThumbnail(thumbnail1)
+    self.RemoveNTPThumbnail(thumbnail1)
+    self.RestoreAllNTPThumbnails()
+    self.RemoveNTPDefaultThumbnails()
+    self.assertFalse(self.IsNTPThumbnailPinned(thumbnail1))
+
+  def testThumbnailPersistence(self):
+    """Tests that thumbnails persist across Chrome restarts"""
+    self.RemoveNTPDefaultThumbnails()
+    for page in self.PAGES:
+      self.AppendTab(pyauto.GURL(page['url']))
+    thumbnails = self.GetNTPThumbnails()
+    self.MoveNTPThumbnail(thumbnails[0], 1)
+    thumbnails = self.GetNTPThumbnails()
+

[Nirnimesh, 2010/11/24 09:58:23]

please assert that |thumbnails| is not empty

[kkania, 2010/11/24 17:40:35]

I am interested in your thoughts about this. I was thinking that this test
should assume that the move function works, since it is tested by
testMoveThumbnailBasic. This makes the test simpler and more focused.
Assumptions like this are made in several places in this and other files.

[Nirnimesh, 2010/11/24 20:38:11]

Ok

+    self.RestartBrowser(False)

[Nirnimesh, 2010/11/24 09:58:23]

use named arg. ie clear_profile=False

[kkania, 2010/11/24 17:40:35]

Done.

+    self.assertEqual(thumbnails, self.GetNTPThumbnails())
+
+  def testRestoreAllRemovedThumbnails(self):
+    """Tests restoring all removed thumnails"""

[Nirnimesh, 2010/11/24 09:58:23]

thumbnails

[kkania, 2010/11/24 17:40:35]

Done.

+    self.RemoveNTPDefaultThumbnails()
+    for page in self.PAGES:
+      self.AppendTab(pyauto.GURL(page['url']))
+
+    thumbnails = self.GetNTPThumbnails()
+    for thumbnail in thumbnails:
+      self.RemoveNTPThumbnail(thumbnail)
+
+    self.RestoreAllNTPThumbnails()
+    self.RemoveNTPDefaultThumbnails()

[Nirnimesh, 2010/11/24 09:58:23]

maybe this line and line 143 should be removed from this test

[kkania, 2010/11/24 17:40:35]

That's a good idea, done.

+    self.assertEquals(thumbnails, self.GetNTPThumbnails())
+
+  def testThumbnailRanking(self):
+    """Tests that the thumbnails are ordered according to visit count"""
+    self.RemoveNTPDefaultThumbnails()
+    for page in self.PAGES:
+      self.AppendTab(pyauto.GURL(page['url']))
+    thumbnails = self.GetNTPThumbnails()
+    self.assertEqual(self.PAGES[0]['url'], self.GetNTPThumbnails()[0]['url'])
+    self.AppendTab(pyauto.GURL(self.PAGES[1]['url']))
+    self.assertEqual(self.PAGES[1]['url'], self.GetNTPThumbnails()[0]['url'])
+    self.AppendTab(pyauto.GURL(self.PAGES[0]['url']))
+    self.AppendTab(pyauto.GURL(self.PAGES[0]['url']))
+    self.assertEqual(self.PAGES[0]['url'], self.GetNTPThumbnails()[0]['url'])
+
+  def testPinnedThumbnailNeverMoves(self):
+    """Testis that once a thumnail is pinned it never moves"""

[Nirnimesh, 2010/11/24 09:58:23]

Testis?!!

[kkania, 2010/11/24 17:40:35]

Done.

+    self.RemoveNTPDefaultThumbnails()
+    for page in self.PAGES:
+      self.AppendTab(pyauto.GURL(page['url']))
+    self.PinNTPThumbnail(self.GetNTPThumbnails()[0])
+    thumbnails = self.GetNTPThumbnails()
+    self.AppendTab(pyauto.GURL(self.PAGES[1]['url']))
+    self.assertEqual(thumbnails, self.GetNTPThumbnails())

[Nirnimesh, 2010/11/24 09:58:23]

won't self.PAGES[1]['url']) show up in GetNTPThumbnails()?

[kkania, 2010/11/24 17:40:35]

yes, but all the pages have already been added to the NTP. Going to page[1]
again triggers it to have a higher rank than page[0]. However, since page[0] is
pinned, it should stay put.

+
+  def testThumbnailTitleChangeAfterPageTitleChange(self):
+    """Tests that once a page title changes, the thumbnail title changes also"""

[Nirnimesh, 2010/11/24 09:58:23]

s/also/too/

[kkania, 2010/11/24 17:40:35]

Done.

+    self.RemoveNTPDefaultThumbnails()
+    self.NavigateToURL(self.PAGES[0]['url'])
+    self.assertEqual(self.PAGES[0]['title'],
+                     self.GetNTPThumbnails()[0]['title'])
+    self.ExecuteJavascript('window.domAutomationController.send(' +
+                           'document.title = "new title")')
+    self.assertEqual('new title', self.GetNTPThumbnails()[0]['title'])
+
+  def testCloseOneTab(self):
+    """Tests that closing a tab populates the recently closed list"""
+    self.RemoveNTPDefaultThumbnails()
+    self.AppendTab(pyauto.GURL(self.PAGES[1]['url']))
+    self.GetBrowserWindow(0).GetTab(1).Close(True)
+    self.assertEqual(self.PAGES[1]['url'],
+                     self.GetNTPRecentlyClosed()[0]['url'])
+    self.assertEqual(self.PAGES[1]['title'],
+                     self.GetNTPRecentlyClosed()[0]['title'])
+
+  def testCloseOneWindow(self):
+    """Tests that closing a window populates the recently closed list"""
+    self.RemoveNTPDefaultThumbnails()
+    self.OpenNewBrowserWindow(True)
+    self.NavigateToURL(self.PAGES[0]['url'], 1, 0)
+    self.AppendTab(pyauto.GURL(self.PAGES[1]['url']), 1)
+    self.CloseBrowserWindow(1)
+    expected = [{ u'type': u'window',
+                  u'tabs': [
+                  { u'type': u'tab',
+                    u'url': self.PAGES[0]['url'],
+                    u'direction': u'ltr' },
+                  { u'type': u'tab',
+                    u'url': self.PAGES[1]['url']}]
+                }]
+    self.assertEquals(expected, test_utils.StripUnmatchedKeys(
+        self.GetNTPRecentlyClosed(), expected))
+
+  def testCloseMultipleTabs(self):
+    """Tests closing multiple tabs"""

[Nirnimesh, 2010/11/24 09:58:23]

Please elaborate. Should mention that it checks the recently closed list

[kkania, 2010/11/24 17:40:35]

Done.

+    self.RemoveNTPDefaultThumbnails()
+    self.AppendTab(pyauto.GURL(self.PAGES[0]['url']))
+    self.AppendTab(pyauto.GURL(self.PAGES[1]['url']))
+    self.GetBrowserWindow(0).GetTab(2).Close(True)
+    self.GetBrowserWindow(0).GetTab(1).Close(True)
+    expected = [{ u'type': u'tab',
+                  u'url': self.PAGES[0]['url']
+                },
+                { u'type': u'tab',
+                  u'url': self.PAGES[1]['url']
+                }]
+    self.assertEquals(expected, test_utils.StripUnmatchedKeys(
+        self.GetNTPRecentlyClosed(), expected))
+
+  def testCloseWindowWithOneTab(self):
+    """Tests that closing a window with only one tab only shows up as a tab in
+    the Recently Closed section"""
+    self.RemoveNTPDefaultThumbnails()
+    self.OpenNewBrowserWindow(True)
+    self.NavigateToURL(self.PAGES[0]['url'], 1, 0)
+    self.CloseBrowserWindow(1)
+    expected = [{ u'type': u'tab',
+                  u'url': self.PAGES[0]['url']
+                }]
+    self.assertEquals(expected, test_utils.StripUnmatchedKeys(
+        self.GetNTPRecentlyClosed(), expected))
+
+  def testCloseMultipleWindows(self):
+    """Tests closing multiple windows populates the Recently Closed list"""
+    self.RemoveNTPDefaultThumbnails()
+    self.OpenNewBrowserWindow(True)
+    self.NavigateToURL(self.PAGES[0]['url'], 1, 0)
+    self.AppendTab(pyauto.GURL(self.PAGES[1]['url']), 1)
+    self.OpenNewBrowserWindow(True)
+    self.NavigateToURL(self.PAGES[1]['url'], 2, 0)
+    self.AppendTab(pyauto.GURL(self.PAGES[0]['url']), 2)
+    self.CloseBrowserWindow(2)
+    self.CloseBrowserWindow(1)
+    expected = [{ u'type': u'window',
+                  u'tabs': [
+                  { u'type': u'tab',
+                    u'url': self.PAGES[0]['url'],
+                    u'direction': u'ltr' },
+                  { u'type': u'tab',
+                    u'url': self.PAGES[1]['url']}]
+                },
+                { u'type': u'window',
+                  u'tabs': [
+                  { u'type': u'tab',
+                    u'url': self.PAGES[1]['url'],
+                    u'direction': u'ltr' },
+                  { u'type': u'tab',
+                    u'url': self.PAGES[0]['url']}]
+                }]
+    self.assertEquals(expected, test_utils.StripUnmatchedKeys(
+        self.GetNTPRecentlyClosed(), expected))
+
+  def testRecentlyClosedShowsUniqueItems(self):
+    """Tests that the Recently Closed section does not show duplicate items"""
+    self.RemoveNTPDefaultThumbnails()
+    self.AppendTab(pyauto.GURL(self.PAGES[0]['url']))
+    self.AppendTab(pyauto.GURL(self.PAGES[0]['url']))
+    self.GetBrowserWindow(0).GetTab(1).Close(True)
+    self.GetBrowserWindow(0).GetTab(1).Close(True)
+    self.assertEquals(1, len(self.GetNTPRecentlyClosed()))
+
+  def testRecentlyClosedIncognito(self):
+    """Tests that we don't record closure of Incognito tabs or windows"""
+    self.RemoveNTPDefaultThumbnails()
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
+    self.NavigateToURL(self.PAGES[0]['url'], 1, 0)
+    self.AppendTab(pyauto.GURL(self.PAGES[0]['url']), 1)
+    self.AppendTab(pyauto.GURL(self.PAGES[1]['url']), 1)
+    self.GetBrowserWindow(1).GetTab(0).Close(True)
+    self.assertEqual([], self.GetNTPRecentlyClosed())

[Nirnimesh, 2010/11/24 09:58:23]

assertFalse

[kkania, 2010/11/24 17:40:35]

Done.

+    self.CloseBrowserWindow(1)
+    self.assertEqual([], self.GetNTPRecentlyClosed())

[Nirnimesh, 2010/11/24 09:58:23]

assertFalse

[kkania, 2010/11/24 17:40:35]

Done.

+
 
 if __name__ == '__main__':
   pyauto_functional.Main()