| Index: chrome/test/functional/rlz/rlztest.py
|
| ===================================================================
|
| --- chrome/test/functional/rlz/rlztest.py (revision 261231)
|
| +++ chrome/test/functional/rlz/rlztest.py (working copy)
|
| @@ -1,270 +0,0 @@
|
| -#!/usr/bin/env 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.
|
| -
|
| -import chromium_proxy_server_ex
|
| -import commands
|
| -import logging
|
| -import optparse
|
| -import os
|
| -import re
|
| -import selenium.selenium
|
| -import shutil
|
| -import subprocess
|
| -import sys
|
| -import tempfile
|
| -import time
|
| -import unittest
|
| -import _winreg
|
| -import selenium.webdriver.common.keys
|
| -
|
| -from selenium import webdriver
|
| -from selenium.webdriver.common.keys import Keys as Keys
|
| -from selenium.webdriver.support.ui import WebDriverWait
|
| -
|
| -class RlzTest(unittest.TestCase):
|
| -
|
| - proxy_server_file = ''
|
| - chrome_driver_path = ''
|
| -
|
| - def setUp(self):
|
| - """Performs necessary setup work before running each test in this class."""
|
| - # Delete RLZ key Folder
|
| - self.deleteRegistryKey()
|
| - # Launch Proxy Server
|
| - print ('Serving clients: 127.0.0.1')
|
| - self.proxy_server = subprocess.Popen([
|
| - 'python',
|
| - RlzTest.proxy_server_file,
|
| - '--client=127.0.0.1',
|
| - '--port=8080',
|
| - '--urls=http://clients1.google.com/tools/pso/ping,www.google.com'])
|
| - print('\nLaunching Chrome...')
|
| - # Launch chrome and set proxy.
|
| - self.temp_data_dir = tempfile.mkdtemp()
|
| - self.launchChrome(self.temp_data_dir)
|
| -
|
| - def tearDown(self):
|
| - """Kills the chrome driver after after the test method has been called."""
|
| - # Terminate the chrome driver.
|
| - print '\nTerminating Chrome Driver...'
|
| - self.driver.quit()
|
| -
|
| - # Kill proxy server.
|
| - print '\nKilling Proxy Server...'
|
| - subprocess.Popen.kill(self.proxy_server)
|
| -
|
| - # Delete temp profile directory
|
| - try:
|
| - shutil.rmtree(self.temp_data_dir) # delete directory
|
| - except OSError, e:
|
| - if e.errno != 2: # code 2 - no such file or directory
|
| - raise
|
| -
|
| - def launchChrome(self, data_directory):
|
| - """Launch chrome using chrome driver.
|
| -
|
| - Args:
|
| - data_directory: Temp directory to store preference data.
|
| - """
|
| - service = webdriver.chrome.service.Service(RlzTest.chrome_driver_path)
|
| - service.start()
|
| - self.driver = webdriver.Remote(
|
| - service.service_url, {
|
| - 'proxy': {'proxyType': 'manual', 'httpProxy': 'localhost:8080'},
|
| - 'chrome.nativeEvents': True,
|
| - 'chrome.switches': ['disable-extensions',
|
| - r'user-data-dir=' + data_directory]})
|
| -
|
| - def deleteRegistryKey(self):
|
| - """Delete RLZ key Folder from win registry."""
|
| - try:
|
| - hkey = _winreg.OpenKey(_winreg.HKEY_CURRENT_USER,
|
| - 'Software\\Google\\Common\\Rlz')
|
| - except _winreg.error, err:
|
| - return True
|
| - if(hkey):
|
| - _winreg.DeleteKey(_winreg.HKEY_CURRENT_USER,
|
| - 'Software\\Google\\Common\\Rlz\\Events\\C')
|
| - _winreg.DeleteKey(_winreg.HKEY_CURRENT_USER,
|
| - 'Software\\Google\\Common\\Rlz\\StatefulEvents\\C')
|
| - _winreg.DeleteKey(_winreg.HKEY_CURRENT_USER,
|
| - 'Software\\Google\\Common\\Rlz\\Events')
|
| - _winreg.DeleteKey(_winreg.HKEY_CURRENT_USER,
|
| - 'Software\\Google\\Common\\Rlz\\StatefulEvents')
|
| - _winreg.DeleteKey(_winreg.HKEY_CURRENT_USER,
|
| - 'Software\\Google\\Common\\Rlz\\PTimes')
|
| - _winreg.DeleteKey(_winreg.HKEY_CURRENT_USER,
|
| - 'Software\\Google\\Common\\Rlz\\RLZs')
|
| - _winreg.DeleteKey(_winreg.HKEY_CURRENT_USER,
|
| - 'Software\\Google\\Common\\Rlz')
|
| -
|
| - def GetKeyValueNames(self, key, subkey):
|
| - """Get the values for particular subkey
|
| -
|
| - Args:
|
| - key: Key is one of the predefined HKEY_* constants.
|
| - subkey: It is a string that identifies the sub_key to delete.
|
| - """
|
| - list_names = []
|
| - counter = 0
|
| - try:
|
| - hkey = _winreg.OpenKey(key, subkey)
|
| - except _winreg.error, err:
|
| - return -1
|
| - while True:
|
| - try:
|
| - value = _winreg.EnumValue(hkey, counter)
|
| - list_names.append(value[0])
|
| - counter += 1
|
| - except _winreg.error:
|
| - break
|
| - hkey.Close()
|
| - return list_names
|
| -
|
| - def _AssertEventsInPing(self, log_file, excepted_event_list, readFile=1):
|
| - """ Asserts events in ping appended.
|
| -
|
| - Args:
|
| - contents: String variable contains contents of log file.
|
| - excepted_event_list: List of expected events in ping.
|
| - readFile: Reading order for file. Default is 1 (Top to Bottom).
|
| - """
|
| - for line in log_file[::readFile]:
|
| - if(re.search('events=', line)):
|
| - event_start = line.find('events=')
|
| - event_end = line.find('&rep', event_start)
|
| - events = line[event_start + 7 : event_end]
|
| - events_List = events.split(',')
|
| - print 'event_list',events_List
|
| - break
|
| - # Validate events in url ping.
|
| - for event in excepted_event_list:
|
| - self.assertTrue(event in events_List)
|
| - # Validate brand code in url ping.
|
| - self.assertTrue(re.search('CHMZ', line))
|
| - # Print first chrome launch ping on Console.
|
| - start = line.find('http://clients1.google.com/tools/'+
|
| - 'pso/ping?as=chrome&brand=CHMZ')
|
| - end = line.find('http://www', start)
|
| - print '\nChrome Launch ping sent :\n', line[start:end]
|
| -
|
| -
|
| - def _AssertEventsInRegistry(self, excepted_reg_keys):
|
| - """ Asserts events reported in win registry.
|
| -
|
| - Args:
|
| - excepted_reg_keys: List of expected events in win registry.
|
| - """
|
| - list_key=self.GetKeyValueNames(_winreg.HKEY_CURRENT_USER,
|
| - 'Software\Google\Common\Rlz\StatefulEvents\C')
|
| - print ('\nList of event reported to registry-'
|
| - 'Software\Google\Common\Rlz\StatefulEvents:', list_key)
|
| - for key in excepted_reg_keys:
|
| - self.assertTrue(key in list_key)
|
| -
|
| - def _AssertRlzValues(self, log_file, readFile=1):
|
| - """ Asserts RLZ values.
|
| -
|
| - Args:
|
| - log_file: String variable contains contents of log file.
|
| - readFile: Reading order for file. Default is 1 (Top to Bottom).
|
| - """
|
| - for line in log_file[::readFile]:
|
| - if(re.search('events=', line)):
|
| - event_start = line.find('rlz=')
|
| - event_end = line.find('&id', event_start)
|
| - events = line[event_start + 4 : event_end]
|
| - events_List = events.split(',')
|
| - self.assertTrue('C1:' in events_List)
|
| - self.assertTrue('C2:' in events_List)
|
| -
|
| - def _searchFromOmnibox(self, searchString):
|
| - """ Asserts RLZ values.
|
| -
|
| - Args:
|
| - searchString: Input string to be searched.
|
| - """
|
| - self.driver.switch_to_active_element().send_keys(Keys.CONTROL + 'l')
|
| - self.driver.switch_to_active_element().send_keys(searchString)
|
| - self.driver.switch_to_active_element().send_keys(Keys.ENTER)
|
| - time.sleep(2)
|
| -
|
| - def testRlzPingAtFirstChromeLaunch(self):
|
| - """Test rlz ping when chrome is launched for first time."""
|
| - # Wait for 100 sec till chrome sends ping to server.
|
| - time.sleep(100)
|
| - self.driver.get('http://www.google.com')
|
| -
|
| - # Open log file.
|
| - log_file = open(
|
| - os.getcwd() + '\chromium_proxy_server.log', 'r', 1).readlines()
|
| -
|
| - # Validate events first chrome launch(C1I,C2I,C1S) are appended in ping.
|
| - excepted_events = ['C1S', 'C1I', 'C2I']
|
| - self._AssertEventsInPing(log_file, excepted_events)
|
| -
|
| - # Validate events in win registry.
|
| - excepted_reg_keys = ['C1I', 'C2I']
|
| - self._AssertEventsInRegistry(excepted_reg_keys)
|
| -
|
| - def testRlzPingForFirstSearch(self):
|
| - """Test rlz ping when first search is performed in chrome."""
|
| - # Type search string in omnibox.
|
| - self._searchFromOmnibox('java')
|
| - print '\nCurrent Url before chrome ping sent:\n', self.driver.current_url
|
| -
|
| - # Assert brand code 'CHMZ' is not appended in search string.
|
| - self.assertFalse(re.search('CHMZ', self.driver.current_url))
|
| -
|
| - # Wait for 100 sec till chrome sends ping to server.
|
| - time.sleep(100)
|
| -
|
| - # Open log file.
|
| - log_file = open(
|
| - os.getcwd() + '\chromium_proxy_server.log', 'r', 1).readlines()
|
| -
|
| - # Validate events first chrome launch(C1I,C2I,C1S) and
|
| - # first search(C1F) are appended in ping.
|
| - excepted_events = ['C1S', 'C1I', 'C2I', 'C1F']
|
| - self._AssertEventsInPing(log_file, excepted_events)
|
| -
|
| - # Assert C1, C2 rlz value appended in ping
|
| - self._AssertRlzValues(log_file)
|
| -
|
| - # Type search string in omnibox after ping is sent to server.
|
| - self._searchFromOmnibox('java')
|
| - print '\nCurrent Url after chrome ping sent:\n', self.driver.current_url
|
| -
|
| - # Assert brand code 'CHMZ' is appended in search string.
|
| - self.assertTrue(re.search('CHMZ', self.driver.current_url))
|
| -
|
| - # Validate events in win registry.
|
| - excepted_reg_keys=['C1I', 'C2I', 'C1F']
|
| - self._AssertEventsInRegistry(excepted_reg_keys)
|
| -
|
| - # Assert the log for search ping with query string/brand code appended.
|
| - log_file = open(
|
| - os.getcwd() + '\chromium_proxy_server.log', 'r', 1).readlines()
|
| - searchStringFound = False
|
| - for line in log_file[::-1]:
|
| - if(re.search('search?', line)):
|
| - self.assertTrue(re.search('java', line))
|
| - self.assertTrue(re.search('CHMZ', line))
|
| - print '\nChrome search ping send\n', line
|
| - searchStringFound = True
|
| - break
|
| - self.assertTrue(searchStringFound, 'Search Query String Not Found')
|
| -
|
| -def rlzInput():
|
| - proxy_server_file = raw_input("Enter Proxy Server File Name: ")
|
| - chrome_driver_path = raw_input("Enter chrome driver path in"+
|
| - "your system(c:\\chrome\\..):")
|
| - return (proxy_server_file, chrome_driver_path)
|
| -
|
| -if __name__ == '__main__':
|
| - server, driver = rlzInput()
|
| - RlzTest.proxy_server_file = server
|
| - RlzTest.chrome_driver_path = driver
|
| - unittest.main()
|
|
|