First pass at adding comment trees. (#2362)

[lemmy.git] / api_tests / src / follow.spec.ts

diff --git a/api_tests/src/follow.spec.ts b/api_tests/src/follow.spec.ts
index 8917efc02414b2835374412bf691365a044a6e8f..75ec9f29f8c0e05292417af8ba005ecbd9ae2510 100644 (file)
--- a/api_tests/src/follow.spec.ts
+++ b/api_tests/src/follow.spec.ts
@@ -1,13 +1,13 @@
 jest.setTimeout(120000);
+import {SubscribedType} from 'lemmy-js-client';
 import {
   alpha,
   setupLogins,
-  searchForBetaCommunity,
+  resolveBetaCommunity,
   followCommunity,
-  checkFollowedCommunities,
   unfollowRemotes,
+  getSite,
   delay,
-  longDelay,
 } from './shared';
 
 beforeAll(async () => {
@@ -19,32 +19,31 @@ afterAll(async () => {
 });
 
 test('Follow federated community', async () => {
-  let search = await searchForBetaCommunity(alpha); // TODO sometimes this is returning null?
+  let betaCommunity = (await resolveBetaCommunity(alpha)).community.unwrap();
   let follow = await followCommunity(
     alpha,
     true,
-    search.communities[0].community.id
+    betaCommunity.community.id
   );
 
   // Make sure the follow response went through
   expect(follow.community_view.community.local).toBe(false);
   expect(follow.community_view.community.name).toBe('main');
-  await longDelay();
+  expect(follow.community_view.subscribed).toBe(SubscribedType.Subscribed);
 
   // Check it from local
-  let followCheck = await checkFollowedCommunities(alpha);
-  await delay();
-  let remoteCommunityId = followCheck.communities.find(
+  let site = await getSite(alpha);
+  let remoteCommunityId = site.my_user.unwrap().follows.find(
     c => c.community.local == false
   ).community.id;
   expect(remoteCommunityId).toBeDefined();
+  expect(site.my_user.unwrap().follows.length).toBe(1);
 
   // Test an unfollow
   let unfollow = await followCommunity(alpha, false, remoteCommunityId);
-  expect(unfollow.community_view.community.local).toBe(false);
-  await delay();
+  expect(unfollow.community_view.subscribed).toBe(SubscribedType.NotSubscribed);
 
   // Make sure you are unsubbed locally
-  let unfollowCheck = await checkFollowedCommunities(alpha);
-  expect(unfollowCheck.communities.length).toBeGreaterThanOrEqual(1);
+  let siteUnfollowCheck = await getSite(alpha);
+  expect(siteUnfollowCheck.my_user.unwrap().follows.length).toBe(0);
 });