pandorafms/tests/console/include/common_classes_60.py

# -*- coding: utf-8 -*-
from unittest2 import TestResult, TestCase
from common_functions_60 import *
from datetime import datetime
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
#from sauceclient import SauceClient
from os import environ
import os


class ArticaTestResult(TestResult):
	success = []
	def addSuccess(self, test):
		self.success.append((test,u'Success'))
		TestResult.addSuccess(self, test)

class PandoraWebDriverTestCase(TestCase):
	test_name = u'' #Name of the test.
	test_description = u'' #Description of the test
	time_started = None
	time_elapsed = None #Total time of the test
	tickets_associated = []
	#sauce_username = environ["SAUCE_USERNAME"]
	#sauce_access_key = environ["SAUCE_ACCESS_KEY"]
	#sauce_client = None
	#sauce_labs_job_id = None

	#desired_cap = {
	#	'tunnel-identifier': environ["TRAVIS_JOB_NUMBER"],
	#	'platform': "Windows 10",
	#	'browserName': "firefox",
	#	'version': "46",
	#}

	@classmethod
	def setUpClass(cls):
		cls.is_development = os.getenv('DEVELOPMENT', False)
		cls.is_enterprise = os.getenv('ENTERPRISE', False)
		if cls.is_development != False:
			cls.driver = webdriver.Firefox()
			cls.base_url = os.getenv('DEVELOPMENT_URL')
		else:
			#Start VM in Sauce Labs
			#cls.driver = webdriver.Remote(command_executor='http://'+cls.sauce_username+':'+cls.sauce_access_key+'@ondemand.saucelabs.com:80/wd/hub',desired_capabilities=cls.desired_cap)
			#cls.sauce_labs_job_id = cls.driver.session_id
			cls.driver = webdriver.Firefox()
			cls.base_url = "http://127.0.0.1/"
		
	@classmethod
	def tearDownClass(cls):
		cls.driver.quit()

	def setUp(self):
		self.time_started = datetime.now()
		self.driver.implicitly_wait(30)
		self.verificationErrors = []
		self.accept_next_alert = True
		#self.is_development = self.is_development
		#TODO Print test name
		print "Starting test"
		super(PandoraWebDriverTestCase, self).setUp()

	def is_element_present(self, how, what):
		try: self.driver.find_element(by=how, value=what)
		except NoSuchElementException, e: return False
		return True

	def is_alert_present(self):
		try: self.driver.switch_to_alert()
		except NoAlertPresentException, e: return False
		return True

	def close_alert_and_get_its_text(self):
		try:
			alert = self.driver.switch_to_alert()
			alert_text = alert.text
			if self.accept_next_alert:
				alert.accept()
			else:
				alert.dismiss()
			return alert_text
		finally: self.accept_next_alert = True

	def tearDown(self):
		tack = datetime.now()
		diff = tack - self.time_started
		self.time_elapsed = diff.seconds

		self.assertEqual([], self.verificationErrors)
		super(PandoraWebDriverTestCase, self).tearDown()


	def login(self,user="admin",passwd="pandora",pandora_url=None):
		print u"Logging in"

		driver = self.driver

		if pandora_url is None:
			pandora_url = self.base_url

		driver.get(pandora_url+"/pandora_console/index.php")
		driver.find_element_by_id("nick").clear()
		driver.find_element_by_id("nick").send_keys(user)
		driver.find_element_by_id("pass").clear()
		driver.find_element_by_id("pass").send_keys(passwd)
		driver.find_element_by_id("submit-login_button").click()

	def logout(self,pandora_url=None):
		print u"Logging out"

		driver = self.driver

		if pandora_url is None:
			pandora_url = self.base_url

		if pandora_url[-1] != '/':
			driver.find_element_by_xpath('//div[@id="container"]//a[@href="'+pandora_url+'/pandora_console/index.php?bye=bye"]').click()
		else:
			driver.find_element_by_xpath('//div[@id="container"]//a[@href="'+pandora_url+'pandora_console/index.php?bye=bye"]').click()

		driver.get(pandora_url+"/pandora_console/index.php")
		refresh_N_times_until_find_element(driver,2,"nick")