Issue71_todos.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 Issue71_todos 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().setModule("TodoModule");
		query.getScope().getControllers().add("TodoCtrl as t");
		query.setExpression("to");

		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().setModule("TodoModule");
		query.getScope().getControllers().add("TodoCtrl as t");
		query.setExpression("t.to");

		doc.setQuery(query);
		return doc;
	}

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

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

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

		TernAngularCompletionsQuery query = new TernAngularCompletionsQuery(
				AngularType.model);
		query.getScope().setModule("TodoModule");
		query.getScope().getControllers().add("TodoCtrl as t");
		query.setExpression("");

		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);
		TernCompletionItem t = collector.get("t");
		Assert.assertNotNull(t);
		TernCompletionItem todo = collector.get("todo");
		Assert.assertNotNull(todo);
		Assert.assertEquals("{done, text}", todo.getType());
	}

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

		TernAngularCompletionsQuery query = new TernAngularCompletionsQuery(
				AngularType.model);
		query.getScope().setModule("TodoModule");
		query.getScope().getControllers().add("TodoCtrl as t");
		query.getScope().addRepeat("todo in t.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().setModule("TodoModule");
		query.getScope().getControllers().add("TodoCtrl as t");
		query.getScope().addRepeat("todo in t.todos");
		query.setExpression("todo.");

		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().setModule("TodoModule");
		query.getScope().getControllers().add("TodoCtrl as t");
		query.getScope().addRepeat("todo in todos"); // bad 'as' it should be
														// t.todos
		query.setExpression("todo.");

		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 todo = collector.get("todo");
		Assert.assertNotNull(todo);
		Assert.assertEquals("{done, text}", todo.getType());
	}

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

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

		doc.setQuery(query);
		return doc;
	}

	public TernDoc createFile() {
		String name = "myfile.js";
		String text = "var app = angular.module('TodoModule',[ ]);\n"
				+ "app.controller('TodoCtrl', function(){\n"
				+ "this.todos = [{text:'learn angular', done:true},\n"
				+ "{text:'build an angular app', done:false}];\n" 
				+ "});";

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

}