javatests/com/google/gerrit/acceptance/api/accounts/MessageIdGeneratorIT.java

// Copyright (C) 2020 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package com.google.gerrit.acceptance.api.accounts;

import static com.google.common.truth.Truth.assertThat;

import com.google.gerrit.acceptance.AbstractDaemonTest;
import com.google.gerrit.acceptance.PushOneCommit;
import com.google.gerrit.entities.Address;
import com.google.gerrit.entities.PatchSet;
import com.google.gerrit.entities.RefNames;
import com.google.gerrit.mail.MailMessage;
import com.google.gerrit.server.mail.send.MessageIdGenerator;
import com.google.gerrit.server.util.time.TimeUtil;
import java.time.Instant;
import javax.inject.Inject;
import org.eclipse.jgit.lib.Repository;
import org.junit.Test;

public class MessageIdGeneratorIT extends AbstractDaemonTest {
  @Inject private MessageIdGenerator messageIdGenerator;

  @Test
  public void fromAccountUpdate() throws Exception {
    try (Repository repo = repoManager.openRepository(allUsers)) {
      String messageId = messageIdGenerator.fromAccountUpdate(admin.id()).id();
      String sha1 =
          repo.getRefDatabase().findRef(RefNames.refsUsers(admin.id())).getObjectId().getName();
      assertThat(sha1).isEqualTo(messageId);
    }
  }

  @Test
  public void fromChangeUpdate() throws Exception {
    try (Repository repo = repoManager.openRepository(project)) {
      PushOneCommit.Result result = createChange();
      PatchSet.Id patchsetId = result.getChange().currentPatchSet().id();
      String messageId = messageIdGenerator.fromChangeUpdate(project, patchsetId).id();
      String sha1 =
          repo.getRefDatabase()
              .findRef(String.format("%smeta", patchsetId.changeId().toRefPrefix()))
              .getObjectId()
              .getName();
      assertThat(sha1).isEqualTo(messageId);
    }
  }

  @Test
  public void fromMailMessage() throws Exception {
    String id = "unique-id";
    MailMessage mailMessage =
        MailMessage.builder()
            .id(id)
            .from(Address.create("email@email.com"))
            .dateReceived(Instant.EPOCH)
            .subject("subject")
            .build();
    assertThat(messageIdGenerator.fromMailMessage(mailMessage).id()).isEqualTo(id + "-REJECTION");
  }

  @Test
  public void fromReasonAccountIdAndTimestamp() throws Exception {
    String reason = "reason";
    Instant timestamp = TimeUtil.now();
    assertThat(
            messageIdGenerator.fromReasonAccountIdAndTimestamp(reason, admin.id(), timestamp).id())
        .isEqualTo(reason + "-" + admin.id().toString() + "-" + timestamp.toString());
  }
}