javatests/com/google/gerrit/mail/TextParserTest.java - gerrit.git - Git at Google

 // Copyright (C) 2016 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.mail;

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

 import com.google.gerrit.entities.HumanComment;
 import java.util.List;
 import org.junit.Test;

 public class TextParserTest extends AbstractParserTest {
   private static final String quotedFooter =
       ""
           + "> To view, visit https://gerrit-review.googlesource.com/c/project/+/123\n"
           + "> To unsubscribe, visit https://gerrit-review.googlesource.com\n"
           + "> \n"
           + "> Gerrit-MessageType: comment\n"
           + "> Gerrit-Change-Id: Ie1234021bf1e8d1425641af58fd648fc011db153\n"
           + "> Gerrit-PatchSet: 1\n"
           + "> Gerrit-Project: gerrit\n"
           + "> Gerrit-Branch: master\n"
           + "> Gerrit-Owner: Foo Bar <foo@bar.com>\n"
           + "> Gerrit-HasComments: Yes";

   @Test
   public void simpleChangeMessage() {
     MailMessage.Builder b = newMailMessageBuilder();
     b.textContent("Looks good to me\n" + quotedFooter);

     List<HumanComment> comments = defaultComments();
     List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

     assertThat(parsedComments).hasSize(1);
     assertPatchsetComment("Looks good to me", parsedComments.get(0));
   }

   @Test
   public void simpleInlineComments() {
     MailMessage.Builder b = newMailMessageBuilder();
     b.textContent(
         newPlaintextBody(
                 "Looks good to me",
                 "I have a comment on this.",
                 null,
                 "Also have a comment here.",
                 null,
                 null,
                 null)
             + quotedFooter);

     List<HumanComment> comments = defaultComments();
     List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

     assertThat(parsedComments).hasSize(3);
     assertPatchsetComment("Looks good to me", parsedComments.get(0));
     assertInlineComment("I have a comment on this.", parsedComments.get(1), comments.get(1));
     assertInlineComment("Also have a comment here.", parsedComments.get(2), comments.get(3));
   }

   @Test
   public void simpleFileComment() {
     MailMessage.Builder b = newMailMessageBuilder();
     b.textContent(
         newPlaintextBody(
                 "Looks good to me",
                 null,
                 null,
                 "Also have a comment here.",
                 "This is a nice file",
                 null,
                 null)
             + quotedFooter);

     List<HumanComment> comments = defaultComments();
     List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

     assertThat(parsedComments).hasSize(3);
     assertPatchsetComment("Looks good to me", parsedComments.get(0));
     assertFileComment("This is a nice file", parsedComments.get(1), comments.get(1).key.filename);
     assertInlineComment("Also have a comment here.", parsedComments.get(2), comments.get(3));
   }

   @Test
   public void noComments() {
     MailMessage.Builder b = newMailMessageBuilder();
     b.textContent(newPlaintextBody(null, null, null, null, null, null, null) + quotedFooter);

     List<HumanComment> comments = defaultComments();
     List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

     assertThat(parsedComments).isEmpty();
   }

   @Test
   public void noChangeMessage() {
     MailMessage.Builder b = newMailMessageBuilder();
     b.textContent(
         newPlaintextBody(
                 null, null, null, "Also have a comment here.", "This is a nice file", null, null)
             + quotedFooter);

     List<HumanComment> comments = defaultComments();
     List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

     assertThat(parsedComments).hasSize(2);
     assertFileComment("This is a nice file", parsedComments.get(0), comments.get(1).key.filename);
     assertInlineComment("Also have a comment here.", parsedComments.get(1), comments.get(3));
   }

   @Test
   public void allCommentsGmail() {
     MailMessage.Builder b = newMailMessageBuilder();
     b.textContent(
         (newPlaintextBody(
                     "Looks good to me",
                     null,
                     null,
                     "Also have a comment here.",
                     "This is a nice file",
                     null,
                     null)
                 + quotedFooter)
             .replace("> ", ">> "));

     List<HumanComment> comments = defaultComments();
     List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

     assertThat(parsedComments).hasSize(3);
     assertPatchsetComment("Looks good to me", parsedComments.get(0));
     assertFileComment("This is a nice file", parsedComments.get(1), comments.get(1).key.filename);
     assertInlineComment("Also have a comment here.", parsedComments.get(2), comments.get(3));
   }

   @Test
   public void replyToFileComment() {
     MailMessage.Builder b = newMailMessageBuilder();
     b.textContent(
         newPlaintextBody(
                 "Looks good to me",
                 null,
                 null,
                 null,
                 null,
                 null,
                 "Comment in reply to file comment")
             + quotedFooter);

     List<HumanComment> comments = defaultComments();
     List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

     assertThat(parsedComments).hasSize(2);
     assertPatchsetComment("Looks good to me", parsedComments.get(0));
     assertInlineComment("Comment in reply to file comment", parsedComments.get(1), comments.get(0));
   }

   @Test
   public void squashComments() {
     MailMessage.Builder b = newMailMessageBuilder();
     b.textContent(
         "Nice change\n> Some quoted content\nMy other comment on the same entity\n" + quotedFooter);

     List<MailComment> parsedComments = TextParser.parse(b.build(), defaultComments(), CHANGE_URL);

     assertThat(parsedComments).hasSize(1);
     assertPatchsetComment(
         "Nice change\n\nMy other comment on the same entity", parsedComments.get(0));
   }

   /**
    * Create a plaintext message body with the specified comments.
    *
    * @param c1 Comment in reply to first inline comment.
    * @param c2 Comment in reply to second inline comment.
    * @param c3 Comment in reply to third inline comment.
    * @param f1 Comment on file one.
    * @param f2 Comment on file two.
    * @param fc1 Comment in reply to a comment of file 1.
    * @return A string with all inline comments and the original quoted email.
    */
   private static String newPlaintextBody(
       String changeMessage, String c1, String c2, String c3, String f1, String f2, String fc1) {
     return (changeMessage == null ? "" : changeMessage + "\n")
         + "On Thu, Feb 9, 2017 at 8:21 AM, ekempin (Gerrit)\n"
         + "<noreply-gerritcodereview-qUgXfQecoDLHwp0MldAzig@google.com> wrote: \n"
         + "> Foo Bar has posted comments on this change. (  \n"
         + "> "
         + CHANGE_URL
         + "/1 )\n"
         + "> \n"
         + "> Change subject: Test change\n"
         + "> ...............................................................\n"
         + "> \n"
         + "> \n"
         + "> Patch Set 1: Code-Review+1\n"
         + "> \n"
         + "> (3 comments)\n"
         + "> \n"
         + "> "
         + CHANGE_URL
         + "/1/gerrit-server/test.txt\n"
         + "> File  \n"
         + "> gerrit-server/test.txt:\n"
         + (f1 == null ? "" : f1 + "\n")
         + "> \n"
         + "> Patch Set #4:\n"
         + "> "
         + CHANGE_URL
         + "/1/gerrit-server/test.txt\n"
         + "> \n"
         + "> Some comment"
         + "> \n"
         + (fc1 == null ? "" : fc1 + "\n")
         + "> "
         + CHANGE_URL
         + "/1/gerrit-server/test.txt@2\n"
         + "> PS1, Line 2: throw new Exception(\"Object has unsupported: \" +\n"
         + ">               :             entry.getValue() +\n"
         + ">               :             \" must be java.util.Date\");\n"
         + "> Should entry.getKey() be included in this message?\n"
         + "> \n"
         + (c1 == null ? "" : c1 + "\n")
         + ">\n"
         + "> \n"
         + "> "
         + CHANGE_URL
         + "/1/gerrit-server/test.txt@3\n"
         + "> PS1, Line 3: throw new Exception(\"Object has: \" +\n"
         + ">               :             entry.getValue().getClass() +\n"
         + ">              :             \" must be java.util.Date\");\n"
         + "> same here\n"
         + "> \n"
         + (c2 == null ? "" : c2 + "\n")
         + "> \n"
         + "> "
         + CHANGE_URL
         + "/1/gerrit-server/readme.txt\n"
         + "> File  \n"
         + "> gerrit-server/readme.txt:\n"
         + (f2 == null ? "" : f2 + "\n")
         + "> \n"
         + "> "
         + CHANGE_URL
         + "/1/gerrit-server/readme.txt@3\n"
         + "> PS1, Line 3: E\n"
         + "> Should this be EEE like in other places?\n"
         + (c3 == null ? "" : c3 + "\n");
   }
 }
	// Copyright (C) 2016 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.mail;

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

	import com.google.gerrit.entities.HumanComment;
	import java.util.List;
	import org.junit.Test;

	public class TextParserTest extends AbstractParserTest {
	private static final String quotedFooter =
	""
	+ "> To view, visit https://gerrit-review.googlesource.com/c/project/+/123\n"
	+ "> To unsubscribe, visit https://gerrit-review.googlesource.com\n"
	+ "> \n"
	+ "> Gerrit-MessageType: comment\n"
	+ "> Gerrit-Change-Id: Ie1234021bf1e8d1425641af58fd648fc011db153\n"
	+ "> Gerrit-PatchSet: 1\n"
	+ "> Gerrit-Project: gerrit\n"
	+ "> Gerrit-Branch: master\n"
	+ "> Gerrit-Owner: Foo Bar <foo@bar.com>\n"
	+ "> Gerrit-HasComments: Yes";

	@Test
	public void simpleChangeMessage() {
	MailMessage.Builder b = newMailMessageBuilder();
	b.textContent("Looks good to me\n" + quotedFooter);

	List<HumanComment> comments = defaultComments();
	List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

	assertThat(parsedComments).hasSize(1);
	assertPatchsetComment("Looks good to me", parsedComments.get(0));
	}

	@Test
	public void simpleInlineComments() {
	MailMessage.Builder b = newMailMessageBuilder();
	b.textContent(
	newPlaintextBody(
	"Looks good to me",
	"I have a comment on this.",
	null,
	"Also have a comment here.",
	null,
	null,
	null)
	+ quotedFooter);

	List<HumanComment> comments = defaultComments();
	List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

	assertThat(parsedComments).hasSize(3);
	assertPatchsetComment("Looks good to me", parsedComments.get(0));
	assertInlineComment("I have a comment on this.", parsedComments.get(1), comments.get(1));
	assertInlineComment("Also have a comment here.", parsedComments.get(2), comments.get(3));
	}

	@Test
	public void simpleFileComment() {
	MailMessage.Builder b = newMailMessageBuilder();
	b.textContent(
	newPlaintextBody(
	"Looks good to me",
	null,
	null,
	"Also have a comment here.",
	"This is a nice file",
	null,
	null)
	+ quotedFooter);

	List<HumanComment> comments = defaultComments();
	List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

	assertThat(parsedComments).hasSize(3);
	assertPatchsetComment("Looks good to me", parsedComments.get(0));
	assertFileComment("This is a nice file", parsedComments.get(1), comments.get(1).key.filename);
	assertInlineComment("Also have a comment here.", parsedComments.get(2), comments.get(3));
	}

	@Test
	public void noComments() {
	MailMessage.Builder b = newMailMessageBuilder();
	b.textContent(newPlaintextBody(null, null, null, null, null, null, null) + quotedFooter);

	List<HumanComment> comments = defaultComments();
	List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

	assertThat(parsedComments).isEmpty();
	}

	@Test
	public void noChangeMessage() {
	MailMessage.Builder b = newMailMessageBuilder();
	b.textContent(
	newPlaintextBody(
	null, null, null, "Also have a comment here.", "This is a nice file", null, null)
	+ quotedFooter);

	List<HumanComment> comments = defaultComments();
	List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

	assertThat(parsedComments).hasSize(2);
	assertFileComment("This is a nice file", parsedComments.get(0), comments.get(1).key.filename);
	assertInlineComment("Also have a comment here.", parsedComments.get(1), comments.get(3));
	}

	@Test
	public void allCommentsGmail() {
	MailMessage.Builder b = newMailMessageBuilder();
	b.textContent(
	(newPlaintextBody(
	"Looks good to me",
	null,
	null,
	"Also have a comment here.",
	"This is a nice file",
	null,
	null)
	+ quotedFooter)
	.replace("> ", ">> "));

	List<HumanComment> comments = defaultComments();
	List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

	assertThat(parsedComments).hasSize(3);
	assertPatchsetComment("Looks good to me", parsedComments.get(0));
	assertFileComment("This is a nice file", parsedComments.get(1), comments.get(1).key.filename);
	assertInlineComment("Also have a comment here.", parsedComments.get(2), comments.get(3));
	}

	@Test
	public void replyToFileComment() {
	MailMessage.Builder b = newMailMessageBuilder();
	b.textContent(
	newPlaintextBody(
	"Looks good to me",
	null,
	null,
	null,
	null,
	null,
	"Comment in reply to file comment")
	+ quotedFooter);

	List<HumanComment> comments = defaultComments();
	List<MailComment> parsedComments = TextParser.parse(b.build(), comments, CHANGE_URL);

	assertThat(parsedComments).hasSize(2);
	assertPatchsetComment("Looks good to me", parsedComments.get(0));
	assertInlineComment("Comment in reply to file comment", parsedComments.get(1), comments.get(0));
	}

	@Test
	public void squashComments() {
	MailMessage.Builder b = newMailMessageBuilder();
	b.textContent(
	"Nice change\n> Some quoted content\nMy other comment on the same entity\n" + quotedFooter);

	List<MailComment> parsedComments = TextParser.parse(b.build(), defaultComments(), CHANGE_URL);

	assertThat(parsedComments).hasSize(1);
	assertPatchsetComment(
	"Nice change\n\nMy other comment on the same entity", parsedComments.get(0));
	}

	/**
	* Create a plaintext message body with the specified comments.
	*
	* @param c1 Comment in reply to first inline comment.
	* @param c2 Comment in reply to second inline comment.
	* @param c3 Comment in reply to third inline comment.
	* @param f1 Comment on file one.
	* @param f2 Comment on file two.
	* @param fc1 Comment in reply to a comment of file 1.
	* @return A string with all inline comments and the original quoted email.
	*/
	private static String newPlaintextBody(
	String changeMessage, String c1, String c2, String c3, String f1, String f2, String fc1) {
	return (changeMessage == null ? "" : changeMessage + "\n")
	+ "On Thu, Feb 9, 2017 at 8:21 AM, ekempin (Gerrit)\n"
	+ "<noreply-gerritcodereview-qUgXfQecoDLHwp0MldAzig@google.com> wrote: \n"
	+ "> Foo Bar has posted comments on this change. ( \n"
	+ "> "
	+ CHANGE_URL
	+ "/1 )\n"
	+ "> \n"
	+ "> Change subject: Test change\n"
	+ "> ...............................................................\n"
	+ "> \n"
	+ "> \n"
	+ "> Patch Set 1: Code-Review+1\n"
	+ "> \n"
	+ "> (3 comments)\n"
	+ "> \n"
	+ "> "
	+ CHANGE_URL
	+ "/1/gerrit-server/test.txt\n"
	+ "> File \n"
	+ "> gerrit-server/test.txt:\n"
	+ (f1 == null ? "" : f1 + "\n")
	+ "> \n"
	+ "> Patch Set #4:\n"
	+ "> "
	+ CHANGE_URL
	+ "/1/gerrit-server/test.txt\n"
	+ "> \n"
	+ "> Some comment"
	+ "> \n"
	+ (fc1 == null ? "" : fc1 + "\n")
	+ "> "
	+ CHANGE_URL
	+ "/1/gerrit-server/test.txt@2\n"
	+ "> PS1, Line 2: throw new Exception(\"Object has unsupported: \" +\n"
	+ "> : entry.getValue() +\n"
	+ "> : \" must be java.util.Date\");\n"
	+ "> Should entry.getKey() be included in this message?\n"
	+ "> \n"
	+ (c1 == null ? "" : c1 + "\n")
	+ ">\n"
	+ "> \n"
	+ "> "
	+ CHANGE_URL
	+ "/1/gerrit-server/test.txt@3\n"
	+ "> PS1, Line 3: throw new Exception(\"Object has: \" +\n"
	+ "> : entry.getValue().getClass() +\n"
	+ "> : \" must be java.util.Date\");\n"
	+ "> same here\n"
	+ "> \n"
	+ (c2 == null ? "" : c2 + "\n")
	+ "> \n"
	+ "> "
	+ CHANGE_URL
	+ "/1/gerrit-server/readme.txt\n"
	+ "> File \n"
	+ "> gerrit-server/readme.txt:\n"
	+ (f2 == null ? "" : f2 + "\n")
	+ "> \n"
	+ "> "
	+ CHANGE_URL
	+ "/1/gerrit-server/readme.txt@3\n"
	+ "> PS1, Line 3: E\n"
	+ "> Should this be EEE like in other places?\n"
	+ (c3 == null ? "" : c3 + "\n");
	}
	}