8
8
openFirebaseScreenshotsDatabase ,
9
9
connectFirebaseScreenshots } from '../util/firebase' ;
10
10
import { setGithubStatus } from '../util/github' ;
11
+ import { isTravisPushBuild } from '../util/travis-ci' ;
11
12
12
13
const imageDiff = require ( 'image-diff' ) ;
13
14
@@ -23,7 +24,13 @@ const FIREBASE_FILELIST = 'screenshot/filenames';
23
24
/** Task which upload screenshots generated from e2e test. */
24
25
task ( 'screenshots' , ( ) => {
25
26
let prNumber = process . env [ 'TRAVIS_PULL_REQUEST' ] ;
26
- if ( prNumber ) {
27
+ if ( isTravisPushBuild ( ) ) {
28
+ // Only update golds and filenames for build
29
+ let database = openFirebaseScreenshotsDatabase ( ) ;
30
+ uploadScreenshots ( )
31
+ . then ( ( ) => setScreenFilenames ( database ) )
32
+ . then ( ( ) => database . goOffline ( ) , ( ) => database . goOffline ( ) ) ;
33
+ } else if ( prNumber ) {
27
34
let firebaseApp = connectFirebaseScreenshots ( ) ;
28
35
let database = firebaseApp . database ( ) ;
29
36
@@ -36,12 +43,6 @@ task('screenshots', () => {
36
43
. then ( ( ) => updateTravis ( database , prNumber ) )
37
44
. then ( ( ) => setScreenFilenames ( database , prNumber ) )
38
45
. then ( ( ) => database . goOffline ( ) , ( ) => database . goOffline ( ) ) ;
39
- } else if ( process . env [ 'TRAVIS' ] ) {
40
- // Only update golds and filenames for build
41
- let database = openFirebaseScreenshotsDatabase ( ) ;
42
- uploadScreenshots ( )
43
- . then ( ( ) => setScreenFilenames ( database ) )
44
- . then ( ( ) => database . goOffline ( ) , ( ) => database . goOffline ( ) ) ;
45
46
}
46
47
} ) ;
47
48
@@ -51,7 +52,8 @@ function updateFileResult(database: firebase.database.Database, prNumber: string
51
52
}
52
53
53
54
function updateResult ( database : firebase . database . Database , prNumber : string , result : boolean ) {
54
- return getPullRequestRef ( database , prNumber ) . child ( 'result' ) . set ( result ) . then ( ( ) => result ) ;
55
+ return getPullRequestRef ( database , prNumber ) . child ( 'result' )
56
+ . child ( process . env [ 'TRAVIS_PULL_REQUEST_SHA' ] ) . set ( result ) . then ( ( ) => result ) ;
55
57
}
56
58
57
59
function getPullRequestRef ( database : firebase . database . Database | admin . database . Database ,
0 commit comments