@Before
public void setUp() throws Exception {
persistedEntry = new PersistedEntry();
persistedEntry.setFeed(FEED_NAME);
persistedEntry.setEntryId(MARKER_ID);
persistedEntry.setEntryBody(ENTRY_BODY);
emptyList = new ArrayList<PersistedEntry>();
entryList = new ArrayList<PersistedEntry>();
entryList.add(persistedEntry);
// Mocks
abdera = mock(Abdera.class);
getFeedRequest = mock(GetFeedRequest.class);
getEntryRequest = mock(GetEntryRequest.class);
jdbcTemplate = mock(JdbcTemplate.class);
postgresFeedSource = new JdbcFeedSource();
postgresFeedSource.setJdbcTemplate(jdbcTemplate);
// Mock GetEntryRequest
when(getEntryRequest.getFeedName()).thenReturn(FEED_NAME);
when(getEntryRequest.getEntryId()).thenReturn(MARKER_ID);
// Mock GetFeedRequest
when(getFeedRequest.getFeedName()).thenReturn(FEED_NAME);
when(getFeedRequest.getPageSize()).thenReturn("25");
when(getFeedRequest.getAbdera()).thenReturn(abdera);
}
@Test
public void shouldGetEntry() throws Exception {
Abdera localAbdera = new Abdera();
when(jdbcTemplate.query(any(String.class), any(Object[].class), any(EntryRowMapper.class)))
.thenReturn(entryList);
when(getEntryRequest.getAbdera()).thenReturn(localAbdera);
assertEquals(
"Should get a 200 response",
HttpStatus.OK,
postgresFeedSource.getEntry(getEntryRequest).getResponseStatus());
}
@Test
public void shouldGetFeedWithMarkerBackward() throws Exception {
when(getFeedRequest.getPageMarker()).thenReturn(MARKER_ID);
when(getFeedRequest.getDirection()).thenReturn(BACKWARD);
Abdera localAbdera = new Abdera();
when(jdbcTemplate.queryForObject(
any(String.class), any(EntryRowMapper.class), any(String.class), any(String.class)))
.thenReturn(persistedEntry);
when(getFeedRequest.getAbdera()).thenReturn(localAbdera);
when(getEntryRequest.getAbdera()).thenReturn(localAbdera);
when(jdbcTemplate.query(any(String.class), any(Object[].class), any(EntryRowMapper.class)))
.thenReturn(entryList);
assertEquals(
"Should get a 200 response",
HttpStatus.OK,
postgresFeedSource.getFeed(getFeedRequest).getResponseStatus());
}
@Test
public void shouldGetFeedHeadWithCategory() throws Exception {
Abdera localAbdera = new Abdera();
when(getFeedRequest.getSearchQuery()).thenReturn(SINGLE_CAT);
when(jdbcTemplate.queryForObject(
any(String.class), any(EntryRowMapper.class), any(String.class), any(String.class)))
.thenReturn(persistedEntry);
when(getFeedRequest.getAbdera()).thenReturn(localAbdera);
when(getEntryRequest.getAbdera()).thenReturn(localAbdera);
when(jdbcTemplate.query(any(String.class), any(Object[].class), any(EntryRowMapper.class)))
.thenReturn(entryList);
when(jdbcTemplate.queryForInt(any(String.class), any(Object[].class))).thenReturn(1);
assertEquals(
"Should get a 200 response",
HttpStatus.OK,
postgresFeedSource.getFeed(getFeedRequest).getResponseStatus());
}