1 files changed, 23 insertions, 0 deletions
diff --git a/py-fake-service/features/environment.py b/py-fake-service/features/environment.py
new file mode 100644
index 00000000..0ec3558e
--- /dev/null
+++ b/py-fake-service/features/environment.py
@@ -0,0 +1,23 @@
+from selenium import webdriver
+
+
+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'))