1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
#
# Copyright (c) 2014 ThoughtWorks, Inc.
#
# Pixelated is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Pixelated is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
from selenium import webdriver
from test.support.integration_helper import SoledadTestBase
import time
import pixelated.user_agent
import multiprocessing
def before_all(context):
context.soledad_test_base = SoledadTestBase()
context.soledad_test_base.setup_soledad()
context.mailboxes = context.soledad_test_base.pixelated_mailboxes
context.app = pixelated.user_agent.app
context.app.mail_service = context.soledad_test_base.mail_service
worker = lambda app, port: pixelated.user_agent.app.run(port=4567, use_reloader=False)
context._process = multiprocessing.Process(target=worker, args=(context.app, 4567))
context._process.start()
# we must wait the server start listening
time.sleep(1)
def after_all(context):
context.soledad_test_base.teardown_soledad()
context._process.terminate()
def before_feature(context, feature):
# context.browser = webdriver.Firefox()
context.browser = webdriver.PhantomJS()
context.browser.set_window_size(1280, 1024)
context.browser.implicitly_wait(5)
context.browser.set_page_load_timeout(60) # wait for data
context.browser.get('http://localhost:4567/')
def after_feature(context, feature):
context.browser.quit()
def take_screenshot(context):
context.browser.save_screenshot('/tmp/screenshot.jpeg')
def save_source(context):
with open('/tmp/source.html', 'w') as out:
out.write(context.browser.page_source.encode('utf8'))
|
