com.google.javascript.jscomp.testing.NodeSubject Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 The Closure Compiler Authors.
 *
 * 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.javascript.jscomp.testing;

import static com.google.common.truth.Truth.THROW_ASSERTION_ERROR;
import static org.junit.Assert.assertEquals;

import com.google.common.truth.FailureStrategy;
import com.google.common.truth.Subject;
import com.google.javascript.rhino.Node;
import com.google.javascript.rhino.Token;

/**
 * A Truth Subject for the Node class. Usage:
 *  *   import static com.google.javascript.jscomp.testing.NodeSubject.assertNode;
 *   ...
 *   assertNode(node1).isEqualTo(node2);
 *   assertNode(node1).hasType(Token.FUNCTION);
 * 
 */
public final class NodeSubject extends Subject {
  public static NodeSubject assertNode(Node node) {
    return new NodeSubject(THROW_ASSERTION_ERROR, node);
  }

  public NodeSubject(FailureStrategy fs, Node node) {
    super(fs, node);
  }

  public void isEqualTo(Node node) {
    String treeDiff = node.checkTreeEquals(getSubject());
    if (treeDiff != null) {
      failWithRawMessage("%s", treeDiff);
    }
  }

  public void hasType(int type) {
    String message = "Node is of type " + Token.name(getSubject().getType())
        + " not of type " + Token.name(type);
    assertEquals(message, type, getSubject().getType());
  }

  public void hasCharno(int charno) {
    assertEquals(charno, getSubject().getCharno());
  }

  public void hasLineno(int lineno) {
    assertEquals(lineno, getSubject().getLineno());
  }

  public void hasLength(int length) {
    assertEquals(length, getSubject().getLength());
  }
}