TestTypeRegister.java

package test.ql;

import static org.junit.Assert.assertEquals;

import org.junit.Test;

import ql.ast.expression.Identifier;
import ql.ast.expression.literal.StringLiteral;
import ql.ast.statement.Question;
import ql.ast.type.QLFloat;
import ql.ast.visitor.typechecker.TypeEnvironment;

public class TestTypeRegister {
		
	@Test
	public void testRegistration() {
		TypeEnvironment register = new TypeEnvironment();
		
		StringLiteral myString = new StringLiteral("My String");
		Identifier myIdentifier = new Identifier("aString");
		
		register.store(myIdentifier, myString.getType());
		
		assertEquals("Should return my previously bound type instance.", 
			register.resolve(myIdentifier).toString(), "string");
		
		myString = new StringLiteral("Other value");
		register.store(myIdentifier, myString.getType());
		
		assertEquals("Should return the newly bound type instance.", "string",
				register.resolve(myIdentifier).toString());
	}
	
	@Test
	public void throwsQLError() {
		TypeEnvironment register = new TypeEnvironment();
		
		Identifier myIdentifier = new Identifier("aString");
				
		assertEquals("Should return null when the identifier is not registered"
				, null, register.resolve(myIdentifier));
	}
	
	@Test
	public void testAdvancedRegistration() {
		TypeEnvironment register = new TypeEnvironment();
		
		Question question = new Question(
				new Identifier("houseValue"), 
				new QLFloat(), 
				new StringLiteral("Value of house"));
		
		register.store(question.getIdentifier(), question.getType());
		
		assertEquals("Should return the QLFLoat type", "float",
				register.resolve(question.getIdentifier()).toString());
	}
	@Test
	public void testDoubleRegistration() {
		TypeEnvironment register = new TypeEnvironment();
		
		Question question = new Question(
				new Identifier("houseValue"), 
				new QLFloat(), 
				new StringLiteral("Value of house"));
		
		Question question2 = new Question(
				new Identifier("carValue"), 
				new QLFloat(), 
				new StringLiteral("Value of car"));
		
		register.store(question.getIdentifier(), question.getType());
		register.store(question2.getIdentifier(), question2.getType());

		assertEquals("Should return the QLFLoat type", "float",
				register.resolve(question.getIdentifier()).toString());
		assertEquals("Should return the QLFLoat type", "float",
				register.resolve(question2.getIdentifier()).toString());
	}
}