fix: update so follow and followedBy returns data

juni-b-queer · Oct 18, 2024 · 1d844d3 · 1d844d3
1 parent 83141b4
commit 1d844d3
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 23 deletions.
diff --git a/src/agent/HandlerAgent.ts b/src/agent/HandlerAgent.ts
@@ -113,7 +113,7 @@ export class HandlerAgent {
             limit: limit,
         };
         const resp = await this.agent?.getFollows(body);
-        return resp?.data.follows;
+        return resp?.data;
     }
 
     /**
@@ -133,7 +133,7 @@ export class HandlerAgent {
             limit: limit,
         };
         const resp = await this.agent?.getFollowers(body);
-        return resp?.data.followers;
+        return resp?.data;
     }
 
     /**

diff --git a/tests/agent/HandlerAgentFollow.test.ts b/tests/agent/HandlerAgentFollow.test.ts
@@ -9,37 +9,37 @@ describe('HandlerAgent', () => {
     const testHandle: string = 'testhandle';
     const testPassword: string = 'testpassword';
 
-    const followingMocks = [
-        {
-            did: 'isFollowing',
-            viewer: {
-                following: 'followLink',
+    const followingMocks = {
+        follows: [
+            {
+                did: 'isFollowing',
+                viewer: {
+                    following: 'followLink',
+                },
             },
-        },
-    ];
+        ],
+    };
 
-    const followedByMocks = [
-        {
-            did: 'isFollowedBy',
-            viewer: {
-                following: 'followLink',
-                followedBy: 'followedByLink',
+    const followedByMocks = {
+        followers: [
+            {
+                did: 'isFollowedBy',
+                viewer: {
+                    following: 'followLink',
+                    followedBy: 'followedByLink',
+                },
             },
-        },
-    ];
+        ],
+    };
     let getFollowsMock: jest.Mock<any, any, any>;
     let getFollowersMock: jest.Mock<any, any, any>;
     const followMock = jest.fn();
     const deleteFollowMock = jest.fn();
     let getProfileMock: jest.Mock<any, any, any>;
     beforeEach(() => {
         jest.clearAllMocks();
-        getFollowsMock = jest
-            .fn()
-            .mockReturnValue({ data: { follows: followingMocks } });
-        getFollowersMock = jest
-            .fn()
-            .mockReturnValue({ data: { followers: followedByMocks } });
+        getFollowsMock = jest.fn().mockReturnValue({ data: followingMocks });
+        getFollowersMock = jest.fn().mockReturnValue({ data: followedByMocks });
 
         // Require mocked module and define class' methods
         const mockedAgent = {