// 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.PatchSet;
import com.google.gerrit.entities.RefNames;
import com.google.gerrit.mail.Address;
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());
  }
}
