File tree 2 files changed +19
-29
lines changed
2 files changed +19
-29
lines changed Original file line number Diff line number Diff line change @@ -24,16 +24,18 @@ var Dataset = require('../lib/bigquery/dataset');
24
24
var env = require ( './env' ) ;
25
25
var fs = require ( 'fs' ) ;
26
26
var Job = require ( '../lib/bigquery/job' ) ;
27
+ var uuid = require ( 'node-uuid' ) ;
27
28
28
29
var gcloud = require ( '../lib' ) ( env ) ;
29
30
var bigquery = gcloud . bigquery ( ) ;
30
- var bucket = gcloud . storage ( ) . bucket ( ) ;
31
+ var storage = gcloud . storage ( ) ;
31
32
32
33
describe ( 'BigQuery' , function ( ) {
33
34
var DATASET_ID = 'testDatasetId' ;
34
35
var dataset ;
35
36
var TABLE_ID = 'myKittens' ;
36
37
var table ;
38
+ var bucket ;
37
39
38
40
var query = 'SELECT url FROM [publicdata:samples.github_nested] LIMIT 100' ;
39
41
@@ -82,23 +84,17 @@ describe('BigQuery', function() {
82
84
} ) ;
83
85
} ,
84
86
85
- // Create a Bucket, if necessary .
87
+ // Create a Bucket.
86
88
function ( next ) {
87
- bucket . getMetadata ( function ( err ) {
88
- if ( ! err ) {
89
- next ( ) ;
89
+ var bucketName = 'gcloud-test-bucket-temp-' + uuid . v1 ( ) ;
90
+ storage . createBucket ( bucketName , function ( err , b ) {
91
+ if ( err ) {
92
+ next ( err ) ;
90
93
return ;
91
94
}
92
95
93
- gcloud . storage ( ) . createBucket ( bucket . name , function ( err , b ) {
94
- if ( err ) {
95
- next ( err ) ;
96
- return ;
97
- }
98
-
99
- bucket = b ;
100
- next ( ) ;
101
- } ) ;
96
+ bucket = b ;
97
+ next ( ) ;
102
98
} ) ;
103
99
}
104
100
] , done ) ;
Original file line number Diff line number Diff line change @@ -111,8 +111,8 @@ describe('storage', function() {
111
111
storage . getBuckets ( getBucketsHandler ) ;
112
112
113
113
var createdBuckets = [ ] ;
114
- var failedTests = 0 ;
115
- var MAX_TRIES = 2 ;
114
+ var retries = 0 ;
115
+ var MAX_RETRIES = 2 ;
116
116
117
117
function getBucketsHandler ( err , buckets , nextQuery ) {
118
118
buckets . forEach ( function ( bucket ) {
@@ -121,23 +121,17 @@ describe('storage', function() {
121
121
}
122
122
} ) ;
123
123
124
- function allCreated ( ) {
125
- assert . equal ( createdBuckets . length , bucketsToCreate . length ) ;
126
- }
127
-
128
- try {
129
- allCreated ( ) ;
130
- done ( ) ;
131
- } catch ( e ) {
132
- failedTests ++ ;
124
+ if ( createdBuckets . length < bucketsToCreate . length && nextQuery ) {
125
+ retries ++ ;
133
126
134
- if ( failedTests <= MAX_TRIES ) {
127
+ if ( retries <= MAX_RETRIES ) {
135
128
storage . getBuckets ( nextQuery , getBucketsHandler ) ;
136
- } else {
137
- // Crash.
138
- allCreated ( ) ;
129
+ return ;
139
130
}
140
131
}
132
+
133
+ assert . equal ( createdBuckets . length , bucketsToCreate . length ) ;
134
+ done ( ) ;
141
135
}
142
136
} ) ;
143
137
} ) ;
You can’t perform that action at this time.
0 commit comments