Added test for table.import() callback invocation

googleapis · Apr 13, 2015 · 285415e · 285415e
1 parent 90e3175
commit 285415e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/test/bigquery/table.js b/test/bigquery/table.js
@@ -698,16 +698,22 @@ describe('BigQuery/Table', function() {
     }, 'file.json');
 
     it('should accept just a File and a callback', function(done) {
+      var mockJob = { id: 'foo' };
+
       table.createWriteStream = function() {
         var ws = new stream.Writable();
         setImmediate(function() {
-          ws.emit('complete');
+          ws.emit('complete', mockJob);
           ws.end();
         });
         return ws;
       };
 
-      table.import(FILEPATH, done);
+      table.import(FILEPATH, function(error, job) {
+        assert.strictEqual(error, null);
+        assert.deepEqual(job, mockJob);
+        done();
+      });
     });
 
     it('should return a stream when a string is given', function() {