|
21 | 21 | import club.minnced.discord.webhook.send.WebhookEmbedBuilder;
|
22 | 22 | import club.minnced.discord.webhook.send.WebhookMessage;
|
23 | 23 | import club.minnced.discord.webhook.send.WebhookMessageBuilder;
|
| 24 | +import net.dv8tion.jda.api.EmbedBuilder; |
| 25 | +import net.dv8tion.jda.api.MessageBuilder; |
| 26 | +import net.dv8tion.jda.api.entities.Message; |
| 27 | +import net.dv8tion.jda.api.entities.MessageEmbed; |
| 28 | +import net.dv8tion.jda.internal.entities.DataMessage; |
| 29 | +import net.dv8tion.jda.internal.entities.ReceivedMessage; |
24 | 30 | import okhttp3.RequestBody;
|
25 | 31 | import org.json.JSONArray;
|
26 | 32 | import org.json.JSONObject;
|
|
38 | 44 | import java.util.List;
|
39 | 45 | import java.util.Map;
|
40 | 46 |
|
| 47 | +import static org.mockito.Mockito.CALLS_REAL_METHODS; |
| 48 | +import static org.powermock.api.mockito.PowerMockito.mock; |
| 49 | +import static org.powermock.api.mockito.PowerMockito.when; |
| 50 | + |
41 | 51 | public class MessageTest {
|
42 | 52 | private WebhookMessageBuilder builder;
|
43 | 53 |
|
@@ -127,6 +137,55 @@ public void buildMessageWithFiles() throws IOException {
|
127 | 137 | Assert.assertEquals("dog.png", message.getAttachments()[1].getName());
|
128 | 138 | Assert.assertEquals("bird.png", message.getAttachments()[2].getName());
|
129 | 139 | }
|
| 140 | + |
| 141 | + @Test |
| 142 | + public void buildMessageWithDataMessage() { |
| 143 | + MessageEmbed jdaEmbed = new EmbedBuilder() |
| 144 | + .setTitle("myEmbed") |
| 145 | + .build(); |
| 146 | + |
| 147 | + Message jdaMessage = new MessageBuilder() |
| 148 | + .setTTS(true) |
| 149 | + .setContent("myContent") |
| 150 | + .setEmbeds(jdaEmbed) |
| 151 | + .build(); |
| 152 | + |
| 153 | + WebhookMessage webhookMessage = WebhookMessageBuilder.fromJDA(jdaMessage).build(); |
| 154 | + List<WebhookEmbed> webhookEmbeds = webhookMessage.getEmbeds(); |
| 155 | + |
| 156 | + Assert.assertEquals(webhookEmbeds.size(), 1); |
| 157 | + |
| 158 | + WebhookEmbed webhookEmbed = webhookEmbeds.get(0); |
| 159 | + |
| 160 | + Assert.assertTrue(jdaMessage instanceof DataMessage); |
| 161 | + Assert.assertTrue(webhookMessage.isTTS()); |
| 162 | + Assert.assertEquals(webhookMessage.getContent(), "myContent"); |
| 163 | + Assert.assertEquals(webhookEmbed.getTitle().getText(), "myEmbed"); |
| 164 | + } |
| 165 | + |
| 166 | + @Test |
| 167 | + public void buildMessageWithReceivedMessage() { |
| 168 | + MessageEmbed jdaEmbed = new EmbedBuilder() |
| 169 | + .setTitle("myEmbed") |
| 170 | + .build(); |
| 171 | + |
| 172 | + Message jdaMessage = mock(ReceivedMessage.class, CALLS_REAL_METHODS); |
| 173 | + when(jdaMessage.isTTS()).thenReturn(true); |
| 174 | + when(jdaMessage.getContentRaw()).thenReturn("myContent"); |
| 175 | + when(jdaMessage.getEmbeds()).thenReturn(Arrays.asList(jdaEmbed)); |
| 176 | + |
| 177 | + WebhookMessage webhookMessage = WebhookMessageBuilder.fromJDA(jdaMessage).build(); |
| 178 | + List<WebhookEmbed> webhookEmbeds = webhookMessage.getEmbeds(); |
| 179 | + |
| 180 | + Assert.assertEquals(webhookEmbeds.size(), 1); |
| 181 | + |
| 182 | + WebhookEmbed webhookEmbed = webhookEmbeds.get(0); |
| 183 | + |
| 184 | + Assert.assertTrue(jdaMessage instanceof ReceivedMessage); |
| 185 | + Assert.assertTrue(webhookMessage.isTTS()); |
| 186 | + Assert.assertEquals(webhookMessage.getContent(), "myContent"); |
| 187 | + Assert.assertEquals(webhookEmbed.getTitle().getText(), "myEmbed"); |
| 188 | + } |
130 | 189 |
|
131 | 190 | @Test
|
132 | 191 | public void factoryEmbeds() {
|
|
0 commit comments