File upload API in chromedriver

chrome/test/webdriver/chromedriver_tests.py

 #!/usr/bin/python
 
 # Copyright (c) 2011 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.
 
 """Tests for ChromeDriver.
 
 If your test is testing a specific part of the WebDriver API, consider adding
 it to the appropriate place in the WebDriver tree instead.
 """
 
 import hashlib
 import os
 import platform
 import sys
+import tempfile
 import unittest
 import urllib
 import urllib2
 import urlparse
 
 from chromedriver_launcher import ChromeDriverLauncher
 import chromedriver_paths
 from gtest_text_test_runner import GTestTextTestRunner
 
 sys.path += [chromedriver_paths.SRC_THIRD_PARTY]
@@ skipping 256 matching lines @@
 
   def tearDown(self):
     self._driver.quit()
     self._launcher.Kill()
 
   def testAddCookie(self):
     self._driver.get(self._launcher.GetURL() + '/test_page.html')
     cookie_dict = None
     cookie_dict = self._driver.get_cookie("chromedriver_cookie_test")
     cookie_dict = {}
-    cookie_dict["name"]= "chromedriver_cookie_test"
+    cookie_dict["name"] = "chromedriver_cookie_test"
     cookie_dict["value"] = "this is a test"
     self._driver.add_cookie(cookie_dict)
     cookie_dict = self._driver.get_cookie("chromedriver_cookie_test")
     self.assertNotEqual(cookie_dict, None)
     self.assertEqual(cookie_dict["value"], "this is a test")
 
   def testDeleteCookie(self):
     self.testAddCookie();
     self._driver.delete_cookie("chromedriver_cookie_test")
     cookie_dict = self._driver.get_cookie("chromedriver_cookie_test")
@@ skipping 285 matching lines @@
         'autofill-edit-credit-card-apply-button').click()
     # Refresh the page to ensure the UI is up-to-date.
     driver.refresh()
     list_entry = driver.find_element_by_class_name('autofill-list-item')
     self.assertTrue(list_entry.is_displayed)
     self.assertEqual(list_entry.text,
                      creditcard_data['CREDIT_CARD_NAME'],
                      'Saved CC line item not same as what was entered.')
 
 
+class FileUploadControlTest(unittest.TestCase):
+  """Tests dealing with file upload control."""
+
+  def setUp(self):
+    self._launcher = ChromeDriverLauncher(root_path=os.path.dirname(__file__))
+    self._driver = WebDriver(self._launcher.GetURL(),
+                             DesiredCapabilities.CHROME)
+
+  def tearDown(self):
+    self._driver.quit()
+    self._launcher.Kill()
+
+  def testSetFilePathToFileUploadControl(self):
+    """Verify a file path is set to the file upload control."""
+    self._driver.get(self._launcher.GetURL() + '/upload.html')
+
+    file = tempfile.NamedTemporaryFile()
+
+    fileupload = self._driver.find_element_by_name('fileupload')
+    fileupload.send_keys(file.name)
+    path = fileupload.value
+    self.assertTrue(path.endswith(os.path.basename(file.name)))

[kkania, 2011/06/07 15:12:04]

do you need to close/delete the temp file?

[nodchip, 2011/06/08 02:37:19]

I think we don't need to manually close/delete the temporary files. The files
will be automatically closed at the end of the script process and deleted at
that time.

10.6. tempfile — Generate temporary files and directories — Python v2.7.1
documentation http://docs.python.org/library/tempfile.html


On 2011/06/07 15:12:04, kkania wrote:

+
+  def testSetMultipleFilePathsToFileUploadControl(self):
+    """Verify multiple file paths are set to the file upload control."""
+    self._driver.get(self._launcher.GetURL() + '/upload.html')
+
+    files = list()
+    filepaths = list()
+    filenames = set()
+    for index in xrange(4):
+      file = tempfile.NamedTemporaryFile()
+      files.append(file)
+      filepath = file.name
+      filepaths.append(filepath)
+      filenames.add(os.path.basename(filepath))
+
+    fileupload = self._driver.find_element_by_name('fileupload')
+    fileupload.send_keys(filepaths[0], filepaths[1], filepaths[2], filepaths[3])
+    path = fileupload.value.replace('\\', '//')
+    self.assertTrue(os.path.basename(path) in filenames)
+
+
 if __name__ == '__main__':
   unittest.main(module='chromedriver_tests',
                 testRunner=GTestTextTestRunner(verbosity=1))