Signal-Server/integration-tests/src/test/java/org/signal/integration/MessagingTest.java

/*
 * Copyright 2023 Signal Messenger, LLC
 * SPDX-License-Identifier: AGPL-3.0-only
 */

package org.signal.integration;

import static org.junit.jupiter.api.Assertions.assertArrayEquals;

import java.nio.charset.StandardCharsets;
import java.util.Base64;
import java.util.List;
import org.apache.commons.lang3.tuple.Pair;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.ValueSource;
import org.whispersystems.textsecuregcm.entities.IncomingMessage;
import org.whispersystems.textsecuregcm.entities.IncomingMessageList;
import org.whispersystems.textsecuregcm.entities.OutgoingMessageEntityList;
import org.whispersystems.textsecuregcm.entities.SendMessageResponse;
import org.whispersystems.textsecuregcm.storage.Device;

public class MessagingTest {

  @ParameterizedTest
  @ValueSource(booleans = {true, false})
  public void testSendMessageUnsealed(final boolean atomicAccountCreation) throws Exception {
    final TestUser userA;
    final TestUser userB;

    if (atomicAccountCreation) {
      userA = Operations.newRegisteredUserAtomic("+19995550102");
      userB = Operations.newRegisteredUserAtomic("+19995550103");
    } else {
      userA = Operations.newRegisteredUser("+19995550104");
      userB = Operations.newRegisteredUser("+19995550105");
    }

    try {
      final byte[] expectedContent = "Hello, World!".getBytes(StandardCharsets.UTF_8);
      final String contentBase64 = Base64.getEncoder().encodeToString(expectedContent);
      final IncomingMessage message = new IncomingMessage(1, Device.PRIMARY_ID, userB.registrationId(), contentBase64);
      final IncomingMessageList messages = new IncomingMessageList(List.of(message), false, true, System.currentTimeMillis());

      final Pair<Integer, SendMessageResponse> sendMessage = Operations
          .apiPut("/v1/messages/%s".formatted(userB.aciUuid().toString()), messages)
          .authorized(userA)
          .execute(SendMessageResponse.class);

      final Pair<Integer, OutgoingMessageEntityList> receiveMessages = Operations.apiGet("/v1/messages")
          .authorized(userB)
          .execute(OutgoingMessageEntityList.class);

      final byte[] actualContent = receiveMessages.getRight().messages().get(0).content();
      assertArrayEquals(expectedContent, actualContent);
    } finally {
      Operations.deleteUser(userA);
      Operations.deleteUser(userB);
    }
  }
}