PostgreSQLLocalInstructionCommentTests.java

/*
Copyright 2014 Google Inc. All Rights Reserved.

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.security.zynamics.binnavi.Database.PostgreSQL.commentTests;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import com.google.common.collect.Iterables;
import com.google.security.zynamics.binnavi.Database.Exceptions.CPartialLoadException;
import com.google.security.zynamics.binnavi.Database.Exceptions.CouldntDeleteException;
import com.google.security.zynamics.binnavi.Database.Exceptions.CouldntLoadDataException;
import com.google.security.zynamics.binnavi.Database.Exceptions.CouldntSaveDataException;
import com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException;
import com.google.security.zynamics.binnavi.Database.PostgreSQL.Functions.PostgreSQLCommentFunctions;
import com.google.security.zynamics.binnavi.Database.PostgreSQL.UniqueTestUserGenerator;
import com.google.security.zynamics.binnavi.Exceptions.MaybeNullException;
import com.google.security.zynamics.binnavi.Gui.GraphWindows.CommentDialogs.CComment;
import com.google.security.zynamics.binnavi.Gui.GraphWindows.CommentDialogs.Interfaces.IComment;
import com.google.security.zynamics.binnavi.Gui.Users.Interfaces.IUser;
import com.google.security.zynamics.binnavi.disassembly.CCodeNode;
import com.google.security.zynamics.binnavi.disassembly.INaviInstruction;
import com.google.security.zynamics.binnavi.disassembly.types.ExpensiveBaseTest;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

import java.util.ArrayList;
import java.util.List;

@RunWith(JUnit4.class)
public class PostgreSQLLocalInstructionCommentTests extends ExpensiveBaseTest {
  private CCodeNode codeNode;
  private INaviInstruction instruction;

  @Before
  public void loadCodeNode() throws CouldntLoadDataException, CPartialLoadException,
      LoadCancelledException, MaybeNullException {
    codeNode = (CCodeNode) loadCodeNode(getNotepadModule(), "sub_1002B87");
    instruction = Iterables.getLast(codeNode.getInstructions());
  }

  @Test(expected = NullPointerException.class)
  public void appendLocalInstructionComment1() throws CouldntSaveDataException {
    getProvider().appendLocalInstructionComment(null, null, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void appendLocalInstructionComment2() throws CouldntSaveDataException {
    getProvider().appendLocalInstructionComment(codeNode, null, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void appendLocalInstructionComment3() throws CouldntSaveDataException {
    getProvider().appendLocalInstructionComment(codeNode, instruction, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void appendLocalInstructionComment4() throws CouldntSaveDataException {
    getProvider().appendLocalInstructionComment(codeNode, instruction, " FAIL ", null);
  }

  @Test
  public void appendLocalInstructionComment5()
      throws CouldntLoadDataException, CouldntSaveDataException {

    final IUser user = new UniqueTestUserGenerator(getProvider()).nextActiveUser();

    final List<IComment> storedComments =
        codeNode.getComments().getLocalInstructionComment(instruction) == null ? new ArrayList<
        IComment>()
        : codeNode.getComments().getLocalInstructionComment(instruction);

    final IComment lastComment =
        storedComments.isEmpty() ? null : Iterables.getLast(storedComments);

    final String firstCommentString = " PASS LOCAL INSTRUCTION COMMENT 1 ";
    final int firstCommentId = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, firstCommentString, user.getUserId());
    final IComment firstComment =
        new CComment(firstCommentId, user, lastComment, firstCommentString);

    final String secondCommentString = " PASS LOCAL INSTRUCTION COMMENT 2 ";
    final int secondCommentId = getProvider().appendLocalInstructionComment(
        codeNode, instruction, secondCommentString, user.getUserId());
    final IComment secondComment =
        new CComment(secondCommentId, user, firstComment, secondCommentString);

    final ArrayList<IComment> commentsFromDatabase = getProvider().loadCommentById(secondCommentId);

    assertNotNull(commentsFromDatabase);
    assertEquals(storedComments.size() + 2, commentsFromDatabase.size());
    assertTrue(commentsFromDatabase.contains(firstComment));
    assertTrue(commentsFromDatabase.contains(secondComment));
  }

  @Test(expected = NullPointerException.class)
  public void deleteLocalInstructionComment1() throws CouldntDeleteException {
    getProvider().deleteLocalInstructionComment(null, null, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void deleteLocalInstructionComment2() throws CouldntDeleteException {
    getProvider().deleteLocalInstructionComment(codeNode, null, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void deleteLocalInstructionComment3() throws CouldntDeleteException {
    getProvider().deleteLocalInstructionComment(codeNode, instruction, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void deleteLocalInstructionComment4() throws CouldntDeleteException {
    getProvider().deleteLocalInstructionComment(codeNode, instruction, 1, null);
  }

  @Test
  public void deleteLocalInstructionComment5()
      throws CouldntLoadDataException, CouldntSaveDataException, CouldntDeleteException {

    final List<IComment> comments = codeNode.getComments().getLocalInstructionComment(instruction)
        == null ? new ArrayList<IComment>()
        : codeNode.getComments().getLocalInstructionComment(instruction);

    final IComment lastComment = comments.size() == 0 ? null : Iterables.getLast(comments);

    final String commentString = " TEST DELETE LOCAL INSTRUCTION COMMENT ";

    final IUser user = new UniqueTestUserGenerator(getProvider()).nextActiveUser();

    final int commentId = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, commentString, user.getUserId());

    final IComment newComment = new CComment(commentId, user, lastComment, commentString);

    final ArrayList<IComment> storedComments = getProvider().loadCommentById(commentId);

    assertNotNull(storedComments);
    assertEquals(comments.size() + 1, storedComments.size());
    assertEquals(newComment, storedComments.get(storedComments.size() - 1));

    getProvider().deleteLocalInstructionComment(
        codeNode, instruction, commentId, newComment.getUser().getUserId());

    final ArrayList<IComment> commentsAfterDelete = getProvider().loadCommentById(commentId);

    assertNotNull(commentsAfterDelete);
    assertTrue(commentsAfterDelete.isEmpty());
  }

  /**
   * This test checks if the delete of a comment in a series of comments works if the comment is the
   * last comment.
   *
   * <pre>
   * Comment 1:      Comment 1:
   * Comment 2:  ->  Comment 2:
   * Comment 3:
   * </pre>
   *
   */
  @Test
  public void deleteLocalInstructionComment6()
      throws CouldntLoadDataException, CouldntSaveDataException, CouldntDeleteException {

    final IUser user = new UniqueTestUserGenerator(getProvider()).nextActiveUser();

    final List<IComment> storedComments =
        codeNode.getComments().getLocalInstructionComment(instruction) == null ? new ArrayList<
        IComment>()
        : codeNode.getComments().getLocalInstructionComment(instruction);
    final IComment lastComment =
        storedComments.isEmpty() ? null : Iterables.getLast(storedComments);
    final String comment1String = " Comment 1: ";
    final int comment1Id = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, comment1String, user.getUserId());
    final IComment comment1 = new CComment(comment1Id, user, lastComment, comment1String);

    final String comment2String = " Comment 2: ";
    final int comment2Id = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, comment2String, user.getUserId());
    final IComment comment2 = new CComment(comment2Id, user, comment1, comment2String);

    final String comment3String = " Comment 3: ";
    final int comment3Id = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, comment3String, user.getUserId());
    final IComment comment3 = new CComment(comment3Id, user, comment2, comment3String);

    final ArrayList<IComment> commentsBeforeDelete = getProvider().loadCommentById(comment3Id);

    assertNotNull(commentsBeforeDelete);
    assertEquals(storedComments.size() + 3, commentsBeforeDelete.size());
    assertTrue(commentsBeforeDelete.contains(comment1));
    assertTrue(commentsBeforeDelete.contains(comment2));
    assertTrue(commentsBeforeDelete.contains(comment3));
    assertEquals(comment3, Iterables.getLast(commentsBeforeDelete));
    assertEquals(comment2, commentsBeforeDelete.get(commentsBeforeDelete.size() - 2));
    assertEquals(comment1, commentsBeforeDelete.get(commentsBeforeDelete.size() - 3));

    getProvider()
        .deleteLocalInstructionComment(codeNode, instruction, comment3Id, user.getUserId());

    final ArrayList<IComment> commentsAfterDelete1 = getProvider().loadCommentById(comment3Id);

    assertNotNull(commentsAfterDelete1);
    assertTrue(commentsAfterDelete1.isEmpty());

    final ArrayList<IComment> commentsAfterDelete2 = getProvider().loadCommentById(comment2Id);

    assertNotNull(commentsAfterDelete2);
    assertEquals(storedComments.size() + 2, commentsAfterDelete2.size());
    assertTrue(commentsAfterDelete2.contains(comment2));
    assertTrue(commentsAfterDelete2.contains(comment1));
    assertEquals(comment2, Iterables.getLast(commentsAfterDelete2));
    assertEquals(comment1, commentsAfterDelete2.get(commentsAfterDelete2.size() - 2));
  }

  /**
   * This test checks if the delete of a comment in a series of comments works if the comment is a
   * comment in the middle.
   *
   * <pre>
   * Comment 1:      Comment 1:
   * Comment 2:  ->  
   * Comment 3:      Comment 3:
   * </pre>
   *
   */
  @Test
  public void deleteLocalInstructionComment7()
      throws CouldntLoadDataException, CouldntSaveDataException, CouldntDeleteException {

    final IUser user = new UniqueTestUserGenerator(getProvider()).nextActiveUser();

    final List<IComment> storedComments =
        codeNode.getComments().getLocalInstructionComment(instruction) == null ? new ArrayList<
        IComment>()
        : codeNode.getComments().getLocalInstructionComment(instruction);

    final IComment lastComment =
        storedComments.isEmpty() ? null : Iterables.getLast(storedComments);
    final String comment1String = " Comment 1: ";
    final int comment1Id = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, comment1String, user.getUserId());
    final IComment comment1 = new CComment(comment1Id, user, lastComment, comment1String);

    final String comment2String = " Comment 2: ";
    final int comment2Id = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, comment2String, user.getUserId());
    final IComment comment2 = new CComment(comment2Id, user, comment1, comment2String);

    final String comment3String = " Comment 3: ";
    final int comment3Id = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, comment3String, user.getUserId());
    final IComment comment3 = new CComment(comment3Id, user, comment2, comment3String);

    final ArrayList<IComment> commentsBeforeDelete = getProvider().loadCommentById(comment3Id);

    assertNotNull(commentsBeforeDelete);
    assertEquals(storedComments.size() + 3, commentsBeforeDelete.size());
    assertTrue(commentsBeforeDelete.contains(comment1));
    assertTrue(commentsBeforeDelete.contains(comment2));
    assertTrue(commentsBeforeDelete.contains(comment3));
    assertEquals(comment3, Iterables.getLast(commentsBeforeDelete));
    assertEquals(comment2, commentsBeforeDelete.get(commentsBeforeDelete.size() - 2));
    assertEquals(comment1, commentsBeforeDelete.get(commentsBeforeDelete.size() - 3));

    getProvider()
        .deleteLocalInstructionComment(codeNode, instruction, comment2Id, user.getUserId());

    final ArrayList<IComment> commentsAfterDelete1 = getProvider().loadCommentById(comment2Id);

    assertNotNull(commentsAfterDelete1);
    assertTrue(commentsAfterDelete1.isEmpty());

    final ArrayList<IComment> commentsAfterDelete2 = getProvider().loadCommentById(comment3Id);

    assertNotNull(commentsAfterDelete2);
    assertEquals(storedComments.size() + 2, commentsAfterDelete2.size());

    final IComment comment3AfterDelete = new CComment(comment3Id, user, comment1, comment3String);

    assertTrue(commentsAfterDelete2.contains(comment3AfterDelete));
    assertTrue(commentsAfterDelete2.contains(comment1));
    assertEquals(comment3AfterDelete, Iterables.getLast(commentsAfterDelete2));
    assertEquals(comment1, commentsAfterDelete2.get(commentsAfterDelete2.size() - 2));
  }

  /**
   * This test checks if the delete of a comment in a series of comments works if the comment is the
   * first comment.
   *
   * <pre>
   * Comment 1:      
   * Comment 2:  ->  Comment 2:
   * Comment 3:      Comment 3:
   * </pre>
   *
   */
  @Test
  public void deleteLocalInstructionComment8()
      throws CouldntLoadDataException, CouldntSaveDataException, CouldntDeleteException {

    final IUser user = new UniqueTestUserGenerator(getProvider()).nextActiveUser();

    final List<IComment> storedComments =
        codeNode.getComments().getLocalInstructionComment(instruction) == null ? new ArrayList<
        IComment>()
        : codeNode.getComments().getLocalInstructionComment(instruction);

    final IComment lastComment =
        storedComments.isEmpty() ? null : Iterables.getLast(storedComments);
    final String comment1String = " Comment 1: ";
    final int comment1Id = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, comment1String, user.getUserId());
    final IComment comment1 = new CComment(comment1Id, user, lastComment, comment1String);

    final String comment2String = " Comment 2: ";
    final int comment2Id = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, comment2String, user.getUserId());
    final IComment comment2 = new CComment(comment2Id, user, comment1, comment2String);

    final String comment3String = " Comment 3: ";
    final int comment3Id = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, comment3String, user.getUserId());
    final IComment comment3 = new CComment(comment3Id, user, comment2, comment3String);

    final ArrayList<IComment> commentsBeforeDelete = getProvider().loadCommentById(comment3Id);

    assertNotNull(commentsBeforeDelete);
    assertEquals(storedComments.size() + 3, commentsBeforeDelete.size());
    assertTrue(commentsBeforeDelete.contains(comment1));
    assertTrue(commentsBeforeDelete.contains(comment2));
    assertTrue(commentsBeforeDelete.contains(comment3));
    assertEquals(comment3, Iterables.getLast(commentsBeforeDelete));
    assertEquals(comment2, commentsBeforeDelete.get(commentsBeforeDelete.size() - 2));
    assertEquals(comment1, commentsBeforeDelete.get(commentsBeforeDelete.size() - 3));

    getProvider()
        .deleteLocalInstructionComment(codeNode, instruction, comment1Id, user.getUserId());

    final ArrayList<IComment> commentsAfterDelete1 = getProvider().loadCommentById(comment1Id);

    assertNotNull(commentsAfterDelete1);
    assertTrue(commentsAfterDelete1.isEmpty());

    final ArrayList<IComment> commentsAfterDelete2 = getProvider().loadCommentById(comment3Id);

    final IComment comment2AfterDelete =
        new CComment(comment2Id, user, lastComment, comment2String);
    final IComment comment3AfterDelete =
        new CComment(comment3Id, user, comment2AfterDelete, comment3String);


    assertNotNull(commentsAfterDelete2);
    assertEquals(storedComments.size() + 2, commentsAfterDelete2.size());
    assertTrue(commentsAfterDelete2.contains(comment3AfterDelete));
    assertTrue(commentsAfterDelete2.contains(comment2AfterDelete));
    assertEquals(comment3AfterDelete, Iterables.getLast(commentsAfterDelete2));
    assertEquals(comment2AfterDelete, commentsAfterDelete2.get(commentsAfterDelete2.size() - 2));
  }

  @Test(expected = NullPointerException.class)
  public void editLocalInstructionComment1() throws CouldntSaveDataException {
    getProvider().editLocalInstructionComment(null, null, null, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void editLocalInstructionComment2() throws CouldntSaveDataException {
    getProvider().editLocalInstructionComment(codeNode, null, null, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void editLocalInstructionComment3() throws CouldntSaveDataException {
    getProvider().editLocalInstructionComment(codeNode, instruction, null, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void editLocalInstructionComment4() throws CouldntSaveDataException {
    getProvider().editLocalInstructionComment(codeNode, instruction, 1, null, null);
  }

  @Test(expected = NullPointerException.class)
  public void editLocalInstructionComment5() throws CouldntSaveDataException {
    getProvider().editLocalInstructionComment(codeNode, instruction, 1, 1, null);
  }

  @Test
  public void editLocalInstructionComment6()
      throws CouldntLoadDataException, CouldntSaveDataException {

    final List<IComment> comments = codeNode.getComments().getLocalInstructionComment(instruction)
        == null ? new ArrayList<IComment>()
        : codeNode.getComments().getLocalInstructionComment(instruction);

    final IComment lastComment = comments.size() == 0 ? null : Iterables.getLast(comments);

    final IUser user = new UniqueTestUserGenerator(getProvider()).nextActiveUser();

    final String commentText = " INSTRUCTION COMMENT TEST BEFORE EDIT ";

    final Integer commentId = getProvider()
        .appendLocalInstructionComment(codeNode, instruction, commentText, user.getUserId());

    final IComment newComment = new CComment(commentId, user, lastComment, commentText);

    final ArrayList<IComment> newComments = getProvider().loadCommentById(commentId);

    assertNotNull(newComments);
    assertEquals(comments.size() + 1, newComments.size());
    assertEquals(newComment, newComments.get(newComments.size() - 1));

    final String commentAfterEdit = " INSTRUCTION COMMENT TEST AFTER EDIT ";

    getProvider().editLocalInstructionComment(
        codeNode, instruction, commentId, user.getUserId(), commentAfterEdit);
    final ArrayList<IComment> commentsAfterEdit =
        PostgreSQLCommentFunctions.loadCommentByCommentId(getProvider(), commentId);

    assertEquals(
        commentAfterEdit, commentsAfterEdit.get(commentsAfterEdit.size() - 1).getComment());
    assertEquals(commentsAfterEdit.size(), newComments.size());
  }

  @Test(expected = CouldntSaveDataException.class)
  public void editLocalInstructionComment7()
      throws CouldntLoadDataException, CouldntSaveDataException {

    final List<IComment> storedComments =
        codeNode.getComments().getLocalInstructionComment(instruction) == null ? new ArrayList<
        IComment>()
        : codeNode.getComments().getLocalInstructionComment(instruction);

    final IUser user = new UniqueTestUserGenerator(getProvider()).nextActiveUser();

    final IComment lastComment =
        storedComments.isEmpty() ? null : Iterables.getLast(storedComments);
    final String beforeEditCommentString = " EDIT LOCAL INSTRUCTION COMMENT BEFORE EDIT ";

    final int editedCommentId = getProvider().appendLocalInstructionComment(
        codeNode, instruction, beforeEditCommentString, user.getUserId());
    final IComment commentBeforeEdit =
        new CComment(editedCommentId, user, lastComment, beforeEditCommentString);

    final ArrayList<IComment> commentsFromDatabase = getProvider().loadCommentById(editedCommentId);

    assertNotNull(commentsFromDatabase);
    assertTrue(commentsFromDatabase.contains(commentBeforeEdit));
    assertEquals(storedComments.size() + 1, commentsFromDatabase.size());

    final IUser wrongUser = new UniqueTestUserGenerator(getProvider()).nextActiveUser();

    getProvider().editLocalInstructionComment(
        codeNode, instruction, editedCommentId, wrongUser.getUserId(), " FAIL ");
  }
}