def setUpClass(cls):
# Start FireFox
try:
cls.client = webdriver.FireFox()
except:
pass
# Skip these tests if the browser could not be started
if cls.client:
# Create the application
cls.app = create_app("testing")
cls.app_context = cls.app.app_context()
cls.app_context.push()
# Suppress logging to keep unittest output clean
import logging
logger = logging.getLogger("werkzeug")
logger.setLevel("ERROR")
# Create the database and populate with some fake data
db.create_all()
Role.insert_roles()
User.generate_fake(10)
Post.generate_fake(10)
# Add an administrator user
admin_role = Role.query.filter_by(permissions=0xff).first()
admin = User(email="*****@*****.**",
username="******", password="******",
role=admin_role, confirmed=True)
db.session.add(admin)
db.session.commit()
# Start the flask server in a thread
threading.Thread(target=cls.app_run).start()
def __init__(self, ghost_log_file=None, firefox=False):
if firefox:
# Visual display in firefox
self.driver = webdriver.FireFox()
else:
#driver = webdriver.PhantomJS(service_log_path='/var/log/phantomjs/ghostdriver.log')
self.driver = webdriver.PhantomJS(PHANTOMJS_PATH, service_log_path=ghost_log_file)
self.scrapers = []
def __init__(self, website, user_name, password, browser='chrome'):
self.website = website
self.user_name = user_name
self.password = password
self.driver = None
if browser == 'chrome':
self.driver = webdriver.Chrome()
if browser == 'firefox':
self.driver = webdriver.FireFox()
def test_Firefox(self):
try:
self.FirefoxDriver = webdriver.FireFox()
firefox = Browser(self.FirefoxDriver)
self.assertTrue(firefox.TestFailLogin())
self.assertTrue(firefox.TestFailRegisterAccount())
self.assertTrue(firefox.TestSuccessRegisterAccount())
self.assertTrue(firefox.TestSucceessLogin())
except:
self.assertTrue(True)
def get_webpage():
## dummyt make call to dependecnies
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
driver = webdriver.FireFox()
driver.get("https://hrms.vinsys.com/Login")
username_field = driver.find_element_by_name("user")
username_field.send_keys("*****@*****.**")
password_field = driver.find_element_by_name("password")
password_field.send_keys("somepassword")
return driver
def driver(request):
#url = request.config.getoption("--baseurl")
browser = request.config.getoption("--browser")
if browser == "chrome":
driver = webdriver.Chrome()
elif browser == "firefox":
driver = webdriver.FireFox()
driver.implicitly_wait(5)
driver.maximize_window()
request.cls.driver = driver #request all class's driver =driver
yield
driver.close()
driver.quit()
print("Test Completed")
def test_setup(request):
from selenium import webdriver
browser = request.config.getoption("--browser")
if browser == "chrome":
driver = webdriver.Chrome()
elif browser == "firefox":
driver = webdriver.FireFox()
driver.implicitly_wait(5)
driver.maximize_window()
request.cls.driver = driver #request all class's driver =driver
yield
driver.close()
driver.quit()
print("Test Completed")
def index(request):
"""
Takes user to the main index page.
"""
if request.method == 'POST':
form = ArticleForm(request.POST)
if form.is_valid():
# Retrieve article
url = form.cleaned_data['article_url']
# Create Article model instance
article = Article(url=url)
# Initiate selenium webdriver
driver = webdriver.FireFox()
# Go to url
driver.get(url)
# Get type of article
article.type = driver.find_element_by_xpath(
'//meta[@property="article:top-level-section"]/@content')
# Get article title
article.title = driver.find_element_by_xpath(
'//meta[@name="hdl"]/@content')
# Get article author
article.author = driver.find_element_by_xpath(
'//meta[@name="author"]/@content')
# Get article text
article.text = driver.find_element_by_xpath('//p')
# Save article to database
article.save()
# store url in session
request.session['article_url'] = url
return HttpResponseRedirect('/results.html')
else:
form = ArticleForm()
return render(request, 'classifier/index.html', {'form': form})
def setupClass(cls):
#start firefox
try:
cls.client = webdriver.FireFox()
except:
pass
#skip these tests if the browser could not be started
if cls.client:
#create the Application
cls.app = create_app('testing')
cls.app_context = cls.app.app_context()
cls.app_context.push()
#suppress logging to keep unittest output clean
import logging
logger = logging.getLogger('werkzeug')
logger.setLevel('ERROR')
#create the database and populate with same fake data
db.create_all()
Role.insert_roles()
User.generate_fake(10)
Post.generate_fake(10)
#add an administrator user
admin_role = Role.query.filter_by(permissions=0xff).first()
admin = User(email='*****@*****.**',
password='******',
role=admin_role,
confirmed=True)
db.session.add(admin)
db.session.commit()
#start the Flask server in a thread
threading.Thread(target=cls.app.run).start()
# give the server a second to ensure it is up
time.sleep(1)
def getWFSlot(productUrl):
driver = webdriver.FireFox()
driver.get(productUrl)
html = driver.page_source
soup = bs4.BeautifulSoup(html)
time.sleep(60)
no_open_slots = True
while no_open_slots:
driver.refresh()
print("refreshed")
html = driver.page_source
soup = bs4.BeautifulSoup(html)
time.sleep(4)
slot_pattern = 'Next available'
try:
next_slot_text = soup.find(
'h4', class_='ufss-slotgroup-heading-text a-text-normal').text
if slot_pattern in next_slot_text:
print('SLOTS OPEN!')
os.system('say "Slots for delivery opened!"')
no_open_slots = False
time.sleep(1400)
except AttributeError:
continue
try:
no_slot_pattern = 'No delivery windows available. New windows are released throughout the day.'
if no_slot_pattern == soup.find('h4',
class_='a-alert-heading').text:
print("NO SLOTS!")
except AttributeError:
print('SLOTS OPEN!')
os.system('say "Slots for delivery opened!"')
no_open_slots = False
def open_browser(self, browser):
if browser == 'Chrome':
self.driver = webdriver.Chrome()
else:
driver = webdriver.FireFox()
from selenium import webdriver
browser = webdriver.FireFox()
url = 'https://www.youtube.com'
search_term = "bla"
enter = webdriver.common.keys.Keys.RETURN
browser.get(url)
search_input = browser.find_element_by_id("masthead-search-term")
search_input.send_keys(search_term)
submit_button = browser.find_element_by_id("search-btn")
submit_button.send_keys(enter)
def setUp(self):
self.browser = webdriver.FireFox()
self.browser.implicitly_wait(3)
def _create_firefox_driver(self):
return webdriver.FireFox()
def app(request, browser_type, base_url):
if browser_type == "Chrome":
driver = webdriver.Chrome()
elif browser_type == "firefox"
driver == webdriver.FireFox ("C:\FF_driver\geckodriver.exe")
def setUp(self): self.driver=webdriver.FireFox() self.driver.maximize_window() self.driver.implicitly_wait(30) self.driver.get(usecsv.getcsv(0,0))
def setUp(self):
self.browser = webdriver.FireFox()
stage_browser = os.environ.get('STAGING_SERVER')
if stage_server:
self.live_server_url = 'http://' + stage_server