HbTodoIndexTest.java

package com.dmarcotte.handlebars.editor;


import com.dmarcotte.handlebars.util.HbTestUtils;
import com.intellij.psi.search.PsiTodoSearchHelper;
import com.intellij.psi.search.TodoItem;
import com.intellij.testFramework.LightPlatformCodeInsightTestCase;
import org.jetbrains.annotations.NotNull;

public class HbTodoIndexTest extends LightPlatformCodeInsightTestCase {
  @Override
  @NotNull
  protected String getTestDataPath() {
    return HbTestUtils.BASE_TEST_DATA_PATH + "/todo/";
  }

  public void testNoTodo() {
    checkTodoCount(0);
  }

  public void testFileWithTwoTodo() {
    checkTodoCount(2);
  }

  public void testHtmlTodoOnly() {
    checkTodoCount(1);
  }

  private void checkTodoCount(int expectedTodoCount) {
    configureByFile(getTestName(true) + ".hbs");
    TodoItem[] items = PsiTodoSearchHelper.SERVICE.getInstance(getProject()).findTodoItems(getFile());
    assertEquals(expectedTodoCount, items.length);
  }
}