Skip to content

Commit c05aaa1

Browse files
fix BQ error + retry retry logic
1 parent a049907 commit c05aaa1

File tree

2 files changed

+19
-29
lines changed

2 files changed

+19
-29
lines changed

regression/bigquery.js

+10-14
Original file line numberDiff line numberDiff line change
@@ -24,16 +24,18 @@ var Dataset = require('../lib/bigquery/dataset');
2424
var env = require('./env');
2525
var fs = require('fs');
2626
var Job = require('../lib/bigquery/job');
27+
var uuid = require('node-uuid');
2728

2829
var gcloud = require('../lib')(env);
2930
var bigquery = gcloud.bigquery();
30-
var bucket = gcloud.storage().bucket();
31+
var storage = gcloud.storage();
3132

3233
describe('BigQuery', function() {
3334
var DATASET_ID = 'testDatasetId';
3435
var dataset;
3536
var TABLE_ID = 'myKittens';
3637
var table;
38+
var bucket;
3739

3840
var query = 'SELECT url FROM [publicdata:samples.github_nested] LIMIT 100';
3941

@@ -82,23 +84,17 @@ describe('BigQuery', function() {
8284
});
8385
},
8486

85-
// Create a Bucket, if necessary.
87+
// Create a Bucket.
8688
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);
9093
return;
9194
}
9295

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();
10298
});
10399
}
104100
], done);

regression/storage.js

+9-15
Original file line numberDiff line numberDiff line change
@@ -111,8 +111,8 @@ describe('storage', function() {
111111
storage.getBuckets(getBucketsHandler);
112112

113113
var createdBuckets = [];
114-
var failedTests = 0;
115-
var MAX_TRIES = 2;
114+
var retries = 0;
115+
var MAX_RETRIES = 2;
116116

117117
function getBucketsHandler(err, buckets, nextQuery) {
118118
buckets.forEach(function(bucket) {
@@ -121,23 +121,17 @@ describe('storage', function() {
121121
}
122122
});
123123

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++;
133126

134-
if (failedTests <= MAX_TRIES) {
127+
if (retries <= MAX_RETRIES) {
135128
storage.getBuckets(nextQuery, getBucketsHandler);
136-
} else {
137-
// Crash.
138-
allCreated();
129+
return;
139130
}
140131
}
132+
133+
assert.equal(createdBuckets.length, bucketsToCreate.length);
134+
done();
141135
}
142136
});
143137
});

0 commit comments

Comments
 (0)