blob: efbaed65a47fb58d00f806f9b2e0fd98f58061b6 [file] [log] [blame]
// Copyright (C) 2018 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.notedb;
import static com.google.common.truth.Truth.assertThat;
import static com.google.common.truth.extensions.proto.ProtoTruth.assertThat;
import static com.google.gerrit.proto.testing.SerializedClassSubject.assertThatSerializedClass;
import static com.google.gerrit.server.notedb.ChangeNotesState.Serializer.toByteString;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableListMultimap;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.ImmutableTable;
import com.google.common.collect.Iterables;
import com.google.gerrit.common.data.SubmitRecord;
import com.google.gerrit.common.data.SubmitRequirement;
import com.google.gerrit.entities.Account;
import com.google.gerrit.entities.AttentionSetUpdate;
import com.google.gerrit.entities.Change;
import com.google.gerrit.entities.ChangeMessage;
import com.google.gerrit.entities.Comment;
import com.google.gerrit.entities.LabelId;
import com.google.gerrit.entities.PatchSet;
import com.google.gerrit.entities.PatchSetApproval;
import com.google.gerrit.entities.converter.ChangeMessageProtoConverter;
import com.google.gerrit.entities.converter.PatchSetApprovalProtoConverter;
import com.google.gerrit.entities.converter.PatchSetProtoConverter;
import com.google.gerrit.mail.Address;
import com.google.gerrit.server.AssigneeStatusUpdate;
import com.google.gerrit.server.ReviewerByEmailSet;
import com.google.gerrit.server.ReviewerSet;
import com.google.gerrit.server.ReviewerStatusUpdate;
import com.google.gerrit.server.cache.proto.Cache.ChangeNotesStateProto;
import com.google.gerrit.server.cache.proto.Cache.ChangeNotesStateProto.AssigneeStatusUpdateProto;
import com.google.gerrit.server.cache.proto.Cache.ChangeNotesStateProto.AttentionSetUpdateProto;
import com.google.gerrit.server.cache.proto.Cache.ChangeNotesStateProto.ChangeColumnsProto;
import com.google.gerrit.server.cache.proto.Cache.ChangeNotesStateProto.ReviewerByEmailSetEntryProto;
import com.google.gerrit.server.cache.proto.Cache.ChangeNotesStateProto.ReviewerSetEntryProto;
import com.google.gerrit.server.cache.proto.Cache.ChangeNotesStateProto.ReviewerStatusUpdateProto;
import com.google.gerrit.server.cache.serialize.ObjectIdConverter;
import com.google.gerrit.server.notedb.ChangeNotesState.ChangeColumns;
import com.google.gerrit.server.notedb.ChangeNotesState.Serializer;
import com.google.inject.TypeLiteral;
import com.google.protobuf.ByteString;
import java.lang.reflect.Type;
import java.sql.Timestamp;
import java.time.Instant;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.UUID;
import org.eclipse.jgit.lib.ObjectId;
import org.junit.Before;
import org.junit.Test;
public class ChangeNotesStateTest {
private static final Change.Id ID = Change.id(123);
private static final ObjectId SHA =
ObjectId.fromString("1234567812345678123456781234567812345678");
private static final ByteString SHA_BYTES = ObjectIdConverter.create().toByteString(SHA);
private static final String CHANGE_KEY = "Iabcdabcdabcdabcdabcdabcdabcdabcdabcdabcd";
private static final String DEFAULT_SERVER_ID = UUID.randomUUID().toString();
private ChangeColumns cols;
private ChangeColumnsProto colsProto;
@Before
public void setUp() throws Exception {
cols =
ChangeColumns.builder()
.changeKey(Change.key(CHANGE_KEY))
.createdOn(new Timestamp(123456L))
.lastUpdatedOn(new Timestamp(234567L))
.owner(Account.id(1000))
.branch("refs/heads/master")
.subject("Test change")
.isPrivate(false)
.workInProgress(false)
.reviewStarted(true)
.build();
colsProto = toProto(newBuilder().build()).getColumns();
}
private ChangeNotesState.Builder newBuilder() {
return ChangeNotesState.Builder.empty(ID).metaId(SHA).columns(cols);
}
private ChangeNotesStateProto.Builder newProtoBuilder() {
return ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto);
}
@Test
public void serializeChangeKey() throws Exception {
assertRoundTrip(
newBuilder()
.columns(
cols.toBuilder()
.changeKey(Change.key("Ieeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"))
.build())
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(
colsProto.toBuilder().setChangeKey("Ieeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"))
.build());
}
@Test
public void serializeCreatedOn() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().createdOn(new Timestamp(98765L)).build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setCreatedOnMillis(98765L))
.build());
}
@Test
public void serializeLastUpdatedOn() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().lastUpdatedOn(new Timestamp(98765L)).build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setLastUpdatedOnMillis(98765L))
.build());
}
@Test
public void serializeOwner() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().owner(Account.id(7777)).build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setOwner(7777))
.build());
}
@Test
public void serializeBranch() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().branch("refs/heads/bar").build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setBranch("refs/heads/bar"))
.build());
}
@Test
public void serializeSubject() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().subject("A different test change").build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setSubject("A different test change"))
.build());
}
@Test
public void serializeCurrentPatchSetId() throws Exception {
assertRoundTrip(
newBuilder()
.columns(cols.toBuilder().currentPatchSetId(PatchSet.id(ID, 2)).build())
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setCurrentPatchSetId(2).setHasCurrentPatchSetId(true))
.build());
}
@Test
public void serializeNullTopic() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().topic(null).build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.build());
}
@Test
public void serializeEmptyTopic() throws Exception {
ChangeNotesState state = newBuilder().columns(cols.toBuilder().topic("").build()).build();
assertRoundTrip(
state,
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setTopic("").setHasTopic(true))
.build());
}
@Test
public void serializeNonEmptyTopic() throws Exception {
ChangeNotesState state = newBuilder().columns(cols.toBuilder().topic("topic").build()).build();
assertRoundTrip(
state,
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setTopic("topic").setHasTopic(true))
.build());
}
@Test
public void serializeOriginalSubject() throws Exception {
assertRoundTrip(
newBuilder()
.columns(cols.toBuilder().originalSubject("The first patch set").build())
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(
colsProto
.toBuilder()
.setOriginalSubject("The first patch set")
.setHasOriginalSubject(true))
.build());
}
@Test
public void serializeSubmissionId() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().submissionId("xyz").build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setSubmissionId("xyz").setHasSubmissionId(true))
.build());
}
@Test
public void serializeStatus() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().status(Change.Status.MERGED).build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setStatus("MERGED").setHasStatus(true))
.build());
}
@Test
public void serializeIsPrivate() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().isPrivate(true).build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setIsPrivate(true))
.build());
}
@Test
public void serializeIsWorkInProgress() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().workInProgress(true).build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setWorkInProgress(true))
.build());
}
@Test
public void serializeHasReviewStarted() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().reviewStarted(true).build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setReviewStarted(true))
.build());
}
@Test
public void serializeRevertOf() throws Exception {
assertRoundTrip(
newBuilder().columns(cols.toBuilder().revertOf(Change.id(999)).build()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto.toBuilder().setRevertOf(999).setHasRevertOf(true))
.build());
}
@Test
public void serializeHashtags() throws Exception {
assertRoundTrip(
newBuilder().hashtags(ImmutableSet.of("tag2", "tag1")).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addHashtag("tag2")
.addHashtag("tag1")
.build());
}
@Test
public void serializePatchSets() throws Exception {
PatchSet ps1 =
PatchSet.builder()
.id(PatchSet.id(ID, 1))
.commitId(ObjectId.fromString("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"))
.uploader(Account.id(2000))
.createdOn(cols.createdOn())
.build();
ByteString ps1Bytes = toByteString(ps1, PatchSetProtoConverter.INSTANCE);
assertThat(ps1Bytes.size()).isEqualTo(66);
PatchSet ps2 =
PatchSet.builder()
.id(PatchSet.id(ID, 2))
.commitId(ObjectId.fromString("bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"))
.uploader(Account.id(3000))
.createdOn(cols.lastUpdatedOn())
.build();
ByteString ps2Bytes = toByteString(ps2, PatchSetProtoConverter.INSTANCE);
assertThat(ps2Bytes.size()).isEqualTo(66);
assertThat(ps2Bytes).isNotEqualTo(ps1Bytes);
assertRoundTrip(
newBuilder().patchSets(ImmutableMap.of(ps2.id(), ps2, ps1.id(), ps1).entrySet()).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addPatchSet(ps2Bytes)
.addPatchSet(ps1Bytes)
.build());
}
@Test
public void serializeApprovals() throws Exception {
PatchSetApproval a1 =
PatchSetApproval.builder()
.key(
PatchSetApproval.key(
PatchSet.id(ID, 1), Account.id(2001), LabelId.create("Code-Review")))
.value(1)
.granted(new Timestamp(1212L))
.build();
ByteString a1Bytes = toByteString(a1, PatchSetApprovalProtoConverter.INSTANCE);
assertThat(a1Bytes.size()).isEqualTo(43);
PatchSetApproval a2 =
PatchSetApproval.builder()
.key(
PatchSetApproval.key(
PatchSet.id(ID, 1), Account.id(2002), LabelId.create("Verified")))
.value(-1)
.granted(new Timestamp(3434L))
.build();
ByteString a2Bytes = toByteString(a2, PatchSetApprovalProtoConverter.INSTANCE);
assertThat(a2Bytes.size()).isEqualTo(49);
assertThat(a2Bytes).isNotEqualTo(a1Bytes);
assertRoundTrip(
newBuilder()
.approvals(ImmutableListMultimap.of(a2.patchSetId(), a2, a1.patchSetId(), a1).entries())
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addApproval(a2Bytes)
.addApproval(a1Bytes)
.build());
}
@Test
public void serializeReviewers() throws Exception {
assertRoundTrip(
newBuilder()
.reviewers(
ReviewerSet.fromTable(
ImmutableTable.<ReviewerStateInternal, Account.Id, Timestamp>builder()
.put(ReviewerStateInternal.CC, Account.id(2001), new Timestamp(1212L))
.put(ReviewerStateInternal.REVIEWER, Account.id(2002), new Timestamp(3434L))
.build()))
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addReviewer(
ReviewerSetEntryProto.newBuilder()
.setState("CC")
.setAccountId(2001)
.setTimestampMillis(1212L))
.addReviewer(
ReviewerSetEntryProto.newBuilder()
.setState("REVIEWER")
.setAccountId(2002)
.setTimestampMillis(3434L))
.build());
}
@Test
public void serializeReviewersByEmail() throws Exception {
assertRoundTrip(
newBuilder()
.reviewersByEmail(
ReviewerByEmailSet.fromTable(
ImmutableTable.<ReviewerStateInternal, Address, Timestamp>builder()
.put(
ReviewerStateInternal.CC,
Address.create("Name1", "email1@example.com"),
new Timestamp(1212L))
.put(
ReviewerStateInternal.REVIEWER,
Address.create("Name2", "email2@example.com"),
new Timestamp(3434L))
.build()))
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addReviewerByEmail(
ReviewerByEmailSetEntryProto.newBuilder()
.setState("CC")
.setAddress("Name1 <email1@example.com>")
.setTimestampMillis(1212L))
.addReviewerByEmail(
ReviewerByEmailSetEntryProto.newBuilder()
.setState("REVIEWER")
.setAddress("Name2 <email2@example.com>")
.setTimestampMillis(3434L))
.build());
}
@Test
public void serializeReviewersByEmailWithNullName() throws Exception {
ChangeNotesState actual =
assertRoundTrip(
newBuilder()
.reviewersByEmail(
ReviewerByEmailSet.fromTable(
ImmutableTable.of(
ReviewerStateInternal.CC,
Address.create("emailonly@example.com"),
new Timestamp(1212L))))
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addReviewerByEmail(
ReviewerByEmailSetEntryProto.newBuilder()
.setState("CC")
.setAddress("emailonly@example.com")
.setTimestampMillis(1212L))
.build());
// Address doesn't consider the name field in equals, so we have to check it manually.
// TODO(dborowitz): Fix Address#equals.
ImmutableSet<Address> ccs = actual.reviewersByEmail().byState(ReviewerStateInternal.CC);
assertThat(ccs).hasSize(1);
Address address = Iterables.getOnlyElement(ccs);
assertThat(address.name()).isNull();
assertThat(address.email()).isEqualTo("emailonly@example.com");
}
@Test
public void serializePendingReviewers() throws Exception {
assertRoundTrip(
newBuilder()
.pendingReviewers(
ReviewerSet.fromTable(
ImmutableTable.<ReviewerStateInternal, Account.Id, Timestamp>builder()
.put(ReviewerStateInternal.CC, Account.id(2001), new Timestamp(1212L))
.put(ReviewerStateInternal.REVIEWER, Account.id(2002), new Timestamp(3434L))
.build()))
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addPendingReviewer(
ReviewerSetEntryProto.newBuilder()
.setState("CC")
.setAccountId(2001)
.setTimestampMillis(1212L))
.addPendingReviewer(
ReviewerSetEntryProto.newBuilder()
.setState("REVIEWER")
.setAccountId(2002)
.setTimestampMillis(3434L))
.build());
}
@Test
public void serializePendingReviewersByEmail() throws Exception {
assertRoundTrip(
newBuilder()
.pendingReviewersByEmail(
ReviewerByEmailSet.fromTable(
ImmutableTable.<ReviewerStateInternal, Address, Timestamp>builder()
.put(
ReviewerStateInternal.CC,
Address.create("Name1", "email1@example.com"),
new Timestamp(1212L))
.put(
ReviewerStateInternal.REVIEWER,
Address.create("Name2", "email2@example.com"),
new Timestamp(3434L))
.build()))
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addPendingReviewerByEmail(
ReviewerByEmailSetEntryProto.newBuilder()
.setState("CC")
.setAddress("Name1 <email1@example.com>")
.setTimestampMillis(1212L))
.addPendingReviewerByEmail(
ReviewerByEmailSetEntryProto.newBuilder()
.setState("REVIEWER")
.setAddress("Name2 <email2@example.com>")
.setTimestampMillis(3434L))
.build());
}
@Test
public void serializeAllPastReviewers() throws Exception {
assertRoundTrip(
newBuilder().allPastReviewers(ImmutableList.of(Account.id(2002), Account.id(2001))).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addPastReviewer(2002)
.addPastReviewer(2001)
.build());
}
@Test
public void serializeReviewerUpdates() throws Exception {
assertRoundTrip(
newBuilder()
.reviewerUpdates(
ImmutableList.of(
ReviewerStatusUpdate.create(
new Timestamp(1212L),
Account.id(1000),
Account.id(2002),
ReviewerStateInternal.CC),
ReviewerStatusUpdate.create(
new Timestamp(3434L),
Account.id(1000),
Account.id(2001),
ReviewerStateInternal.REVIEWER)))
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addReviewerUpdate(
ReviewerStatusUpdateProto.newBuilder()
.setTimestampMillis(1212L)
.setUpdatedBy(1000)
.setReviewer(2002)
.setState("CC"))
.addReviewerUpdate(
ReviewerStatusUpdateProto.newBuilder()
.setTimestampMillis(3434L)
.setUpdatedBy(1000)
.setReviewer(2001)
.setState("REVIEWER"))
.build());
}
@Test
public void serializeAttentionSetUpdates() throws Exception {
assertRoundTrip(
newBuilder()
.attentionSet(
ImmutableSet.of(
AttentionSetUpdate.createFromRead(
Instant.EPOCH.plusSeconds(23),
Account.id(1000),
AttentionSetUpdate.Operation.ADD,
"reason 1"),
AttentionSetUpdate.createFromRead(
Instant.EPOCH.plusSeconds(42),
Account.id(2000),
AttentionSetUpdate.Operation.REMOVE,
"reason 2")))
.build(),
newProtoBuilder()
.addAttentionSetUpdate(
AttentionSetUpdateProto.newBuilder()
.setTimestampMillis(23_000) // epoch millis
.setAccount(1000)
.setOperation("ADD")
.setReason("reason 1"))
.addAttentionSetUpdate(
AttentionSetUpdateProto.newBuilder()
.setTimestampMillis(42_000) // epoch millis
.setAccount(2000)
.setOperation("REMOVE")
.setReason("reason 2"))
.build());
}
@Test
public void serializeAssigneeUpdates() throws Exception {
assertRoundTrip(
newBuilder()
.assigneeUpdates(
ImmutableList.of(
AssigneeStatusUpdate.create(
new Timestamp(1212L), Account.id(1000), Optional.of(Account.id(2001))),
AssigneeStatusUpdate.create(
new Timestamp(3434L), Account.id(1000), Optional.empty())))
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addAssigneeUpdate(
AssigneeStatusUpdateProto.newBuilder()
.setTimestampMillis(1212L)
.setUpdatedBy(1000)
.setCurrentAssignee(2001)
.setHasCurrentAssignee(true))
.addAssigneeUpdate(
AssigneeStatusUpdateProto.newBuilder()
.setTimestampMillis(3434L)
.setUpdatedBy(1000)
.setHasCurrentAssignee(false))
.build());
}
@Test
public void serializeSubmitRecords() throws Exception {
SubmitRecord sr1 = new SubmitRecord();
sr1.status = SubmitRecord.Status.OK;
SubmitRecord sr2 = new SubmitRecord();
sr2.status = SubmitRecord.Status.FORCED;
assertRoundTrip(
newBuilder().submitRecords(ImmutableList.of(sr2, sr1)).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addSubmitRecord("{\"status\":\"FORCED\"}")
.addSubmitRecord("{\"status\":\"OK\"}")
.build());
}
@Test
public void serializeChangeMessages() throws Exception {
ChangeMessage m1 =
new ChangeMessage(
ChangeMessage.key(ID, "uuid1"),
Account.id(1000),
new Timestamp(1212L),
PatchSet.id(ID, 1));
ByteString m1Bytes = toByteString(m1, ChangeMessageProtoConverter.INSTANCE);
assertThat(m1Bytes.size()).isEqualTo(35);
ChangeMessage m2 =
new ChangeMessage(
ChangeMessage.key(ID, "uuid2"),
Account.id(2000),
new Timestamp(3434L),
PatchSet.id(ID, 2));
ByteString m2Bytes = toByteString(m2, ChangeMessageProtoConverter.INSTANCE);
assertThat(m2Bytes.size()).isEqualTo(35);
assertThat(m2Bytes).isNotEqualTo(m1Bytes);
assertRoundTrip(
newBuilder().changeMessages(ImmutableList.of(m2, m1)).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addChangeMessage(m2Bytes)
.addChangeMessage(m1Bytes)
.build());
}
@Test
public void serializePublishedComments() throws Exception {
Comment c1 =
new Comment(
new Comment.Key("uuid1", "file1", 1),
Account.id(1001),
new Timestamp(1212L),
(short) 1,
"message 1",
"serverId",
false);
c1.setCommitId(ObjectId.fromString("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"));
String c1Json = Serializer.GSON.toJson(c1);
Comment c2 =
new Comment(
new Comment.Key("uuid2", "file2", 2),
Account.id(1002),
new Timestamp(3434L),
(short) 2,
"message 2",
"serverId",
true);
c2.setCommitId(ObjectId.fromString("bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"));
String c2Json = Serializer.GSON.toJson(c2);
assertRoundTrip(
newBuilder()
.publishedComments(ImmutableListMultimap.of(c2.getCommitId(), c2, c1.getCommitId(), c1))
.build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.addPublishedComment(c2Json)
.addPublishedComment(c1Json)
.build());
}
@Test
public void serializeUpdateCount() throws Exception {
assertRoundTrip(
newBuilder().updateCount(234).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setColumns(colsProto)
.setUpdateCount(234)
.build());
}
@Test
public void changeNotesStateMethods() throws Exception {
assertThatSerializedClass(ChangeNotesState.class)
.hasAutoValueMethods(
ImmutableMap.<String, Type>builder()
.put("metaId", ObjectId.class)
.put("changeId", Change.Id.class)
.put("serverId", String.class)
.put("columns", ChangeColumns.class)
.put("hashtags", new TypeLiteral<ImmutableSet<String>>() {}.getType())
.put(
"patchSets",
new TypeLiteral<ImmutableList<Map.Entry<PatchSet.Id, PatchSet>>>() {}.getType())
.put(
"approvals",
new TypeLiteral<
ImmutableList<Map.Entry<PatchSet.Id, PatchSetApproval>>>() {}.getType())
.put("reviewers", ReviewerSet.class)
.put("reviewersByEmail", ReviewerByEmailSet.class)
.put("pendingReviewers", ReviewerSet.class)
.put("pendingReviewersByEmail", ReviewerByEmailSet.class)
.put("allPastReviewers", new TypeLiteral<ImmutableList<Account.Id>>() {}.getType())
.put(
"reviewerUpdates",
new TypeLiteral<ImmutableList<ReviewerStatusUpdate>>() {}.getType())
.put(
"attentionSet",
new TypeLiteral<ImmutableSet<AttentionSetUpdate>>() {}.getType())
.put(
"assigneeUpdates",
new TypeLiteral<ImmutableList<AssigneeStatusUpdate>>() {}.getType())
.put("submitRecords", new TypeLiteral<ImmutableList<SubmitRecord>>() {}.getType())
.put("changeMessages", new TypeLiteral<ImmutableList<ChangeMessage>>() {}.getType())
.put(
"publishedComments",
new TypeLiteral<ImmutableListMultimap<ObjectId, Comment>>() {}.getType())
.put("updateCount", int.class)
.build());
}
@Test
public void changeColumnsMethods() throws Exception {
assertThatSerializedClass(ChangeColumns.class)
.hasAutoValueMethods(
ImmutableMap.<String, Type>builder()
.put("changeKey", Change.Key.class)
.put("createdOn", Timestamp.class)
.put("lastUpdatedOn", Timestamp.class)
.put("owner", Account.Id.class)
.put("branch", String.class)
.put("currentPatchSetId", PatchSet.Id.class)
.put("subject", String.class)
.put("topic", String.class)
.put("originalSubject", String.class)
.put("submissionId", String.class)
.put("status", Change.Status.class)
.put("isPrivate", boolean.class)
.put("workInProgress", boolean.class)
.put("reviewStarted", boolean.class)
.put("revertOf", Change.Id.class)
.put("cherryPickOf", PatchSet.Id.class)
.put("toBuilder", ChangeNotesState.ChangeColumns.Builder.class)
.build());
}
@Test
public void patchSetFields() throws Exception {
assertThatSerializedClass(PatchSet.class)
.hasAutoValueMethods(
ImmutableMap.<String, Type>builder()
.put("id", PatchSet.Id.class)
.put("commitId", ObjectId.class)
.put("uploader", Account.Id.class)
.put("createdOn", Timestamp.class)
.put("groups", new TypeLiteral<ImmutableList<String>>() {}.getType())
.put("pushCertificate", new TypeLiteral<Optional<String>>() {}.getType())
.put("description", new TypeLiteral<Optional<String>>() {}.getType())
.build());
}
@Test
public void patchSetApprovalFields() throws Exception {
assertThatSerializedClass(PatchSetApproval.Key.class)
.hasAutoValueMethods(
ImmutableMap.<String, Type>builder()
.put("patchSetId", PatchSet.Id.class)
.put("accountId", Account.Id.class)
.put("labelId", LabelId.class)
.build());
assertThatSerializedClass(PatchSetApproval.class)
.hasAutoValueMethods(
ImmutableMap.<String, Type>builder()
.put("key", PatchSetApproval.Key.class)
.put("value", short.class)
.put("granted", Timestamp.class)
.put("tag", new TypeLiteral<Optional<String>>() {}.getType())
.put("realAccountId", Account.Id.class)
.put("postSubmit", boolean.class)
.put("toBuilder", PatchSetApproval.Builder.class)
.build());
}
@Test
public void reviewerSetFields() throws Exception {
assertThatSerializedClass(ReviewerSet.class)
.hasFields(
ImmutableMap.of(
"table",
new TypeLiteral<
ImmutableTable<ReviewerStateInternal, Account.Id, Timestamp>>() {}.getType(),
"accounts",
new TypeLiteral<ImmutableSet<Account.Id>>() {}.getType()));
}
@Test
public void reviewerByEmailSetFields() throws Exception {
assertThatSerializedClass(ReviewerByEmailSet.class)
.hasFields(
ImmutableMap.of(
"table",
new TypeLiteral<
ImmutableTable<ReviewerStateInternal, Address, Timestamp>>() {}.getType(),
"users",
new TypeLiteral<ImmutableSet<Address>>() {}.getType()));
}
@Test
public void reviewerStatusUpdateMethods() throws Exception {
assertThatSerializedClass(ReviewerStatusUpdate.class)
.hasAutoValueMethods(
ImmutableMap.of(
"date", Timestamp.class,
"updatedBy", Account.Id.class,
"reviewer", Account.Id.class,
"state", ReviewerStateInternal.class));
}
@Test
public void assigneeStatusUpdateMethods() throws Exception {
assertThatSerializedClass(AssigneeStatusUpdate.class)
.hasAutoValueMethods(
ImmutableMap.of(
"date",
Timestamp.class,
"updatedBy",
Account.Id.class,
"currentAssignee",
new TypeLiteral<Optional<Account.Id>>() {}.getType()));
}
@Test
public void submitRecordFields() throws Exception {
assertThatSerializedClass(SubmitRecord.class)
.hasFields(
ImmutableMap.of(
"status",
SubmitRecord.Status.class,
"labels",
new TypeLiteral<List<SubmitRecord.Label>>() {}.getType(),
"requirements",
new TypeLiteral<List<SubmitRequirement>>() {}.getType(),
"errorMessage",
String.class));
assertThatSerializedClass(SubmitRecord.Label.class)
.hasFields(
ImmutableMap.of(
"label", String.class,
"status", SubmitRecord.Label.Status.class,
"appliedBy", Account.Id.class));
assertThatSerializedClass(SubmitRequirement.class)
.hasAutoValueMethods(
ImmutableMap.of(
"fallbackText", String.class,
"type", String.class));
}
@Test
public void changeMessageFields() throws Exception {
assertThatSerializedClass(ChangeMessage.Key.class)
.hasAutoValueMethods(ImmutableMap.of("changeId", Change.Id.class, "uuid", String.class));
assertThatSerializedClass(ChangeMessage.class)
.hasFields(
ImmutableMap.<String, Type>builder()
.put("key", ChangeMessage.Key.class)
.put("author", Account.Id.class)
.put("writtenOn", Timestamp.class)
.put("message", String.class)
.put("patchset", PatchSet.Id.class)
.put("tag", String.class)
.put("realAuthor", Account.Id.class)
.build());
}
@Test
public void commentFields() throws Exception {
assertThatSerializedClass(Comment.Key.class)
.hasFields(
ImmutableMap.of(
"uuid", String.class, "filename", String.class, "patchSetId", int.class));
assertThatSerializedClass(Comment.Identity.class).hasFields(ImmutableMap.of("id", int.class));
assertThatSerializedClass(Comment.Range.class)
.hasFields(
ImmutableMap.of(
"startLine", int.class,
"startChar", int.class,
"endLine", int.class,
"endChar", int.class));
assertThatSerializedClass(Comment.class)
.hasFields(
ImmutableMap.<String, Type>builder()
.put("key", Comment.Key.class)
.put("lineNbr", int.class)
.put("author", Comment.Identity.class)
.put("realAuthor", Comment.Identity.class)
.put("writtenOn", Timestamp.class)
.put("side", short.class)
.put("message", String.class)
.put("parentUuid", String.class)
.put("range", Comment.Range.class)
.put("tag", String.class)
.put("revId", String.class)
.put("serverId", String.class)
.put("unresolved", boolean.class)
.build());
}
@Test
public void serializeServerId() throws Exception {
assertRoundTrip(
newBuilder().serverId(DEFAULT_SERVER_ID).build(),
ChangeNotesStateProto.newBuilder()
.setMetaId(SHA_BYTES)
.setChangeId(ID.get())
.setServerId(DEFAULT_SERVER_ID)
.setHasServerId(true)
.setColumns(colsProto.toBuilder())
.build());
}
private static ChangeNotesStateProto toProto(ChangeNotesState state) throws Exception {
return ChangeNotesStateProto.parseFrom(Serializer.INSTANCE.serialize(state));
}
private static ChangeNotesState assertRoundTrip(
ChangeNotesState state, ChangeNotesStateProto expectedProto) throws Exception {
ChangeNotesStateProto actualProto = toProto(state);
assertThat(actualProto).isEqualTo(expectedProto);
ChangeNotesState actual = Serializer.INSTANCE.deserialize(Serializer.INSTANCE.serialize(state));
assertThat(actual).isEqualTo(state);
// It's possible that ChangeNotesState contains objects which implement equals without taking
// into account all fields. Return the actual deserialized instance so that callers can perform
// additional assertions if necessary.
return actual;
}
}