javatests/org/eclipse/lsp4j/util/DocumentSymbolsTest.java

/******************************************************************************
 * Copyright (c) 2018 TypeFox and others.
 * 
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0,
 * or the Eclipse Distribution License v. 1.0 which is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 * 
 * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause
 ******************************************************************************/
package org.eclipse.lsp4j.test.util;

import static java.util.stream.Collectors.toList;

import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;

import org.eclipse.lsp4j.DocumentSymbol;
import org.eclipse.lsp4j.util.DocumentSymbols;
import org.junit.Assert;
import org.junit.Test;

public class DocumentSymbolsTest {

	@Test(expected = NullPointerException.class)
	public void asIterator_null() {
		DocumentSymbols.asIterator(null);
	}

	@Test
	public void asIterator() {
		DocumentSymbol depth0 = new DocumentSymbol();
		depth0.setName("root");

		DocumentSymbol depth1_0 = new DocumentSymbol();
		depth1_0.setName("A1");
		DocumentSymbol depth1_1 = new DocumentSymbol();
		depth1_1.setName("B1");
		DocumentSymbol depth1_2 = new DocumentSymbol();
		depth1_2.setName("C1");
		depth0.setChildren(Arrays.asList(depth1_0, depth1_1, depth1_2));

		DocumentSymbol depth2_0_0 = new DocumentSymbol();
		depth2_0_0.setName("A11");
		DocumentSymbol depth2_0_1 = new DocumentSymbol();
		depth2_0_1.setName("A12");
		depth1_0.setChildren(Arrays.asList(depth2_0_0, depth2_0_1));

		DocumentSymbol depth2_1_0 = new DocumentSymbol();
		depth2_1_0.setName("B11");
		DocumentSymbol depth2_1_1 = new DocumentSymbol();
		depth2_1_1.setName("B12");
		depth1_1.setChildren(Arrays.asList(depth2_1_0, depth2_1_1));

		DocumentSymbol depth2_2_0 = new DocumentSymbol();
		depth2_2_0.setName("C11");
		DocumentSymbol depth2_2_1 = new DocumentSymbol();
		depth2_2_1.setName("C12");
		depth1_2.setChildren(Arrays.asList(depth2_2_0, depth2_2_1));

		Iterator<DocumentSymbol> iterator = DocumentSymbols.asIterator(depth0);
		Iterable<DocumentSymbol> iterable = () -> iterator;
		Stream<DocumentSymbol> stream = StreamSupport.stream(iterable.spliterator(), false);
		List<String> actual = stream.map(symbol -> symbol.getName()).collect(toList());
		List<String> expected = Arrays.asList("root, A1", "B1", "C1", "A11", "A12", "B11", "B12", "C11", "C12");

		Assert.assertEquals(Arrays.toString(expected.toArray()), Arrays.toString(actual.toArray()));
	}

}