|
1 | 1 | package io.getstream.client; |
2 | 2 |
|
3 | 3 | import io.getstream.core.LookupKind; |
4 | | -import io.getstream.core.models.Activity; |
5 | | -import io.getstream.core.models.FeedID; |
6 | | -import io.getstream.core.models.Paginated; |
7 | | -import io.getstream.core.models.Reaction; |
| 4 | +import io.getstream.core.models.*; |
| 5 | + |
8 | 6 | import java.util.List; |
| 7 | +import java.util.Map; |
| 8 | +import java.util.function.Function; |
| 9 | +import java.util.stream.Collectors; |
9 | 10 |
|
10 | 11 | import io.getstream.core.options.Filter; |
11 | 12 | import io.getstream.core.options.Limit; |
@@ -102,15 +103,19 @@ public void batchFetchReactions() throws Exception { |
102 | 103 | Reaction r5=client.reactions().add("user2", "comment", activity.getID()).join(); |
103 | 104 | Reaction r6=client.reactions().add("user3", "comment", activity.getID()).join(); |
104 | 105 |
|
105 | | - List<Reaction> result = client.reactions().getBatch(List.of(r1.getId(), r2.getId(), r3.getId(), r4.getId(), r5.getId(), r6.getId())).join(); |
106 | | - assertEquals(6, result.size()); |
| 106 | + Map<String, Reaction> reactionsRequest = Map.of(r1.getId(), r1, r2.getId(), r2, r3.getId(), r3, r4.getId(), r4, r5.getId(), r5, r6.getId(), r6); |
| 107 | + |
| 108 | + ReactionBatch response = client.reactions().getBatch(List.of(r1.getId(), r2.getId(), r3.getId(), r4.getId(), r5.getId(), r6.getId())).join(); |
| 109 | + List<Reaction> result = response.getReactions(); |
107 | 110 |
|
108 | | - assertEquals("like", result.get(0).getKind()); |
109 | | - assertEquals("comment", result.get(1).getKind()); |
110 | | - assertEquals("share", result.get(2).getKind()); |
111 | | - assertEquals("like", result.get(3).getKind()); |
112 | | - assertEquals("comment", result.get(4).getKind()); |
113 | | - assertEquals("comment", result.get(5).getKind()); |
| 111 | + //convert result to map and compare each id and type mapping from reactionsRequest to result |
| 112 | + Map<String, Reaction> resultMap = result.stream().collect(Collectors.toMap(Reaction::getId, Function.identity())); |
| 113 | + assertEquals(6, resultMap.size()); |
| 114 | + for (Reaction r : result) { |
| 115 | + Reaction req = reactionsRequest.get(r.getId()); |
| 116 | + assertEquals(req.getActivityID(), r.getActivityID()); |
| 117 | + assertEquals(req.getKind(), r.getKind()); |
| 118 | + } |
114 | 119 | } |
115 | 120 |
|
116 | 121 | @Test |
|
0 commit comments