From aab6aee5695229139dafec6ca326c5365de0faf4 Mon Sep 17 00:00:00 2001
From: tinayuangao <yuan.gao.tina@gmail.com>
Date: Fri, 20 Jan 2017 09:39:36 -0800
Subject: [PATCH] test(e2e): Use current directory so e2e test won't complain
 about directory not exists (#2693)

* Use current directory so e2e test wont complain about directory not exists

* change .gitignore

* Only take screenshot when run on travis

* remove log
---
 .gitignore        | 1 +
 e2e/screenshot.ts | 6 ++++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index 08cd4a9d5ec..67f5b41b67d 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,6 +4,7 @@
 /dist
 /tmp
 /deploy
+/screenshots
 
 # dependencies
 /node_modules
diff --git a/e2e/screenshot.ts b/e2e/screenshot.ts
index 46990e695dd..28fad25225a 100644
--- a/e2e/screenshot.ts
+++ b/e2e/screenshot.ts
@@ -3,7 +3,7 @@ import * as gulp from 'gulp';
 import * as path from 'path';
 import {browser} from 'protractor';
 
-const OUTPUT_DIR = '/tmp/angular-material2-build/screenshots/';
+const OUTPUT_DIR = './screenshots/';
 
 let currentJasmineSpecName = '';
 
@@ -53,5 +53,7 @@ export class Screenshot {
 }
 
 export function screenshot(id: string) {
-  return new Screenshot(id);
+  if (process.env['TRAVIS']) {
+    return new Screenshot(id);
+  }
 }