putting tests inside "unittest" classes

selenium/wallet-test.py

@@ -1,31 +1,36 @@
 from selenium import webdriver
 import time
+import unittest
 
-co = webdriver.ChromeOptions()
-co.add_argument("load-extension=/home/marcello/Taler/wallet-webex")
-cap = webdriver.DesiredCapabilities.CHROME.copy()
-cap['loggingPrefs'] = {'driver': 'INFO', 'browser': 'INFO'}
-client = webdriver.Chrome(chrome_options=co, desired_capabilities=cap)
-client.get('https://taler.net')
-listener = """\
+
+def client_setup():
+    """Return a Chrome browser the extension's id"""
+    co = webdriver.ChromeOptions()
+    co.add_argument("load-extension=/home/marcello/wallet-webex")
+    cap = webdriver.DesiredCapabilities.CHROME.copy()
+    cap['loggingPrefs'] = {'driver': 'INFO', 'browser': 'INFO'}
+    client = webdriver.Chrome(chrome_options=co, desired_capabilities=cap)
+    client.get('https://taler.net')
+    listener = """\
         document.addEventListener('taler-id', function(evt){
           window['extId'] = evt.detail.id;    
         }); 
         var evt = new CustomEvent('taler-query-id');
         document.dispatchEvent(evt);
         """
-client.execute_script(listener)
-poll = """\
+    client.execute_script(listener)
+    poll = """\
         if(window.extId)
           return window.extId;
         else return false;
         """
-# TODO intelligent poller needed
-time.sleep(1)
-ext_id = client.execute_script(poll)
+    # Todo: put some delay in polling
+    ext_id = client.execute_script(poll)
+    return {'client': client, 'ext_id': ext_id}
+
 
-# if client has error from its activity, ptints it and returns True
 def is_error(client):
+    """In case of errors in the browser, print them and return True"""
     for log_type in ['browser']:
         for log in client.get_log(log_type):
             if log['level'] is 'error':
@@ -33,24 +38,49 @@ def is_error(client):
                 return True
 
 
-labels = ['balance']
-# labels = ['balance', 'history', 'debug']
-for l in labels:
-    client.get('chrome-extension://' + ext_id + '/popup/popup.html#/' + l)
 
-# TODO assert here
-is_error(client)
+class PopupTestCase(unittest.TestCase):
+    """Test wallet's popups"""
+    def setUp(self):
+        ret = client_setup()
+        self.client = ret['client']
+        self.ext_id = ret['ext_id']
 
-# visit bank and trigger withdrawal
-client.get('https://bank.test.taler.net')
-client.get('https://bank.test.taler.net/accounts/register')
+    def tearDown(self):
+        self.client.close()
 
-register = """\
-    var form = document.getElementsByTagName('form');
-    form.username.value = 'test';
+    def test_popup(self):
+        # keeping only 'balance' to get tests faster. To be
+        # extended with 'history' and 'debug'
+        labels = ['balance']
+        for l in labels:
+            self.client.get('chrome-extension://' + self.ext_id + '/popup/popup.html#/' + l)
+        self.assertNotEqual(True, is_error(self.client))
+
+class BankTestCase(unittest.TestCase):
+    """Test withdrawal (after registering a new user)"""
+    def setUp(self):
+        ret = client_setup()
+        self.client = ret['client']
+        self.ext_id = ret['ext_id']
+
+    def tearDown(self):
+        self.client.close()
+    
+
+    def test_withdrawal(self):
+        bank_url = 'http://127.0.0.1:9898'
+        self.client.get(bank_url + '/accounts/register')
+
+        register = """\
+            var form = document.getElementsByTagName('form')[0];
+            form.username.value = '%s';
             form.password.value = 'test';
             form.submit();
-    """
+            """ % str(int(time.time())) # need fresh username
 
-# TODO assert here
-client.execute_script(register)
+        self.client.execute_script(register)
+        self.assertNotEqual(True, is_error(self.client))
+
+if __name__ == '__main__':
+    unittest.main()