From f685dc1bf7af46988b59035b041ecbf085bb2884 Mon Sep 17 00:00:00 2001
From: tiagosiebler <tiagosiebler@github.com>
Date: Fri, 8 Nov 2024 13:04:17 +0000
Subject: [PATCH] fix(): failing tests

---
 test/v5/private.read.test.ts  | 23 ++++++++++++++---------
 test/v5/private.write.test.ts | 24 +++++++++++++-----------
 2 files changed, 27 insertions(+), 20 deletions(-)

diff --git a/test/v5/private.read.test.ts b/test/v5/private.read.test.ts
index 7282081..b4c60f9 100644
--- a/test/v5/private.read.test.ts
+++ b/test/v5/private.read.test.ts
@@ -22,6 +22,7 @@ describe('Private READ V5 REST API Endpoints', () => {
 
   const settleCoin = 'USDT';
   const linearSymbol = 'BTCUSDT';
+  const accountType = 'UNIFIED';
 
   describe('misc endpoints', () => {
     it('fetchServerTime()', async () => {
@@ -84,21 +85,19 @@ describe('Private READ V5 REST API Endpoints', () => {
   describe('Account APIs', () => {
     it('getWalletBalance()', async () => {
       expect(
-        await api.getWalletBalance({ accountType: 'CONTRACT' }),
+        await api.getWalletBalance({ accountType: accountType }),
       ).toMatchObject({ ...successResponseObjectV3() });
     });
 
     it('getBorrowHistory()', async () => {
       expect(await api.getBorrowHistory()).toMatchObject({
         ...successResponseObjectV3(),
-        retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
       });
     });
 
     it('getCollateralInfo()', async () => {
       expect(await api.getCollateralInfo()).toMatchObject({
         ...successResponseObjectV3(),
-        retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
       });
     });
 
@@ -127,7 +126,7 @@ describe('Private READ V5 REST API Endpoints', () => {
     it('getTransactionLog()', async () => {
       expect(await api.getTransactionLog()).toMatchObject({
         ...successResponseObjectV3(),
-        retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
+        // retMsg: '',
       });
     });
 
@@ -151,7 +150,7 @@ describe('Private READ V5 REST API Endpoints', () => {
       expect(await api.getDeliveryRecord({ category: 'option' })).toMatchObject(
         {
           ...successResponseObjectV3(),
-          retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
+          // retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
         },
       );
     });
@@ -161,7 +160,7 @@ describe('Private READ V5 REST API Endpoints', () => {
         await api.getSettlementRecords({ category: 'linear' }),
       ).toMatchObject({
         ...successResponseObjectV3(),
-        retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
+        // retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
       });
     });
 
@@ -173,13 +172,19 @@ describe('Private READ V5 REST API Endpoints', () => {
 
     it('getAllCoinsBalance()', async () => {
       expect(
-        await api.getAllCoinsBalance({ accountType: 'SPOT' }),
-      ).toMatchObject({ ...successResponseObjectV3() });
+        await api.getAllCoinsBalance({ accountType: accountType }),
+      ).toMatchObject({
+        ...successResponseObjectV3(),
+        // retMsg: '',
+      });
     });
 
     it('getCoinBalance()', async () => {
       expect(
-        await api.getCoinBalance({ accountType: 'SPOT', coin: settleCoin }),
+        await api.getCoinBalance({
+          accountType: accountType,
+          coin: settleCoin,
+        }),
       ).toMatchObject({ ...successResponseObjectV3() });
     });
 
diff --git a/test/v5/private.write.test.ts b/test/v5/private.write.test.ts
index c228245..d115116 100644
--- a/test/v5/private.write.test.ts
+++ b/test/v5/private.write.test.ts
@@ -199,7 +199,8 @@ describe('Private WRITE V5 REST API Endpoints', () => {
       });
     });
 
-    it('switchIsolatedMargin()', async () => {
+    // Not compatible with account mode
+    it.skip('switchIsolatedMargin()', async () => {
       expect(
         await api.switchIsolatedMargin({
           category: 'linear',
@@ -211,7 +212,7 @@ describe('Private WRITE V5 REST API Endpoints', () => {
         }),
       ).toMatchObject({
         // ...successResponseObjectV3(),
-        // retMsg: '',
+        retMsg: '',
         retCode: API_ERROR_CODE.V5_CROSS_ISOLATED_MARGIN_NOT_CHANGED,
       });
     });
@@ -275,7 +276,8 @@ describe('Private WRITE V5 REST API Endpoints', () => {
       });
     });
 
-    it('setAutoAddMargin()', async () => {
+    // Not compatible with account mode
+    it.skip('setAutoAddMargin()', async () => {
       expect(
         await api.setAutoAddMargin({
           category: 'linear',
@@ -284,8 +286,8 @@ describe('Private WRITE V5 REST API Endpoints', () => {
           positionIdx: 0,
         }),
       ).toMatchObject({
-        // ...successResponseObjectV3(),
-        // retMsg: '',
+        ...successResponseObjectV3(),
+        retMsg: '',
         retCode: API_ERROR_CODE.V5_AUTO_ADD_MARGIN_NOT_CHANGED,
       });
     });
@@ -294,9 +296,9 @@ describe('Private WRITE V5 REST API Endpoints', () => {
   describe('Account APIs', () => {
     it('setMarginMode()', async () => {
       expect(await api.setMarginMode('REGULAR_MARGIN')).toMatchObject({
-        // ...successResponseObjectV3(),
+        ...successResponseObjectV3(),
         // retMsg: '',
-        retCode: API_ERROR_CODE.V5_MARGIN_MODE_NOT_CHANGED,
+        // retCode: API_ERROR_CODE.V5_MARGIN_MODE_NOT_CHANGED,
       });
     });
 
@@ -468,17 +470,17 @@ describe('Private WRITE V5 REST API Endpoints', () => {
   describe('Spot Margin APIs', () => {
     it('toggleSpotMarginTrade()', async () => {
       expect(await api.toggleSpotMarginTrade('1')).toMatchObject({
-        // ...successResponseObjectV3(),
+        ...successResponseObjectV3(),
         // retMsg: '',
-        retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
+        // retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
       });
     });
 
     it('setSpotMarginLeverage()', async () => {
       expect(await api.setSpotMarginLeverage('2')).toMatchObject({
-        // ...successResponseObjectV3(),
+        ...successResponseObjectV3(),
         // retMsg: '',
-        retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
+        // retCode: API_ERROR_CODE.INCORRECT_API_KEY_PERMISSIONS,
       });
     });
   });