[PECO-1390] Configure Typescript for tests (#256)

* Configure TypeScript for unit and e2e tests Signed-off-by: Levko Kravets <[email protected]> * Fix imports to properly collect coverage Signed-off-by: Levko Kravets <[email protected]> --------- Signed-off-by: Levko Kravets <[email protected]>
databricks · Apr 29, 2024 · fac3345 · fac3345
1 parent 9913384
commit fac3345
Show file tree

Hide file tree

Showing 57 changed files with 388 additions and 107 deletions.
diff --git a/nyc.config.js b/nyc.config.js
@@ -1,5 +1,9 @@
 'use strict';
 
 module.exports = {
+  require: ['ts-node/register'],
+  reporter: ['lcov'],
+  all: true,
+  include: ['lib/**'],
   exclude: ['thrift/**', 'tests/**'],
 };
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -47,7 +47,9 @@
   ],
   "license": "Apache 2.0",
   "devDependencies": {
+    "@types/chai": "^4.3.14",
     "@types/lz4": "^0.6.4",
+    "@types/mocha": "^10.0.6",
     "@types/node": "^18.11.9",
     "@types/node-fetch": "^2.6.4",
     "@types/node-int64": "^0.4.29",
@@ -69,6 +71,7 @@
     "nyc": "^15.1.0",
     "prettier": "^2.8.4",
     "sinon": "^14.0.0",
+    "ts-node": "^10.9.2",
     "typescript": "^4.9.3"
   },
   "dependencies": {

diff --git a/tests/e2e/arrow.test.js b/tests/e2e/arrow.test.js
@@ -2,10 +2,10 @@ const { expect } = require('chai');
 const sinon = require('sinon');
 const config = require('./utils/config');
 const logger = require('./utils/logger')(config.logger);
-const { DBSQLClient } = require('../..');
-const ArrowResultHandler = require('../../dist/result/ArrowResultHandler').default;
-const ArrowResultConverter = require('../../dist/result/ArrowResultConverter').default;
-const ResultSlicer = require('../../dist/result/ResultSlicer').default;
+const { DBSQLClient } = require('../../lib');
+const ArrowResultHandler = require('../../lib/result/ArrowResultHandler').default;
+const ArrowResultConverter = require('../../lib/result/ArrowResultConverter').default;
+const ResultSlicer = require('../../lib/result/ResultSlicer').default;
 
 const fixtures = require('../fixtures/compatibility');
 const { expected: expectedColumn } = require('../fixtures/compatibility/column');

diff --git a/tests/e2e/batched_fetch.test.js b/tests/e2e/batched_fetch.test.js
@@ -2,7 +2,7 @@ const { expect } = require('chai');
 const sinon = require('sinon');
 const config = require('./utils/config');
 const logger = require('./utils/logger')(config.logger);
-const { DBSQLClient } = require('../..');
+const { DBSQLClient } = require('../../lib');
 
 async function openSession(customConfig) {
   const client = new DBSQLClient();

diff --git a/tests/e2e/cloudfetch.test.js b/tests/e2e/cloudfetch.test.js
@@ -1,10 +1,10 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const config = require('./utils/config');
-const { DBSQLClient } = require('../..');
-const CloudFetchResultHandler = require('../../dist/result/CloudFetchResultHandler').default;
-const ArrowResultConverter = require('../../dist/result/ArrowResultConverter').default;
-const ResultSlicer = require('../../dist/result/ResultSlicer').default;
+const { DBSQLClient } = require('../../lib');
+const CloudFetchResultHandler = require('../../lib/result/CloudFetchResultHandler').default;
+const ArrowResultConverter = require('../../lib/result/ArrowResultConverter').default;
+const ResultSlicer = require('../../lib/result/ResultSlicer').default;
 
 async function openSession(customConfig) {
   const client = new DBSQLClient();

diff --git a/tests/e2e/data_types.test.js b/tests/e2e/data_types.test.js
@@ -2,7 +2,7 @@ const { expect } = require('chai');
 const sinon = require('sinon');
 const config = require('./utils/config');
 const logger = require('./utils/logger')(config.logger);
-const { DBSQLClient } = require('../..');
+const { DBSQLClient } = require('../../lib');
 
 async function openSession(customConfig) {
   const client = new DBSQLClient();

diff --git a/tests/e2e/proxy.test.js b/tests/e2e/proxy.test.js
@@ -3,7 +3,7 @@ const sinon = require('sinon');
 const httpProxy = require('http-proxy');
 const https = require('https');
 const config = require('./utils/config');
-const { DBSQLClient } = require('../..');
+const { DBSQLClient } = require('../../lib');
 
 class HttpProxyMock {
   constructor(target, port) {

diff --git a/tests/e2e/query_parameters.test.js b/tests/e2e/query_parameters.test.js
@@ -1,8 +1,8 @@
 const { expect, AssertionError } = require('chai');
 const Int64 = require('node-int64');
 const config = require('./utils/config');
-const { DBSQLClient, DBSQLParameter, DBSQLParameterType } = require('../..');
-const ParameterError = require('../../dist/errors/ParameterError').default;
+const { DBSQLClient, DBSQLParameter, DBSQLParameterType } = require('../../lib');
+const ParameterError = require('../../lib/errors/ParameterError').default;
 
 const openSession = async () => {
   const client = new DBSQLClient();

diff --git a/tests/e2e/staging_ingestion.test.js b/tests/e2e/staging_ingestion.test.js
@@ -4,8 +4,8 @@ const path = require('path');
 const os = require('os');
 const uuid = require('uuid');
 const config = require('./utils/config');
-const { DBSQLClient } = require('../..');
-const StagingError = require('../../dist/errors/StagingError').default;
+const { DBSQLClient } = require('../../lib');
+const StagingError = require('../../lib/errors/StagingError').default;
 
 describe('Staging Test', () => {
   const catalog = config.database[0];

diff --git a/tests/e2e/timeouts.test.js b/tests/e2e/timeouts.test.js
@@ -1,7 +1,7 @@
 const { expect, AssertionError } = require('chai');
 const sinon = require('sinon');
 const config = require('./utils/config');
-const { DBSQLClient } = require('../..');
+const { DBSQLClient } = require('../../lib');
 
 async function openSession(socketTimeout, customConfig) {
   const client = new DBSQLClient();

diff --git a/tests/unit/DBSQLClient.test.js b/tests/unit/DBSQLClient.test.js
@@ -1,16 +1,13 @@
 const { expect, AssertionError } = require('chai');
 const sinon = require('sinon');
-const DBSQLClient = require('../../dist/DBSQLClient').default;
-const DBSQLSession = require('../../dist/DBSQLSession').default;
+const DBSQLClient = require('../../lib/DBSQLClient').default;
+const DBSQLSession = require('../../lib/DBSQLSession').default;
 
-const PlainHttpAuthentication = require('../../dist/connection/auth/PlainHttpAuthentication').default;
-const DatabricksOAuth = require('../../dist/connection/auth/DatabricksOAuth').default;
-const {
-  DatabricksOAuthManager,
-  AzureOAuthManager,
-} = require('../../dist/connection/auth/DatabricksOAuth/OAuthManager');
+const PlainHttpAuthentication = require('../../lib/connection/auth/PlainHttpAuthentication').default;
+const DatabricksOAuth = require('../../lib/connection/auth/DatabricksOAuth').default;
+const { DatabricksOAuthManager, AzureOAuthManager } = require('../../lib/connection/auth/DatabricksOAuth/OAuthManager');
 
-const HttpConnectionModule = require('../../dist/connection/connections/HttpConnection');
+const HttpConnectionModule = require('../../lib/connection/connections/HttpConnection');
 
 const { default: HttpConnection } = HttpConnectionModule;
 

diff --git a/tests/unit/DBSQLOperation.test.js b/tests/unit/DBSQLOperation.test.js
@@ -1,16 +1,16 @@
 const { expect, AssertionError } = require('chai');
 const sinon = require('sinon');
-const { DBSQLLogger, LogLevel } = require('../../dist');
+const { DBSQLLogger, LogLevel } = require('../../lib');
 const { TStatusCode, TOperationState, TTypeId, TSparkRowSetType } = require('../../thrift/TCLIService_types');
-const DBSQLOperation = require('../../dist/DBSQLOperation').default;
-const StatusError = require('../../dist/errors/StatusError').default;
-const OperationStateError = require('../../dist/errors/OperationStateError').default;
-const HiveDriverError = require('../../dist/errors/HiveDriverError').default;
-const JsonResultHandler = require('../../dist/result/JsonResultHandler').default;
-const ArrowResultConverter = require('../../dist/result/ArrowResultConverter').default;
-const ArrowResultHandler = require('../../dist/result/ArrowResultHandler').default;
-const CloudFetchResultHandler = require('../../dist/result/CloudFetchResultHandler').default;
-const ResultSlicer = require('../../dist/result/ResultSlicer').default;
+const DBSQLOperation = require('../../lib/DBSQLOperation').default;
+const StatusError = require('../../lib/errors/StatusError').default;
+const OperationStateError = require('../../lib/errors/OperationStateError').default;
+const HiveDriverError = require('../../lib/errors/HiveDriverError').default;
+const JsonResultHandler = require('../../lib/result/JsonResultHandler').default;
+const ArrowResultConverter = require('../../lib/result/ArrowResultConverter').default;
+const ArrowResultHandler = require('../../lib/result/ArrowResultHandler').default;
+const CloudFetchResultHandler = require('../../lib/result/CloudFetchResultHandler').default;
+const ResultSlicer = require('../../lib/result/ResultSlicer').default;
 
 class OperationHandleMock {
   constructor(hasResultSet = true) {

diff --git a/tests/unit/DBSQLParameter.test.js b/tests/unit/DBSQLParameter.test.js
@@ -2,7 +2,7 @@ const { expect } = require('chai');
 
 const Int64 = require('node-int64');
 const { TSparkParameterValue, TSparkParameter } = require('../../thrift/TCLIService_types');
-const { DBSQLParameter, DBSQLParameterType } = require('../../dist/DBSQLParameter');
+const { DBSQLParameter, DBSQLParameterType } = require('../../lib/DBSQLParameter');
 
 describe('DBSQLParameter', () => {
   it('should infer types correctly', () => {

diff --git a/tests/unit/DBSQLSession.test.js b/tests/unit/DBSQLSession.test.js
@@ -1,12 +1,12 @@
 const { expect, AssertionError } = require('chai');
-const { DBSQLLogger, LogLevel } = require('../../dist');
+const { DBSQLLogger, LogLevel } = require('../../lib');
 const sinon = require('sinon');
-const DBSQLSession = require('../../dist/DBSQLSession').default;
-const InfoValue = require('../../dist/dto/InfoValue').default;
-const Status = require('../../dist/dto/Status').default;
-const DBSQLOperation = require('../../dist/DBSQLOperation').default;
-const HiveDriver = require('../../dist/hive/HiveDriver').default;
-const DBSQLClient = require('../../dist/DBSQLClient').default;
+const DBSQLSession = require('../../lib/DBSQLSession').default;
+const InfoValue = require('../../lib/dto/InfoValue').default;
+const Status = require('../../lib/dto/Status').default;
+const DBSQLOperation = require('../../lib/DBSQLOperation').default;
+const HiveDriver = require('../../lib/hive/HiveDriver').default;
+const DBSQLClient = require('../../lib/DBSQLClient').default;
 
 // Create logger that won't emit
 //

diff --git a/tests/unit/connection/auth/DatabricksOAuth/AuthorizationCode.test.js b/tests/unit/connection/auth/DatabricksOAuth/AuthorizationCode.test.js
@@ -2,8 +2,8 @@ const { expect, AssertionError } = require('chai');
 const { EventEmitter } = require('events');
 const sinon = require('sinon');
 const http = require('http');
-const { DBSQLLogger, LogLevel } = require('../../../../../dist');
-const AuthorizationCode = require('../../../../../dist/connection/auth/DatabricksOAuth/AuthorizationCode').default;
+const { DBSQLLogger, LogLevel } = require('../../../../../lib');
+const AuthorizationCode = require('../../../../../lib/connection/auth/DatabricksOAuth/AuthorizationCode').default;
 
 const logger = new DBSQLLogger({ level: LogLevel.error });
 

diff --git a/tests/unit/connection/auth/DatabricksOAuth/OAuthManager.test.js b/tests/unit/connection/auth/DatabricksOAuth/OAuthManager.test.js
@@ -1,15 +1,15 @@
 const { expect, AssertionError } = require('chai');
 const sinon = require('sinon');
 const openidClientLib = require('openid-client');
-const { DBSQLLogger, LogLevel } = require('../../../../../dist');
+const { DBSQLLogger, LogLevel } = require('../../../../../lib');
 const {
   DatabricksOAuthManager,
   AzureOAuthManager,
   OAuthFlow,
-} = require('../../../../../dist/connection/auth/DatabricksOAuth/OAuthManager');
-const OAuthToken = require('../../../../../dist/connection/auth/DatabricksOAuth/OAuthToken').default;
-const { OAuthScope, scopeDelimiter } = require('../../../../../dist/connection/auth/DatabricksOAuth/OAuthScope');
-const AuthorizationCodeModule = require('../../../../../dist/connection/auth/DatabricksOAuth/AuthorizationCode');
+} = require('../../../../../lib/connection/auth/DatabricksOAuth/OAuthManager');
+const OAuthToken = require('../../../../../lib/connection/auth/DatabricksOAuth/OAuthToken').default;
+const { OAuthScope, scopeDelimiter } = require('../../../../../lib/connection/auth/DatabricksOAuth/OAuthScope');
+const AuthorizationCodeModule = require('../../../../../lib/connection/auth/DatabricksOAuth/AuthorizationCode');
 
 const { createValidAccessToken, createExpiredAccessToken } = require('./utils');
 

diff --git a/tests/unit/connection/auth/DatabricksOAuth/OAuthToken.test.js b/tests/unit/connection/auth/DatabricksOAuth/OAuthToken.test.js
@@ -1,5 +1,5 @@
 const { expect } = require('chai');
-const OAuthToken = require('../../../../../dist/connection/auth/DatabricksOAuth/OAuthToken').default;
+const OAuthToken = require('../../../../../lib/connection/auth/DatabricksOAuth/OAuthToken').default;
 
 const { createAccessToken } = require('./utils');
 

diff --git a/tests/unit/connection/auth/DatabricksOAuth/index.test.js b/tests/unit/connection/auth/DatabricksOAuth/index.test.js
@@ -1,8 +1,8 @@
 const { expect, AssertionError } = require('chai');
 const sinon = require('sinon');
-const DatabricksOAuth = require('../../../../../dist/connection/auth/DatabricksOAuth/index').default;
-const OAuthToken = require('../../../../../dist/connection/auth/DatabricksOAuth/OAuthToken').default;
-const OAuthManager = require('../../../../../dist/connection/auth/DatabricksOAuth/OAuthManager').default;
+const DatabricksOAuth = require('../../../../../lib/connection/auth/DatabricksOAuth/index').default;
+const OAuthToken = require('../../../../../lib/connection/auth/DatabricksOAuth/OAuthToken').default;
+const OAuthManager = require('../../../../../lib/connection/auth/DatabricksOAuth/OAuthManager').default;
 
 const { createValidAccessToken, createExpiredAccessToken } = require('./utils');
 

diff --git a/tests/unit/connection/auth/PlainHttpAuthentication.test.js b/tests/unit/connection/auth/PlainHttpAuthentication.test.js
@@ -1,5 +1,5 @@
 const { expect } = require('chai');
-const PlainHttpAuthentication = require('../../../../dist/connection/auth/PlainHttpAuthentication').default;
+const PlainHttpAuthentication = require('../../../../lib/connection/auth/PlainHttpAuthentication').default;
 
 describe('PlainHttpAuthentication', () => {
   it('username and password must be anonymous if nothing passed', () => {

diff --git a/tests/unit/connection/connections/HttpConnection.test.js b/tests/unit/connection/connections/HttpConnection.test.js
@@ -1,8 +1,8 @@
 const http = require('http');
 const { expect } = require('chai');
-const HttpConnection = require('../../../../dist/connection/connections/HttpConnection').default;
-const ThriftHttpConnection = require('../../../../dist/connection/connections/ThriftHttpConnection').default;
-const DBSQLClient = require('../../../../dist/DBSQLClient').default;
+const HttpConnection = require('../../../../lib/connection/connections/HttpConnection').default;
+const ThriftHttpConnection = require('../../../../lib/connection/connections/ThriftHttpConnection').default;
+const DBSQLClient = require('../../../../lib/DBSQLClient').default;
 
 describe('HttpConnection.connect', () => {
   it('should create Thrift connection', async () => {

diff --git a/tests/unit/connection/connections/HttpRetryPolicy.test.js b/tests/unit/connection/connections/HttpRetryPolicy.test.js
@@ -1,9 +1,9 @@
 const { expect, AssertionError } = require('chai');
 const sinon = require('sinon');
 const { Request, Response } = require('node-fetch');
-const HttpRetryPolicy = require('../../../../dist/connection/connections/HttpRetryPolicy').default;
-const { default: RetryError, RetryErrorCode } = require('../../../../dist/errors/RetryError');
-const DBSQLClient = require('../../../../dist/DBSQLClient').default;
+const HttpRetryPolicy = require('../../../../lib/connection/connections/HttpRetryPolicy').default;
+const { default: RetryError, RetryErrorCode } = require('../../../../lib/errors/RetryError');
+const DBSQLClient = require('../../../../lib/DBSQLClient').default;
 
 class ClientContextMock {
   constructor(configOverrides) {

diff --git a/tests/unit/connection/connections/NullRetryPolicy.test.js b/tests/unit/connection/connections/NullRetryPolicy.test.js
@@ -1,6 +1,6 @@
 const { expect, AssertionError } = require('chai');
 const sinon = require('sinon');
-const NullRetryPolicy = require('../../../../dist/connection/connections/NullRetryPolicy').default;
+const NullRetryPolicy = require('../../../../lib/connection/connections/NullRetryPolicy').default;
 
 describe('NullRetryPolicy', () => {
   it('should never allow retries', async () => {

diff --git a/tests/unit/dto/InfoValue.test.js b/tests/unit/dto/InfoValue.test.js
@@ -1,5 +1,5 @@
 const { expect } = require('chai');
-const InfoValue = require('../../../dist/dto/InfoValue').default;
+const InfoValue = require('../../../lib/dto/InfoValue').default;
 const NodeInt64 = require('node-int64');
 
 const createInfoValueMock = (value) =>

diff --git a/tests/unit/dto/Status.test.js b/tests/unit/dto/Status.test.js
@@ -1,6 +1,6 @@
 const { expect } = require('chai');
-const { TCLIService_types } = require('../../../').thrift;
-const Status = require('../../../dist/dto/Status').default;
+const { TCLIService_types } = require('../../../lib').thrift;
+const Status = require('../../../lib/dto/Status').default;
 
 describe('StatusFactory', () => {
   it('should be success', () => {

diff --git a/tests/unit/hive/HiveDriver.test.js b/tests/unit/hive/HiveDriver.test.js
@@ -1,7 +1,7 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
-const { TCLIService_types } = require('../../../').thrift;
-const HiveDriver = require('../../../dist/hive/HiveDriver').default;
+const { TCLIService_types } = require('../../../lib').thrift;
+const HiveDriver = require('../../../lib/hive/HiveDriver').default;
 
 const toTitleCase = (str) => str[0].toUpperCase() + str.slice(1);
 

diff --git a/tests/unit/hive/commands/BaseCommand.test.js b/tests/unit/hive/commands/BaseCommand.test.js
@@ -1,10 +1,10 @@
 const { expect, AssertionError } = require('chai');
 const { Request, Response } = require('node-fetch');
 const { Thrift } = require('thrift');
-const HiveDriverError = require('../../../../dist/errors/HiveDriverError').default;
-const BaseCommand = require('../../../../dist/hive/Commands/BaseCommand').default;
-const HttpRetryPolicy = require('../../../../dist/connection/connections/HttpRetryPolicy').default;
-const DBSQLClient = require('../../../../dist/DBSQLClient').default;
+const HiveDriverError = require('../../../../lib/errors/HiveDriverError').default;
+const BaseCommand = require('../../../../lib/hive/Commands/BaseCommand').default;
+const HttpRetryPolicy = require('../../../../lib/connection/connections/HttpRetryPolicy').default;
+const DBSQLClient = require('../../../../lib/DBSQLClient').default;
 
 class ThriftClientMock {
   constructor(context, methodHandler) {

diff --git a/tests/unit/hive/commands/CancelDelegationTokenCommand.test.js b/tests/unit/hive/commands/CancelDelegationTokenCommand.test.js
@@ -1,7 +1,7 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const TCLIService_types = require('../../../../thrift/TCLIService_types');
-const CancelDelegationTokenCommand = require('../../../../dist/hive/Commands/CancelDelegationTokenCommand').default;
+const CancelDelegationTokenCommand = require('../../../../lib/hive/Commands/CancelDelegationTokenCommand').default;
 
 const requestMock = {
   sessionHandle: {

diff --git a/tests/unit/hive/commands/CancelOperationCommand.test.js b/tests/unit/hive/commands/CancelOperationCommand.test.js
@@ -1,7 +1,7 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const TCLIService_types = require('../../../../thrift/TCLIService_types');
-const CancelOperationCommand = require('../../../../dist/hive/Commands/CancelOperationCommand').default;
+const CancelOperationCommand = require('../../../../lib/hive/Commands/CancelOperationCommand').default;
 
 const requestMock = {
   operationHandle: {

diff --git a/tests/unit/hive/commands/CloseOperationCommand.test.js b/tests/unit/hive/commands/CloseOperationCommand.test.js
@@ -1,7 +1,7 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const TCLIService_types = require('../../../../thrift/TCLIService_types');
-const CloseOperationCommand = require('../../../../dist/hive/Commands/CloseOperationCommand').default;
+const CloseOperationCommand = require('../../../../lib/hive/Commands/CloseOperationCommand').default;
 
 const requestMock = {
   operationHandle: {

diff --git a/tests/unit/hive/commands/CloseSessionCommand.test.js b/tests/unit/hive/commands/CloseSessionCommand.test.js
@@ -1,7 +1,7 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const TCLIService_types = require('../../../../thrift/TCLIService_types');
-const CloseSessionCommand = require('../../../../dist/hive/Commands/CloseSessionCommand').default;
+const CloseSessionCommand = require('../../../../lib/hive/Commands/CloseSessionCommand').default;
 
 const responseMock = {
   status: { statusCode: 0 },

diff --git a/tests/unit/hive/commands/ExecuteStatementCommand.test.js b/tests/unit/hive/commands/ExecuteStatementCommand.test.js
@@ -1,7 +1,7 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const TCLIService_types = require('../../../../thrift/TCLIService_types');
-const ExecuteStatementCommand = require('../../../../dist/hive/Commands/ExecuteStatementCommand').default;
+const ExecuteStatementCommand = require('../../../../lib/hive/Commands/ExecuteStatementCommand').default;
 
 const requestMock = {
   sessionHandle: {

diff --git a/tests/unit/hive/commands/FetchResultsCommand.test.js b/tests/unit/hive/commands/FetchResultsCommand.test.js
@@ -1,7 +1,7 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const TCLIService_types = require('../../../../thrift/TCLIService_types');
-const FetchResultsCommand = require('../../../../dist/hive/Commands/FetchResultsCommand').default;
+const FetchResultsCommand = require('../../../../lib/hive/Commands/FetchResultsCommand').default;
 
 const requestMock = {
   operationHandle: {

diff --git a/tests/unit/hive/commands/GetCatalogsCommand.test.js b/tests/unit/hive/commands/GetCatalogsCommand.test.js
@@ -1,7 +1,7 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const TCLIService_types = require('../../../../thrift/TCLIService_types');
-const GetCatalogsCommand = require('../../../../dist/hive/Commands/GetCatalogsCommand').default;
+const GetCatalogsCommand = require('../../../../lib/hive/Commands/GetCatalogsCommand').default;
 
 const requestMock = {
   sessionHandle: {

diff --git a/tests/unit/hive/commands/GetColumnsCommand.test.js b/tests/unit/hive/commands/GetColumnsCommand.test.js
@@ -1,7 +1,7 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const TCLIService_types = require('../../../../thrift/TCLIService_types');
-const GetColumnsCommand = require('../../../../dist/hive/Commands/GetColumnsCommand').default;
+const GetColumnsCommand = require('../../../../lib/hive/Commands/GetColumnsCommand').default;
 
 const requestMock = {
   sessionHandle: {