javatests/com/google/gerrit/server/restapi/change/ListChangeCommentsTest.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.server.restapi.change;

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

import com.google.common.collect.ImmutableList;
import com.google.gerrit.entities.Change;
import com.google.gerrit.entities.ChangeMessage;
import com.google.gerrit.extensions.common.CommentInfo;
import com.google.gerrit.server.CommentsUtil;
import java.sql.Timestamp;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

@RunWith(JUnit4.class)
public class ListChangeCommentsTest {

  @Test
  public void commentsLinkedToChangeMessages() {
    CommentInfo c1 = getNewCommentInfo("c1", Timestamp.valueOf("2018-01-01 09:01:00"));
    CommentInfo c2 = getNewCommentInfo("c2", Timestamp.valueOf("2018-01-01 09:01:15"));
    CommentInfo c3 = getNewCommentInfo("c3", Timestamp.valueOf("2018-01-01 09:01:25"));

    ChangeMessage cm1 =
        getNewChangeMessage("cm1key", "cm1", Timestamp.valueOf("2018-01-01 00:00:00"));
    ChangeMessage cm2 =
        getNewChangeMessage("cm2key", "cm2", Timestamp.valueOf("2018-01-01 09:01:15"));
    ChangeMessage cm3 =
        getNewChangeMessage("cm3key", "cm3", Timestamp.valueOf("2018-01-01 09:01:27"));

    assertThat(c1.changeMessageId).isNull();
    assertThat(c2.changeMessageId).isNull();
    assertThat(c3.changeMessageId).isNull();

    ImmutableList<CommentInfo> comments = ImmutableList.of(c1, c2, c3);
    ImmutableList<ChangeMessage> changeMessages = ImmutableList.of(cm1, cm2, cm3);

    CommentsUtil.linkCommentsToChangeMessages(comments, changeMessages);

    assertThat(c1.changeMessageId).isEqualTo(changeMessageKey(cm2));
    assertThat(c2.changeMessageId).isEqualTo(changeMessageKey(cm2));
    assertThat(c3.changeMessageId).isEqualTo(changeMessageKey(cm3));
  }

  private static CommentInfo getNewCommentInfo(String message, Timestamp ts) {
    CommentInfo c = new CommentInfo();
    c.message = message;
    c.updated = ts;
    return c;
  }

  private static ChangeMessage getNewChangeMessage(String id, String message, Timestamp ts) {
    ChangeMessage.Key key = ChangeMessage.key(Change.id(1), id);
    ChangeMessage cm = new ChangeMessage(key, null, ts, null);
    cm.setMessage(message);
    return cm;
  }

  private static String changeMessageKey(ChangeMessage changeMessage) {
    return changeMessage.getKey().uuid();
  }
}