fix api kit tests

safe-global · Jan 23, 2025 · b95c3e9 · b95c3e9
1 parent d2caa9c
commit b95c3e9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/packages/api-kit/tests/e2e/addSafeOperation.test.ts b/packages/api-kit/tests/e2e/addSafeOperation.test.ts
@@ -6,7 +6,7 @@ import Safe from '@safe-global/protocol-kit'
 import SafeApiKit from '@safe-global/api-kit/index'
 import { getAddSafeOperationProps } from '@safe-global/api-kit/utils/safeOperation'
 import { BundlerClient, Safe4337Pack } from '@safe-global/relay-kit'
-import * as utils from '@safe-global/relay-kit/src/packs/safe-4337/utils'
+import * as relayKit from '@safe-global/relay-kit'
 import {
   generateTransferCallData,
   ENTRYPOINT_ABI,
@@ -40,7 +40,7 @@ describe('addSafeOperation', () => {
   const requestStub = sinon.stub()
   // Setup mocks for the bundler client
   before(async () => {
-    sinon.stub(utils, 'getEip4337BundlerProvider').returns({
+    sinon.stub(relayKit, 'getEip4337BundlerProvider').returns({
       request: requestStub,
       readContract: sinon
         .stub()

diff --git a/packages/api-kit/tests/e2e/confirmSafeOperation.test.ts b/packages/api-kit/tests/e2e/confirmSafeOperation.test.ts
@@ -8,7 +8,7 @@ import {
   ENTRYPOINT_ADDRESS_V06,
   RPC_4337_CALLS
 } from '@safe-global/relay-kit/test-utils'
-import * as utils from '@safe-global/relay-kit/src/packs/safe-4337/utils'
+import * as relayKit from '@safe-global/relay-kit'
 import SafeApiKit from '@safe-global/api-kit/index'
 import { getAddSafeOperationProps } from '@safe-global/api-kit/utils/safeOperation'
 import { SafeOperation } from '@safe-global/types-kit'
@@ -72,7 +72,7 @@ describe('confirmSafeOperation', () => {
   const requestStub = sinon.stub()
 
   before(async () => {
-    sinon.stub(utils, 'getEip4337BundlerProvider').returns({
+    sinon.stub(relayKit, 'getEip4337BundlerProvider').returns({
       request: requestStub
     } as unknown as BundlerClient)
 

diff --git a/packages/relay-kit/src/index.ts b/packages/relay-kit/src/index.ts
@@ -8,6 +8,7 @@ export { default as EthSafeOperation } from './packs/safe-4337/SafeOperation'
 
 export * from './packs/safe-4337/estimators'
 export * from './packs/safe-4337/types'
+export * from './packs/safe-4337/utils'
 
 export * from './RelayKitBasePack'