Added rlz testcases.

rlz_test/rlztest.py

Index: rlz_test/rlztest.py
===================================================================
--- rlz_test/rlztest.py	(revision 0)
+++ rlz_test/rlztest.py	(revision 0)
@@ -0,0 +1,200 @@
+import subprocess
+import sys
+import selenium.selenium
+import re
+import unittest
+import commands
+import time
+import _winreg
+import logging
+import optparse
+import logging
+import selenium.webdriver.common.keys
+import os

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

please put all these imports in alpha order.
please use standard chromium source file header.

[prachij, 2012/09/19 19:30:16]

Done.

+
+from selenium import webdriver
+#from rlz_tester import RlzTester
+from chromium_proxy_server_version2 import Main
+from chromium_proxy_server_version3 import ChromiumProxyServer

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

you can remove the imports at lines 16-18

[prachij, 2012/09/19 19:30:16]

Done.

+from subprocess import Popen, PIPE, STDOUT

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

remove line 19, not needed since you do it at line 1

[prachij, 2012/09/19 19:30:16]

Done.

+from selenium.webdriver.common.keys import Keys as Keys

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

why do you need the "as Keys" part?

[prachij, 2012/09/19 19:30:16]

I am using native events like ctrl, enter in line 167 where I am adding this to
Keys. If I don't use "as Keys" I end up getting some exception though not sure
why.
On 2012/09/14 19:43:36, Roger Tawa wrote:

[Roger Tawa OOO till Jul 10th, 2012/09/20 19:13:19]

Can you tell me what exceptions you are getting?

+
+
+class RlzTest(unittest.TestCase):
+
+  def setUp(self):
+    """Performs necessary setup work before running each test in this class."""
+    # Delete RLZ key Folder
+    self.deleteRegistryKey()
+    proxy_server = raw_input("Enter Proxy Server File Name: ")
+    chrome_driver = raw_input("Enter chrome driver path in"+
+                              "your system(c:\\chrome\\..):")

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

Is this script meant to be use interactively?

This will get repetitive I think since setUp() is called before each testXXX
function.  I think it would be better to do as command line args, or at least
ask only once before calling unittest.main()

[prachij, 2012/09/19 19:30:16]

My intent was to ask only once. I was planning to move this in
'setupbeforeclass()' method, but its added in python2.7. Most of people use
python2.6 so I didn't went forward with it. Another alternate was to use command
line arg, but since this script might be used my manual tester in future I
wanted to make sure they add correct data before running. So now I have created
a separate method to handle this, which I am going to call before main.So, now
this will only be called once.
On 2012/09/14 19:43:36, Roger Tawa wrote:

[Roger Tawa OOO till Jul 10th, 2012/09/20 19:13:19]

Looks good.  Better to stick with 2.6 since that is what is installed by default
everywhere.

+    # Launch Proxy Server
+    print ('Serving clients: 127.0.0.1')
+    self.proxy_server = subprocess.Popen([
+          'python',
+          proxy_server,
+          '--client=127.0.0.1',
+          '--port=8080',
+          '--urls=http://clients1.google.com/tools/pso/ping,www.google.com'])

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

lines 35-39 should be indented only 4 spaces from line 34.

+
+    print('\nLaunching Chrome...')
+    # Launch chrome and set proxy.
+    service = webdriver.chrome.service.Service(chrome_driver)
+    service.start()
+    self.driver = webdriver.Remote(
+        service.service_url, {
+        'proxy': {'proxyType': 'manual', 'httpProxy': 'localhost:8080'},
+        'chrome.switches': ['disable-extensions'],
+        'chrome.nativeEvents': True})
+
+  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)
+
+  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"""

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

plese document args and return.

[prachij, 2012/09/19 19:30:16]

Done.

+    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 testRlzPingAtFirstChromeLaunch(self):
+    """Test rlz ping when chrome is lauched for first time."""
+    # Check the ping in Events folder in registry.
+    list_key=[]
+    list_key=self.GetKeyValueNames(_winreg.HKEY_CURRENT_USER,
+                              'Software\Google\Common\Rlz\Events\C')

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

spaces around = in two lines above.  Please also fix indentation.

don't you need to double up the backslash \?  Or use as raw string?  this
applies to all string with backslahes, like line 108 below,

[prachij, 2012/09/19 19:30:16]

This was working fine with single \, but just to be on safe side I have changed
it to double.
On 2012/09/14 19:43:36, Roger Tawa wrote:

[Roger Tawa OOO till Jul 10th, 2012/09/20 19:13:19]

Single \ works when the following letter happens to not be a special escape
sequence.  So better to double up.  Thanks.

+    print ('\nList of event reported to registry-'
+          'Software\Google\Common\Rlz\Events :\n', list_key)

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

line up the two starting single quotes.

[prachij, 2012/09/19 19:30:16]

Done.

+    #self.assertEqual('C1S', list_key[0])
+    #self.assertEqual('C2I', list_key[1])

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

should these be commented out?

[prachij, 2012/09/19 19:30:16]

This should not be commented out, but for some reason these assertions are
failing now, they were passing before. Till I  investigate what the problem is I
have temporary commented out.
On 2012/09/14 19:43:36, Roger Tawa wrote:

+
+    # Wait for 90 sec till chrome sends ping to server.
+    time.sleep(90)
+    self.driver.get('http://www.google.com')
+
+    # Open log file and validate events chrome launch(C1I,C2I,C1S)
+    # are appended in ping.
+    contents = open(os.getcwd()+'\ServerXXX.log', 'r', 1).read()
+    event_start = contents.find('events=')
+    event_end = contents.find('&rep', event_start)
+    events=contents[event_start+7 : event_end]
+    events_List=events.split(',')
+    self.assertTrue('C1S' in events_List)
+    self.assertTrue('C1I' in events_List)
+    self.assertTrue('C2I' in events_List)
+
+    # Validate brand code in url ping.
+    self.assertTrue(re.search('CHMZ', contents))
+
+    # Print first chrome launch ping on Console.
+    start = contents.find('http://clients1.google.com/tools/'+
+                          'pso/ping?as=chrome&brand=CHMZ')
+    end = contents.find('http://www', start)
+    print '\nChrome Launch ping send :\n', contents[start:end]

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

send -> sent ?

[prachij, 2012/09/19 19:30:16]

Done.

+
+    # Validate events in win registry.
+    list_key=[]
+    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)

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

line up single quotes.

[prachij, 2012/09/19 19:30:16]

Done.

+    self.assertEqual('C1I', list_key[0])
+    self.assertEqual('C2I', list_key[1])
+
+  def testRlzPingForFirstSearch(self):
+    """Test rlz ping when first search is performed in chrome."""
+    # Type search string in omnibox.
+    self.driver.switch_to_active_element().send_keys(Keys.CONTROL + 'l')
+    self.driver.switch_to_active_element().send_keys('java')
+    self.driver.switch_to_active_element().send_keys(Keys.ENTER)
+    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 90 sec till chrome sends ping to server.
+    time.sleep(100)

[Roger Tawa OOO till Jul 10th, 2012/09/14 19:43:36]

comment says 90

[prachij, 2012/09/19 19:30:16]

Done.

+
+    # Open log file and validate events chrome launch(C1I,C2I,C1S)
+    # and first search(C1F) are appended in ping.
+    contents = open(os.getcwd()+'\ServerXXX.log', 'r', 1).read()
+    event_start = contents.find('events=')
+    event_end = contents.find('&rep', event_start)
+    events=contents[event_start+7 : event_end]
+    events_List=events.split(',')
+    self.assertTrue('C1S' in events_List)
+    self.assertTrue('C1I' in events_List)
+    self.assertTrue('C2I' in events_List)
+    self.assertTrue('C1F' in events_List)
+
+    # Print first chrome launch /default first search pso ping on Console.
+    start = contents.find('http://clients1.google.com/tools/pso/'+
+                          'ping?as=chrome&brand=CHMZ')
+    end = contents.find('http://www', start)
+    print '\nChrome Launch ping send\n', contents[start:end]
+
+    # Type search string in omnibox after ping is sent to server.
+    self.driver.switch_to_active_element().send_keys(Keys.CONTROL + 'l')
+    self.driver.switch_to_active_element().send_keys('java')
+    self.driver.switch_to_active_element().send_keys(Keys.ENTER)
+    time.sleep(2)
+    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.
+    list_key=[]
+    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:\n', list_key)
+    self.assertTrue('C1I' in list_key)
+    self.assertTrue('C2I' in list_key)
+    self.assertTrue('C1F' in list_key)
+
+  
+if __name__ == '__main__':
+  unittest.main()
+