@@ -2,48 +2,46 @@ import { mockServiceContext, setupDatabricksEnv } from "@tools/test-helpers";
22import { afterEach , beforeEach , describe , expect , test , vi } from "vitest" ;
33import { ServiceContext } from "../../../context/service-context" ;
44import { FilesPlugin , files } from "../plugin" ;
5- import { streamFromString } from "./utils" ;
6-
7- const { mockFilesApi, mockClient, MockApiError, mockCacheInstance } =
8- vi . hoisted ( ( ) => {
9- const mockFilesApi = {
10- listDirectoryContents : vi . fn ( ) ,
11- download : vi . fn ( ) ,
12- getMetadata : vi . fn ( ) ,
13- upload : vi . fn ( ) ,
14- createDirectory : vi . fn ( ) ,
15- delete : vi . fn ( ) ,
16- } ;
17-
18- const mockClient = {
19- files : mockFilesApi ,
20- config : {
21- host : "https://test.databricks.com" ,
22- authenticate : vi . fn ( ) ,
23- } ,
24- } ;
25-
26- class MockApiError extends Error {
27- statusCode : number ;
28- constructor ( message : string , statusCode : number ) {
29- super ( message ) ;
30- this . name = "ApiError" ;
31- this . statusCode = statusCode ;
32- }
33- }
34-
35- const mockCacheInstance = {
36- get : vi . fn ( ) ,
37- set : vi . fn ( ) ,
38- delete : vi . fn ( ) ,
39- getOrExecute : vi . fn ( async ( _key : unknown [ ] , fn : ( ) => Promise < unknown > ) =>
40- fn ( ) ,
41- ) ,
42- generateKey : vi . fn ( ) ,
43- } ;
445
45- return { mockFilesApi, mockClient, MockApiError, mockCacheInstance } ;
46- } ) ;
6+ const { mockClient, MockApiError, mockCacheInstance } = vi . hoisted ( ( ) => {
7+ const mockFilesApi = {
8+ listDirectoryContents : vi . fn ( ) ,
9+ download : vi . fn ( ) ,
10+ getMetadata : vi . fn ( ) ,
11+ upload : vi . fn ( ) ,
12+ createDirectory : vi . fn ( ) ,
13+ delete : vi . fn ( ) ,
14+ } ;
15+
16+ const mockClient = {
17+ files : mockFilesApi ,
18+ config : {
19+ host : "https://test.databricks.com" ,
20+ authenticate : vi . fn ( ) ,
21+ } ,
22+ } ;
23+
24+ class MockApiError extends Error {
25+ statusCode : number ;
26+ constructor ( message : string , statusCode : number ) {
27+ super ( message ) ;
28+ this . name = "ApiError" ;
29+ this . statusCode = statusCode ;
30+ }
31+ }
32+
33+ const mockCacheInstance = {
34+ get : vi . fn ( ) ,
35+ set : vi . fn ( ) ,
36+ delete : vi . fn ( ) ,
37+ getOrExecute : vi . fn ( async ( _key : unknown [ ] , fn : ( ) => Promise < unknown > ) =>
38+ fn ( ) ,
39+ ) ,
40+ generateKey : vi . fn ( ) ,
41+ } ;
42+
43+ return { mockFilesApi, mockClient, MockApiError, mockCacheInstance } ;
44+ } ) ;
4745
4846vi . mock ( "@databricks/sdk-experimental" , ( ) => ( {
4947 WorkspaceClient : vi . fn ( ( ) => mockClient ) ,
0 commit comments