AbstractAngularModelCompletionTest.java

/**
 *  Copyright (c) 2013-2016 Angelo ZERR.
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  which accompanies this distribution, and is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  Contributors:
 *  Angelo Zerr <angelo.zerr@gmail.com> - initial API and implementation
 */
package tern.server.protocol.angular;

import org.junit.Assert;
import org.junit.Test;

import tern.TernException;
import tern.angular.AngularType;
import tern.angular.protocol.completions.TernAngularCompletionsQuery;
import tern.server.protocol.TernDoc;
import tern.server.protocol.completions.MockTernCompletionCollector;
import tern.server.protocol.completions.TernCompletionItem;

/**
 * Tests with tern angular model completion.
 * 
 */
public abstract class AbstractAngularModelCompletionTest extends
		AbstractTernServerAngularTest {

	@Test
	public void noCompletionWithTodosAndControllerAs() throws TernException {
		TernDoc doc = createDocForNoCompletionWithTodosAndControllerAs();
		MockTernCompletionCollector collector = new MockTernCompletionCollector();
		server.request(doc, collector);

		Assert.assertTrue(collector.getCompletions().size() == 0);
	}

	private TernDoc createDocForNoCompletionWithTodosAndControllerAs() {
		TernDoc doc = createFile();

		TernAngularCompletionsQuery query = new TernAngularCompletionsQuery(
				AngularType.model);
		query.getScope().getControllers().add("TodoCtrl");
		query.setExpression("xx");

		doc.setQuery(query);
		return doc;
	}

	@Test
	public void completionWithTodosAndControllerAs() throws TernException {
		TernDoc doc = createDocForCompletionWithTodosAndControllerAs();
		MockTernCompletionCollector collector = new MockTernCompletionCollector();
		server.request(doc, collector);

		Assert.assertTrue(collector.getCompletions().size() == 1);
		TernCompletionItem item = collector.get("todos");
		Assert.assertNotNull(item);
	}

	private TernDoc createDocForCompletionWithTodosAndControllerAs() {
		TernDoc doc = createFile();

		TernAngularCompletionsQuery query = new TernAngularCompletionsQuery(
				AngularType.model);
		query.getScope().getControllers().add("TodoCtrl");
		query.setExpression("to");

		doc.setQuery(query);
		return doc;
	}
	@Test
	public void completionWithNgRepeatAndAs() throws TernException {
		TernDoc doc = createDocForCompletionWithNgRepeatAndAs();
		MockTernCompletionCollector collector = new MockTernCompletionCollector();
		server.request(doc, collector);

		Assert.assertTrue(collector.getCompletions().size() == 2);
		Assert.assertNotNull(collector.get("todos"));
		TernCompletionItem atodo = collector.get("atodo");
		Assert.assertNotNull(atodo);
		Assert.assertEquals("{done, text}", atodo.getType());
	}

	private TernDoc createDocForCompletionWithNgRepeatAndAs() {
		TernDoc doc = createFile();

		TernAngularCompletionsQuery query = new TernAngularCompletionsQuery(
				AngularType.model);
		query.getScope().getControllers().add("TodoCtrl");
		query.getScope().addRepeat("atodo in todos");
		query.setExpression("");

		doc.setQuery(query);
		return doc;
	}

	@Test
	public void completionWithNgRepeatAndAsOnTodo() throws TernException {
		TernDoc doc = createDocForCompletionWithNgRepeatAndAsOnTodo();
		MockTernCompletionCollector collector = new MockTernCompletionCollector();
		server.request(doc, collector);

		Assert.assertTrue(collector.getCompletions().size() > 0);
		Assert.assertNotNull(collector.get("toLocaleString"));
		Assert.assertNotNull(collector.get("text"));
		Assert.assertNotNull(collector.get("done"));
	}

	private TernDoc createDocForCompletionWithNgRepeatAndAsOnTodo() {
		TernDoc doc = createFile();

		TernAngularCompletionsQuery query = new TernAngularCompletionsQuery(
				AngularType.model);
		query.getScope().getControllers().add("TodoCtrl");
		query.getScope().addRepeat("atodo in todos");
		query.setExpression("atodo.");

		doc.setQuery(query);
		return doc;
	}

	@Test
	public void completionWithNgRepeatAndAsOnTodoBadAs() throws TernException {
		TernDoc doc = createDocForCompletionWithNgRepeatAndAsOnTodoBadAs();
		MockTernCompletionCollector collector = new MockTernCompletionCollector();
		server.request(doc, collector);

		Assert.assertTrue(collector.getCompletions().size() == 0);
	}

	private TernDoc createDocForCompletionWithNgRepeatAndAsOnTodoBadAs() {
		TernDoc doc = createFile();

		TernAngularCompletionsQuery query = new TernAngularCompletionsQuery(
				AngularType.model);
		query.getScope().getControllers().add("TodoCtrl");
		query.getScope().addRepeat("atodo in t.todos"); // bad 
		query.setExpression("atodo.");

		doc.setQuery(query);
		return doc;
	}

	@Test
	public void completionWithNgRepeatAndAsOnTodoTodoSimple()
			throws TernException {
		TernDoc doc = createDocForCompletionWithNgRepeatAndAsOnTodoSimple();
		MockTernCompletionCollector collector = new MockTernCompletionCollector();
		server.request(doc, collector);

		Assert.assertTrue(collector.getCompletions().size() == 1);
		TernCompletionItem atodo = collector.get("atodo");
		Assert.assertNotNull(atodo);
		Assert.assertEquals("?", atodo.getType());
	}

	private TernDoc createDocForCompletionWithNgRepeatAndAsOnTodoSimple() {
		TernDoc doc = createFile();

		TernAngularCompletionsQuery query = new TernAngularCompletionsQuery(
				AngularType.model);
		query.getScope().getControllers().add("TodoCtrl");
		query.getScope().addRepeat("atodo in t.todos"); // bad 'as' it should be
														// t.todos
		query.setExpression("atod");

		doc.setQuery(query);
		return doc;
	}

	public TernDoc createFile() {
		String name = "myfile.js";
		String text = "function TodoCtrl($scope) {" + "$scope.todos = ["
				+ "{text:'learn angular', done:true},"
				+ "{text:'build an angular app', done:false}];" + "}";

		TernDoc doc = new TernDoc();
		doc.addFile(name, text, null,  null);
		return doc;
	}
}