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