selenium tests: fixes and template creation test added

circle/dashboard/tests/selenium/test_vm_controlls.py

+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
 # Copyright 2014 Budapest University of Technology and Economics (BME IK)
 #
 # This file is part of CIRCLE Cloud.
@@ -19,11 +21,15 @@ from django.contrib.auth.models import User
 import random
 import urlparse
 import re
+import time
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.support.select import Select
 from selenium.webdriver.common.by import By
 random_pass = "".join([random.choice(
     '0123456789abcdefghijklmnopqrstvwxyz') for n in xrange(10)])
+random_accents = random_pass + "".join([random.choice(
+    u"áéíöóúűÁÉÍÖÓÜÚŰ") for n in xrange(5)])
 wait_max_sec = 10
 host = 'https:127.0.0.1'
 
@@ -52,28 +58,68 @@ class UtilityMixin(object):
             password_input.clear()
             password_input.send_keys(password)
             submit_input.click()
-            try:  # If selenium runs only in a small (virtual) screen
+            try:
+                # If selenium runs only in a small (virtual) screen
                 driver.find_element_by_class_name('navbar-toggle').click()
-            except:
-                pass
                 WebDriverWait(self.driver, wait_max_sec).until(
                     EC.element_to_be_clickable((
                         By.CSS_SELECTOR, "a[href*='/dashboard/profile/']")))
             except:
+                time.sleep(0.5)
+        except:
             raise Exception('Selenium cannot find the form controls')
 
+    def list_options(self, select):
+        try:
+            option_dic = {}
+            select = Select(select)
+            for option in select.options:
+                key = option.get_attribute('value')
+                if key is not None and key:
+                    option_dic[key] = [option.text]
+            return option_dic
+        except:
+            raise Exception(
+                'Selenium cannot list the select possibilities')
+
+    def select_option(self, select, what=None):
+        try:
+            my_choice = None
+            options = self.list_options(select)
+            select = Select(select)
+            if what is not None:
+                for key, value in options.iteritems():
+                    if what in key:
+                        my_choice = key
+                    else:
+                        if isinstance(value, list):
+                            for single_value in value:
+                                if what in single_value:
+                                    my_choice = key
+                        else:
+                            if what in value:
+                                my_choice = key
+            if my_choice is None:
+                my_choose_list = options.keys()
+                my_choice = my_choose_list[random.randint(
+                    0, len(my_choose_list) - 1)]
+            select.select_by_value(my_choice)
+        except:
+            raise Exception(
+                'Selenium cannot select the choosen one')
+
     def get_link_by_href(self, target_href, attributes=None):
         try:
             links = self.driver.find_elements_by_tag_name('a')
             for link in links:
                 href = link.get_attribute('href')
-                if href is not None:
+                if href is not None and href:
                     if target_href in href:
                         perfect_fit = True
                         if isinstance(attributes, dict):
                             for key, target_value in attributes.iteritems():
                                 attr_check = link.get_attribute(key)
-                                if attr_check is not None:
+                                if attr_check is not None and attr_check:
                                     if target_value not in attr_check:
                                         perfect_fit = False
                         if perfect_fit:
@@ -82,6 +128,105 @@ class UtilityMixin(object):
             raise Exception(
                 'Selenium cannot find the href=%s link' % target_href)
 
+    def click_on_link(self, link):
+        """
+        There are situations when selenium built in click() function
+        doesn't work as intended, that's when this function is used.
+        Fires a click event via javascript injection.
+        """
+        try:
+            #  Javascript function to simulate a click on a link
+            javascript = (
+                "var link = arguments[0];"
+                "var cancelled = false;"
+                "if(document.createEvent) {"
+                "   var event = document.createEvent(\"MouseEvents\");"
+                "   event.initMouseEvent("
+                "       \"click\", true, true, window, 0, 0, 0, 0, 0,"
+                "       false,false,false,false,0,null);"
+                "   cancelled = !link.dispatchEvent(event);"
+                "} else if(link.fireEvent) {"
+                "   cancelled = !link.fireEvent(\"onclick\");"
+                "} if (!cancelled) {"
+                "   window.location = link.href;"
+                "}")
+            self.driver.execute_script("%s" % javascript, link)
+        except:
+            raise Exception(
+                'Selenium cannot inject javascript to the page')
+
+    def wait_and_accept_operation(self, argument=None):
+        try:
+            accept = WebDriverWait(self.driver, wait_max_sec).until(
+                EC.element_to_be_clickable((
+                    By.ID, "op-form-send")))
+            if argument is not None:
+                possible = self.driver.find_elements_by_css_selector(
+                    "div.controls > input[type='text']")
+                if isinstance(argument, list):
+                    for x in range(0, len(possible)):
+                        possible[x].clear()
+                        possible[x].send_keys(argument[x % len(argument)])
+                else:
+                    for form in possible:
+                        form.clear()
+                        form.send_keys(argument)
+            accept.click()
+        except:
+            raise Exception("Selenium cannot accept the"
+                            " operation confirmation")
+
+    def create_base_template(self, name=None, architecture="x86-64",
+                             method=None, op_system=None, lease=None,
+                             network="vm"):
+        if name is None:
+            name = "template_%s" % random_accents
+        if op_system is None:
+            op_system = "!os %s" % random_accents
+        try:
+            self.driver.get('%s/dashboard/template/choose/' % host)
+            self.driver.find_element_by_css_selector(
+                "input[type='radio'][value='base_vm']").click()
+            next_button = self.driver.find_element_by_id(
+                "template-choose-next-button")
+            next_button.click()
+            template_name = WebDriverWait(self.driver, wait_max_sec).until(
+                EC.visibility_of_element_located((
+                    By.ID, 'id_name')))
+            template_name.clear()
+            template_name.send_keys(name)
+            self.select_option(self.driver.find_element_by_id(
+                "id_arch"), architecture)
+            self.select_option(self.driver.find_element_by_id(
+                "id_access_method"), method)
+            system_name = self.driver.find_element_by_id("id_system")
+            system_name.clear()
+            system_name.send_keys(op_system)
+            self.select_option(self.driver.find_element_by_id(
+                "id_lease"), lease)
+            self.select_option(self.driver.find_element_by_id(
+                "id_networks"), network)
+            self.driver.find_element_by_css_selector(
+                "input.btn[type='submit']").click()
+            WebDriverWait(self.driver, wait_max_sec).until(
+                EC.visibility_of_element_located((
+                    By.ID, 'ops')))
+            self.click_on_link(self.get_link_by_href('/op/deploy/'))
+            self.wait_and_accept_operation()
+            self.click_on_link(WebDriverWait(self.driver, wait_max_sec).until(
+                EC.element_to_be_clickable((
+                    By.CSS_SELECTOR, "a[href$='/op/shut_off/']"))))
+            self.wait_and_accept_operation()
+            WebDriverWait(self.driver, wait_max_sec).until(
+                EC.element_to_be_clickable((
+                    By.CSS_SELECTOR, "a[href$='/op/deploy/']")))
+            self.click_on_link(self.get_link_by_href('/op/save_as_template/'))
+            self.wait_and_accept_operation(name)
+            return name
+        except:
+            raise Exception(
+                'Selenium cannot create a base template virtual machine')
+
     def create_random_vm(self):
         try:
             self.driver.get('%s/dashboard/vm/create/' % host)
@@ -89,7 +234,6 @@ class UtilityMixin(object):
             pk = None
             vm_list = self.driver.find_elements_by_class_name(
                 'vm-create-template-summary')
-            self.driver.save_screenshot('screenie.png')
             choice = random.randint(0, len(vm_list) - 1)
             vm_list[choice].click()
             create = WebDriverWait(self.driver, wait_max_sec).until(
@@ -118,13 +262,13 @@ class UtilityMixin(object):
         list_view_link = self.get_link_by_href(
             '#index-list-view',
             required_attributes).find_element_by_tag_name('i')
-        list_view_link.click()
+        self.click_on_link(list_view_link)
         states = [driver.execute_script("%s" % js_script, list_view),
                   driver.execute_script("%s" % js_script, graph_view)]
-        graph_view_link.click()
+        self.click_on_link(graph_view_link)
         states.extend([driver.execute_script("%s" % js_script, list_view),
                        driver.execute_script("%s" % js_script, graph_view)])
-        list_view_link.click()
+        self.click_on_link(list_view_link)
         states.extend([driver.execute_script("%s" % js_script, list_view),
                        driver.execute_script("%s" % js_script, graph_view)])
         return states
@@ -149,17 +293,23 @@ class UtilityMixin(object):
 
 
 class VmDetailTest(UtilityMixin, SeleniumTestCase):
-    def setUp(self):
-        self.u1 = User.objects.create(username='test_%s' % random_pass,
+    template_id = None
+
+    @classmethod
+    def setup_class(cls):
+        cls._user = User.objects.create(username='test_%s' % random_accents,
                                         is_superuser=True)
-        self.u1.set_password(random_pass)
-        self.u1.save()
-        self.addCleanup(self.u1.delete)
+        cls._user.set_password(random_accents)
+        cls._user.save()
+
+    @classmethod
+    def teardown_class(cls):
+        cls._user.delete()
 
     def test_01_login(self):
         title = 'Dashboard | CIRCLE'
         location = '/dashboard/'
-        self.login('test_%s' % random_pass, random_pass)
+        self.login('test_%s' % random_accents, random_accents)
         self.driver.get('%s/dashboard/' % host)
         url = urlparse.urlparse(self.driver.current_url)
         (self.assertIn('%s' % title, self.driver.title,
@@ -167,8 +317,38 @@ class VmDetailTest(UtilityMixin, SeleniumTestCase):
             self.assertEqual(url.path, '%s' % location,
                              'URL path is not equal with %s' % location))
 
-    def test_02_able_to_create_vm(self):
-        self.login('test_%s' % random_pass, random_pass)
+    def test_02_able_to_create_template(self):
+        self.login('test_%s' % random_accents, random_accents)
+        template_list = None
+        create_template = self.get_link_by_href('/dashboard/template/choose/')
+        self.click_on_link(create_template)
+        WebDriverWait(self.driver, wait_max_sec).until(
+            EC.visibility_of_element_located((
+                By.ID, 'create-modal')))
+        template_list = self.driver.find_elements_by_class_name(
+            'template-choose-list-element')
+        print 'Selenium found %s template possibilities' % len(template_list)
+        (self.assertIsNotNone(
+            template_list, "Selenium can not find the create template list") or
+            self.assertGreater(len(template_list), 0,
+                               "The create template list is empty"))
+
+    def test_03_create_base_vm(self):
+        self.login('test_%s' % random_accents, random_accents)
+        template_name = self.create_base_template()
+        self.driver.get('%s/dashboard/template/list/' % host)
+        templates = self.driver.find_elements_by_css_selector("td.name")
+        found = False
+        for template in templates:
+            if template_name in template.text:
+                found = True
+                self.template_id = re.search(r'\d+', template.text).group()
+        self.assertTrue(
+            found,
+            "Coud not found the created template in the template list")
+
+    def test_10_able_to_create_vm(self):
+        self.login('test_%s' % random_accents, random_accents)
         vm_list = None
         create_vm_link = self.get_link_by_href('/dashboard/vm/create/')
         create_vm_link.click()
@@ -177,18 +357,20 @@ class VmDetailTest(UtilityMixin, SeleniumTestCase):
                 By.ID, 'create-modal')))
         vm_list = self.driver.find_elements_by_class_name(
             'vm-create-template-summary')
-        print 'Selenium found %s template possibilities' % len(vm_list)
+        print ("Selenium found %(vm_number)s virtual machine template "
+               " possibilities" % {
+                   'vm_number': len(vm_list)})
         (self.assertIsNotNone(
             vm_list, "Selenium can not find the VM list") or
             self.assertGreater(len(vm_list), 0, "The create VM list is empty"))
 
-    def test_03_create_vm(self):
-        self.login('test_%s' % random_pass, random_pass)
+    def test_11_create_vm(self):
+        self.login('test_%s' % random_accents, random_accents)
         pk = self.create_random_vm()
         self.assertIsNotNone(pk, "Can not create a VM")
 
-    def test_04_vm_view_change(self):
-        self.login('test_%s' % random_pass, random_pass)
+    def test_12_vm_view_change(self):
+        self.login('test_%s' % random_accents, random_accents)
         expected_states = ["", "none",
                            "none", "",
                            "block", "none"]
@@ -198,8 +380,8 @@ class VmDetailTest(UtilityMixin, SeleniumTestCase):
         self.assertListEqual(states, expected_states,
                              "The view mode does not change for VM listing")
 
-    def test_05_node_view_change(self):
-        self.login('test_%s' % random_pass, random_pass)
+    def test_13_node_view_change(self):
+        self.login('test_%s' % random_accents, random_accents)
         expected_states = ["", "none",
                            "none", "",
                            "block", "none"]
@@ -209,7 +391,7 @@ class VmDetailTest(UtilityMixin, SeleniumTestCase):
         self.assertListEqual(states, expected_states,
                              "The view mode does not change for NODE listing")
 
-    def test_06_delete_vm(self):
-        self.login('test_%s' % random_pass, random_pass)
+    def test_14_delete_vm(self):
+        self.login('test_%s' % random_accents, random_accents)
         pk = self.create_random_vm()
         self.assertTrue(self.delete_vm(pk), "Can not delete a VM")