Skip to content

Commit dc981e3

Browse files
committed
fix and clean up tests
1 parent 9629250 commit dc981e3

File tree

1 file changed

+8
-11
lines changed

1 file changed

+8
-11
lines changed

libs/wire-subsystems/test/unit/Wire/MockInterpreters/UserGroupStore.hs

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -97,20 +97,17 @@ createUserGroupImpl tid nug managedBy = do
9797
pure ug
9898

9999
getUserGroupImpl :: (UserGroupStoreInMemEffectConstraints r) => TeamId -> UserGroupId -> Bool -> Sem r (Maybe UserGroup)
100-
getUserGroupImpl tid gid includeChannels = fmap filterChannels . Map.lookup (tid, gid) <$> get @UserGroupInMemState
101-
where
102-
filterChannels ug =
103-
if includeChannels
104-
then ug
105-
else (ug :: UserGroup) {channels = mempty}
100+
getUserGroupImpl tid gid includeChannels = fmap (filterChannels includeChannels) . Map.lookup (tid, gid) <$> get @UserGroupInMemState
101+
102+
filterChannels :: Bool -> UserGroup -> UserGroup
103+
filterChannels includeChannels ug =
104+
if includeChannels
105+
then (ug :: UserGroup) {channelsCount = Just $ maybe 0 length ug.channels}
106+
else (ug :: UserGroup) {channels = mempty}
106107

107108
getUserGroupsImpl :: (UserGroupStoreInMemEffectConstraints r) => UserGroupPageRequest -> Sem r UserGroupPage
108109
getUserGroupsImpl UserGroupPageRequest {..} = do
109-
let filterChannels ug =
110-
if includeChannels
111-
then (ug :: UserGroup) {channels = mempty, channelsCount = Just $ maybe 0 length ug.channels}
112-
else (ug :: UserGroup) {channels = mempty}
113-
meta <- ((snd <$>) . sieve . fmap (_2 %~ userGroupToMeta . filterChannels) . Map.toList) <$> get @UserGroupInMemState
110+
meta <- ((snd <$>) . sieve . fmap (_2 %~ userGroupToMeta . (filterChannels includeChannels)) . Map.toList) <$> get @UserGroupInMemState
114111
pure $ UserGroupPage meta (length meta)
115112
where
116113
sieve,

0 commit comments

Comments
 (0)