chore: add string type to Table.query & Dataset.query (#1000)

Author: steffnay

handwritten/bigquery/src/dataset.ts

@@ -1139,7 +1139,9 @@ class Dataset extends ServiceObject {
   }
 
   query(options: Query): Promise<QueryRowsResponse>;
+  query(options: string): Promise<QueryRowsResponse>;
   query(options: Query, callback: SimpleQueryRowsCallback): void;
+  query(options: string, callback: SimpleQueryRowsCallback): void;
   /**
    * Run a query scoped to your dataset.
    *
@@ -1151,7 +1153,7 @@ class Dataset extends ServiceObject {
    * @returns {Promise<QueryRowsResponse>} See {@link BigQuery#query} for full documentation of this method.
    */
   query(
-    options: Query,
+    options: Query | string,
     callback?: SimpleQueryRowsCallback
   ): void | Promise<QueryRowsResponse> {
     if (typeof options === 'string') {

handwritten/bigquery/src/table.ts

@@ -2271,6 +2271,7 @@ class Table extends common.ServiceObject {
   }
 
   query(query: Query): Promise<SimpleQueryRowsResponse>;
+  query(query: string): Promise<SimpleQueryRowsResponse>;
   query(query: Query, callback: SimpleQueryRowsCallback): void;
   /**
    * Run a query scoped to your dataset.
@@ -2281,9 +2282,14 @@ class Table extends common.ServiceObject {
    * @returns {Promise<SimpleQueryRowsResponse>}
    */
   query(
-    query: Query,
+    query: Query | string,
     callback?: SimpleQueryRowsCallback
   ): void | Promise<SimpleQueryRowsResponse> {
+    if (typeof query === 'string') {
+      query = {
+        query,
+      };
+    }
     this.dataset.query(query, callback!);
   }
 

handwritten/bigquery/test/table.ts

@@ -2784,7 +2784,7 @@ describe('BigQuery/Table', () => {
   describe('query', () => {
     it('should pass args through to datasetInstance.query()', done => {
       table.dataset.query = (a: {}, b: {}) => {
-        assert.strictEqual(a, 'a');
+        assert.deepStrictEqual(a, {query: 'a'});
         assert.strictEqual(b, 'b');
         done();
       };