From c572a195aa1de69348f24deb08fcd720c6960781 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Du=C5=A1an=20Jureti=C4=87?= <dusan.juretic@gmail.com>
Date: Sat, 28 Dec 2013 23:44:41 -0300
Subject: [PATCH] Take screenshot of failed e2e tests

---
 test/e2e/e2e.js | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/test/e2e/e2e.js b/test/e2e/e2e.js
index f346ad0e78..26e76640d6 100644
--- a/test/e2e/e2e.js
+++ b/test/e2e/e2e.js
@@ -7,6 +7,22 @@ describe('front page', function() {
     browser.sleep(1000);
   });
 
+  // based on https://github.com/angular/protractor/issues/114#issuecomment-29046939
+  afterEach(function(){
+    var currentSpec = jasmine.getEnv().currentSpec;
+    var passed = currentSpec.results().passed();
+    if(!passed){
+      var filename = 'exception_' + currentSpec.description + '.png';
+      browser.takeScreenshot().then(function(png){
+        var fs = require('fs');
+        var buffer = new Buffer(png, 'base64');
+        var stream = fs.createWriteStream(filename);
+        stream.write(buffer);
+        stream.end();
+      });
+    }
+  });
+
   it('shows the front page', function(){
     var button = element(by.className('btn'));
     expect(button.getText()).toEqual('Play');