Add pyauto tests for Web SQL databases.

chrome/test/functional/databases.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 simplejson
 import os
 
 import pyauto_functional
 import pyauto
@@ skipping 106 matching lines @@
       array of all the records in the database
 
     Args:
       tab_index: index of the tab that will query the database
       windex: index of the window containing the tab that will query the
               database
     """
     json = self.CallJavascriptFunc('getRecords', [], tab_index, windex)
     return self._ParseAndCheckResultFromTestPage(json)['returnValue']
 
+  def _HasTable(self, tab_index=0, windex=0):
+    """Returns whether the page has a table in its database."""
+    try:
+      self._GetRecords(tab_index, windex)
+    except SQLExecutionError:
+      return False
+    return True
+
   def testInsertRecord(self):
     """Insert records to the database."""
     self.NavigateToURL(self.TEST_PAGE_URL)
     self._CreateTable()
     self._InsertRecord('text')
     self.assertEquals(['text'], self._GetRecords())
     self._InsertRecord('text2')
     self.assertEquals(['text', 'text2'], self._GetRecords())
 
   def testUpdateRecord(self):
@@ skipping 32 matching lines @@
     self.assertEquals(['0', '2'], self._GetRecords())
 
   def testDeleteNonexistentRow(self):
     """Attempts to delete a nonexistent row in the table."""
     self.NavigateToURL(self.TEST_PAGE_URL)
     self._CreateTable()
     self._InsertRecord('text')
     did_throw_exception = False
     try:
       self._DeleteRecord(1)
-    except:
+    except SQLExecutionError:
       did_throw_exception = True
     self.assertTrue(did_throw_exception)
     self.assertEquals(['text'], self._GetRecords())
 
   def testDatabaseOperations(self):
     """Insert, update, and delete records in the database."""
     self.NavigateToURL(self.TEST_PAGE_URL)
     self._CreateTable()
 
     for i in range(10):
@@ skipping 20 matching lines @@
 
   def testIncognitoCannotReadRegularDatabase(self):
     """Attempt to read a database created in a regular browser from an incognito
     browser.
     """
     self.NavigateToURL(self.TEST_PAGE_URL)
     self._CreateTable()
     self._InsertRecord('text')
     self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
     self.NavigateToURL(self.TEST_PAGE_URL, 1, 0)
-    can_read_regular_database = False
-    try:
-      # |_GetRecords| should throw an error because the table does not exist.
-      if len(self._GetRecords(windex=1)) == 1:
-        can_read_regular_database = True
-    except SQLExecutionError:
-      pass
-    self.assertFalse(can_read_regular_database)
+    self.assertFalse(self._HasTable(windex=1))
     self._CreateTable(windex=1)
-    self.assertEqual(0, len(self._GetRecords(windex=1)))
+    self.assertFalse(self._GetRecords(windex=1))
 
   def testRegularCannotReadIncognitoDatabase(self):
     """Attempt to read a database created in an incognito browser from a regular
     browser.
     """
     self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
     self.NavigateToURL(self.TEST_PAGE_URL, 1, 0)
     self._CreateTable(windex=1)
     self._InsertRecord('text', windex=1)
 
     # Verify a regular browser cannot read the incognito database.
     self.NavigateToURL(self.TEST_PAGE_URL)
-    can_read_incognito_database = False
-    try:
-      # |_GetRecords| should throw an error because the table does not exist.
-      if len(self._GetRecords()) == 1:
-        can_read_incognito_database = True
-    except SQLExecutionError:
-      pass
-    self.assertFalse(can_read_incognito_database)
+    self.assertFalse(self._HasTable())
+    self._CreateTable()
+    self.assertFalse(self._GetRecords())
 
   def testDbModificationPersistInSecondTab(self):
     """Verify DB changes within first tab are present in the second tab."""
     self.NavigateToURL(self.TEST_PAGE_URL)
     self._CreateTable()
     self._InsertRecord('text')
     self.AppendTab(pyauto.GURL(self.TEST_PAGE_URL))
     self._UpdateRecord(0, '0', tab_index=0)
     tab1_records = self._GetRecords(tab_index=0)
     tab2_records = self._GetRecords(tab_index=1)
     self.assertEquals(1, len(tab1_records))
     self.assertEquals('0', tab1_records[0])
     self.assertEquals(1, len(tab2_records))
     self.assertEquals(tab1_records[0], tab2_records[0])
 
+  def testIncognitoDatabaseNotPersistent(self):
+    """Verify incognito database is removed after incognito window closes."""
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
+    self.NavigateToURL(self.TEST_PAGE_URL, 1, 0)
+    self._CreateTable(windex=1)
+    self._InsertRecord('text', windex=1)
+    self.CloseBrowserWindow(1)
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
+    self.NavigateToURL(self.TEST_PAGE_URL, 1, 0)
+    self.assertFalse(self._HasTable(windex=1))
+
+  def testModificationsPersistAfterRendererCrash(self):
+    """Verify database modifications persist after crashing tab."""
+    self.NavigateToURL(self.TEST_PAGE_URL)
+    self._CreateTable()
+    self._InsertRecord('1')
+    self.NavigateToURL('about:crash')

[Nirnimesh, 2010/11/24 23:07:57]

This will create a .dmp file, and there're checks in ui_test.cc. Make sure you
clean up the .dmp file you created.

Or else you could just kill the renderer PID.

+    self.NavigateToURL(self.TEST_PAGE_URL)
+    self.assertEqual(['1'], self._GetRecords())
+
+  def testIncognitoDBPersistentAcrossTabs(self):
+    """Test to check if database modifications are persistent across tabs
+    in incognito window.
+    """
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
+    self.NavigateToURL(self.TEST_PAGE_URL, 1, 0)
+    self._CreateTable(windex=1)
+    self._InsertRecord('text', windex=1)
+    self.AppendTab(pyauto.GURL(self.TEST_PAGE_URL), 1)
+    self.assertEquals(['text'], self._GetRecords(1, 1))
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()