GoogleSpreadsheetRepositoryTest.java

package org.molgenis.googlespreadsheet;

import com.google.gdata.client.spreadsheet.SpreadsheetService;
import com.google.gdata.data.IFeed;
import com.google.gdata.data.TextConstruct;
import com.google.gdata.data.spreadsheet.*;
import com.google.gdata.util.ServiceException;
import org.molgenis.data.Entity;
import org.molgenis.data.meta.model.Attribute;
import org.molgenis.data.meta.model.AttributeFactory;
import org.molgenis.data.meta.model.EntityType;
import org.molgenis.data.meta.model.EntityTypeFactory;
import org.molgenis.test.data.AbstractMolgenisSpringTest;
import org.springframework.beans.factory.annotation.Autowired;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import java.io.IOException;
import java.net.URL;
import java.util.*;

import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import static org.testng.Assert.*;

public class GoogleSpreadsheetRepositoryTest extends AbstractMolgenisSpringTest
{
	@Autowired
	private EntityTypeFactory entityTypeFactory;

	@Autowired
	private AttributeFactory attrMetaFactory;

	private GoogleSpreadsheetRepository spreadsheetRepository;
	private SpreadsheetService spreadsheetService;
	private ListFeed listFeed;
	private CellFeed cellFeed;

	@BeforeMethod
	public void setUp() throws IOException, ServiceException
	{
		spreadsheetService = mock(SpreadsheetService.class);
		listFeed = mock(ListFeed.class);
		TextConstruct textConstruct = when(mock(TextConstruct.class).getPlainText()).thenReturn("name").getMock();
		when(listFeed.getTitle()).thenReturn(textConstruct);
		List<ListEntry> entries = new ArrayList<>();
		ListEntry entry = mock(ListEntry.class);
		CustomElementCollection customElements = mock(CustomElementCollection.class);
		when(customElements.getTags()).thenReturn(new LinkedHashSet<>(Arrays.asList("col1", "col2", "col3")));
		when(customElements.getValue("col1")).thenReturn("val1");
		when(customElements.getValue("col2")).thenReturn("val2");
		when(customElements.getValue("col3")).thenReturn("val3");
		when(entry.getCustomElements()).thenReturn(customElements);
		entries.add(entry);
		when(listFeed.getEntries()).thenReturn(entries);
		cellFeed = mock(CellFeed.class);
		List<CellEntry> cells = new ArrayList<>();

		Cell cell1 = mock(Cell.class);
		when(cell1.getRow()).thenReturn(1);
		when(cell1.getValue()).thenReturn("col1");
		Cell cell2 = mock(Cell.class);
		when(cell2.getRow()).thenReturn(1);
		when(cell2.getValue()).thenReturn("col2");
		Cell cell3 = mock(Cell.class);
		when(cell3.getRow()).thenReturn(1);
		when(cell3.getValue()).thenReturn("col3");
		CellEntry entry1 = when(mock(CellEntry.class).getCell()).thenReturn(cell1).getMock();
		CellEntry entry2 = when(mock(CellEntry.class).getCell()).thenReturn(cell2).getMock();
		CellEntry entry3 = when(mock(CellEntry.class).getCell()).thenReturn(cell3).getMock();
		cells.add(entry1);
		cells.add(entry2);
		cells.add(entry3);
		when(cellFeed.getEntries()).thenReturn(cells);
		when(cellFeed.getTitle()).thenReturn(textConstruct);
		spreadsheetRepository = new GoogleSpreadsheetRepository(spreadsheetService, "key", "id", entityTypeFactory,
				attrMetaFactory);
	}

	@Test(expectedExceptions = NullPointerException.class)
	public void GoogleSpreadsheetRepository() throws IOException, ServiceException
	{
		try (GoogleSpreadsheetRepository repo = new GoogleSpreadsheetRepository(null, null, null, null, null))
		{
		}
	}

	@SuppressWarnings("unchecked")
	@Test
	public void iterator() throws IOException, ServiceException
	{
		when(spreadsheetService.getFeed(any(URL.class), (Class<IFeed>) any(Class.class))).thenReturn(cellFeed)
				.thenReturn(listFeed);
		Iterator<Entity> it = spreadsheetRepository.iterator();
		assertTrue(it.hasNext());
		Entity entity = it.next();
		assertEquals(entity.getString("col1"), "val1");
		assertEquals(entity.getString("col2"), "val2");
		assertEquals(entity.getString("col3"), "val3");
		assertFalse(it.hasNext());
	}

	@SuppressWarnings("unchecked")
	@Test
	public void getEntityType() throws IOException, ServiceException
	{
		when(spreadsheetService.getFeed(any(URL.class), (Class<CellFeed>) any(Class.class))).thenReturn(cellFeed);
		EntityType entityType = spreadsheetRepository.getEntityType();
		assertEquals(entityType.getName(), "name");
		Iterator<Attribute> it = entityType.getAttributes().iterator();
		assertEquals(it.next().getName(), "col1");
		assertTrue(it.hasNext());
		assertEquals(it.next().getName(), "col2");
		assertTrue(it.hasNext());
		assertEquals(it.next().getName(), "col3");
		assertFalse(it.hasNext());
	}
}