|
15 | 15 |
|
16 | 16 | var async = require('async'); |
17 | 17 | var path = require('path'); |
18 | | -var test = require('ava'); |
19 | 18 | var loadDataFromCsvExample = require('../../bigquery/load_data_from_csv'); |
20 | 19 | var pathToCsvFile = path.join(__dirname, '/../../bigquery/resources/data.csv'); |
21 | 20 | var datasetId = 'nodejs_docs_samples'; |
22 | 21 | var tableName = 'test_' + new Date().getTime() + '_' + |
23 | 22 | Math.floor(Math.random() * 10000); |
24 | 23 |
|
25 | | -test.cb.serial('should load data from a csv file', function (t) { |
26 | | - async.series([ |
27 | | - function (cb) { |
28 | | - loadDataFromCsvExample.createTable(datasetId, tableName, cb); |
29 | | - }, |
30 | | - function (cb) { |
31 | | - loadDataFromCsvExample.main(pathToCsvFile, datasetId, tableName, cb); |
32 | | - }, |
33 | | - function (cb) { |
34 | | - loadDataFromCsvExample.deleteTable(datasetId, tableName, cb); |
35 | | - } |
36 | | - ], function (err, results) { |
37 | | - if (err) { |
38 | | - loadDataFromCsvExample.deleteTable(datasetId, tableName, function () { |
39 | | - t.end(err); |
40 | | - }); |
41 | | - } else { |
42 | | - t.ifError(err); |
43 | | - // metadata |
44 | | - t.is(results[1].status.state, 'DONE'); |
45 | | - t.end(); |
46 | | - } |
| 24 | +describe('bigquery:load_data_from_csv', function () { |
| 25 | + it('should load data from a csv file', function (done) { |
| 26 | + async.series([ |
| 27 | + function (cb) { |
| 28 | + loadDataFromCsvExample.createTable(datasetId, tableName, cb); |
| 29 | + }, |
| 30 | + function (cb) { |
| 31 | + loadDataFromCsvExample.main(pathToCsvFile, datasetId, tableName, cb); |
| 32 | + }, |
| 33 | + function (cb) { |
| 34 | + loadDataFromCsvExample.deleteTable(datasetId, tableName, cb); |
| 35 | + } |
| 36 | + ], function (err, results) { |
| 37 | + if (err) { |
| 38 | + loadDataFromCsvExample.deleteTable(datasetId, tableName, function () { |
| 39 | + done(err); |
| 40 | + }); |
| 41 | + } else { |
| 42 | + assert(!err); |
| 43 | + // metadata |
| 44 | + assert.equal(results[1].status.state, 'DONE'); |
| 45 | + done(); |
| 46 | + } |
| 47 | + }); |
47 | 48 | }); |
48 | | -}); |
49 | 49 |
|
50 | | -test('should require correct arguments', function (t) { |
51 | | - t.throws(function () { |
52 | | - loadDataFromCsvExample.main(); |
53 | | - }, Error, 'pathToCsvFile is required!'); |
54 | | - t.throws(function () { |
55 | | - loadDataFromCsvExample.main(pathToCsvFile); |
56 | | - }, Error, 'datasetId is required!'); |
57 | | - t.throws(function () { |
58 | | - loadDataFromCsvExample.main(pathToCsvFile, datasetId); |
59 | | - }, Error, 'tableName is required!'); |
| 50 | + it('should require correct arguments', function () { |
| 51 | + assert.throws(function () { |
| 52 | + loadDataFromCsvExample.main(); |
| 53 | + }, Error, 'pathToCsvFile is required!'); |
| 54 | + assert.throws(function () { |
| 55 | + loadDataFromCsvExample.main(pathToCsvFile); |
| 56 | + }, Error, 'datasetId is required!'); |
| 57 | + assert.throws(function () { |
| 58 | + loadDataFromCsvExample.main(pathToCsvFile, datasetId); |
| 59 | + }, Error, 'tableName is required!'); |
| 60 | + }); |
60 | 61 | }); |
0 commit comments