use of org.nextprot.api.web.service.impl.writer.EntryVelocityBasedStreamWriter in project nextprot-api by calipho-sib.
the class ExportXMLHeaderTest method testXMLExportHeaderRelease.
@Test
public void testXMLExportHeaderRelease() throws Exception {
ByteArrayOutputStream out = new ByteArrayOutputStream();
Writer writer = new PrintWriter(out);
EntryVelocityBasedStreamWriter exporter = new EntryXMLStreamWriter(writer, "overview");
ReleaseInfoVersions rc = new ReleaseInfoVersions();
rc.setApiRelease("api-test-version");
rc.setDatabaseRelease("database-test-version");
Map<String, Object> infos = new HashMap<>();
infos.put(EntryStreamWriter.getReleaseInfoKey(), rc);
exporter.write(Collections.emptyList(), infos);
NodeList dbReleaseNodes = XMLUnitUtils.getMatchingNodes(out.toString(), "//*[local-name()='database-release']");
assertEquals("database-test-version", dbReleaseNodes.item(0).getTextContent());
NodeList apiReleaseNodes = XMLUnitUtils.getMatchingNodes(out.toString(), "//*[local-name()='api-release']");
assertEquals("api-test-version", apiReleaseNodes.item(0).getTextContent());
NodeList entriesCountNode = XMLUnitUtils.getMatchingNodes(out.toString(), "//*[local-name()='number-of-entries']");
assertEquals("0", entriesCountNode.item(0).getTextContent());
}
use of org.nextprot.api.web.service.impl.writer.EntryVelocityBasedStreamWriter in project nextprot-api by calipho-sib.
the class ExportTXTHeaderTest method shouldContainTheNumberOfEntriesInTheHeader.
// TEST for https://issues.isb-sib.ch/browse/CALIPHOMISC-330
@Test
public void shouldContainTheNumberOfEntriesInTheHeader() throws Exception {
ByteArrayOutputStream out = new ByteArrayOutputStream();
Writer writer = new PrintWriter(out);
EntryVelocityBasedStreamWriter exporter = new EntryTXTStreamWriter(writer);
exporter.setEntryBuilderService(entryBuilderMockService);
when(entryBuilderMockService.build(any(EntryConfig.class))).thenReturn(new Entry("NX_1")).thenReturn(new Entry("NX_2"));
Map<String, Object> infos = new HashMap<>();
infos.put(EntryStreamWriter.getReleaseInfoKey(), Mockito.mock(ReleaseInfoVersions.class));
exporter.write(Arrays.asList("NX_1", "NX_2"), infos);
String[] rows = out.toString().split(StringUtils.CR_LF);
assertEquals(rows[0], "#nb entries=2");
assertEquals(rows[1], "NX_1");
assertEquals(rows[2], "NX_2");
}
Aggregations