refactor to connector itself

Author: ptiurin

src/connection/base.ts

@@ -54,6 +54,8 @@ export abstract class Connection {
 
   abstract resolveEngineEndpoint(): Promise<string>;
 
+  abstract testConnection(): Promise<void>;
+
   protected getRequestUrl(executeQueryOptions: ExecuteQueryOptions): string {
     const params = this.getBaseParameters(executeQueryOptions);
 

src/connection/connection_v1.ts

@@ -59,4 +59,8 @@ export class ConnectionV1 extends BaseConnection {
     }
     return this.accountInfo;
   }
+
+  async testConnection() {
+    await this.execute("select 1");
+  }
 }

src/connection/connection_v2.ts

@@ -69,4 +69,9 @@ export class ConnectionV2 extends BaseConnection {
 
     return this.engineEndpoint;
   }
+
+  async testConnection() {
+    const settings = { internal: [{ auto_start_stop_control: "ignore" }] };
+    await this.execute("select 1", { settings });
+  }
 }

src/core/index.ts

@@ -31,10 +31,6 @@ export class FireboltCore {
     const connection = makeConnection(this.context, connectionOptions);
     await auth.authenticate();
     await connection.resolveEngineEndpoint();
-    // auto_start_stop_control is only available for v2
-    const settings = auth.isServiceAccount()
-      ? { internal: [{ auto_start_stop_control: "ignore" }] }
-      : {};
-    await connection.execute("select 1", { settings });
+    await connection.testConnection();
   }
 }