-
Notifications
You must be signed in to change notification settings - Fork 40
/
ActiveMQTransportsTest.java
84 lines (67 loc) · 3.05 KB
/
ActiveMQTransportsTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// Copyright (c) Committed Software 2018, opensource@committed.io
package uk.gov.dstl.baleen.transports.activemq;
import static uk.gov.dstl.baleen.uima.BaleenCollectionReader.KEY_CONTENT_EXTRACTOR;
import java.io.IOException;
import org.apache.uima.UIMAException;
import org.apache.uima.analysis_engine.AnalysisEngine;
import org.apache.uima.fit.factory.AnalysisEngineFactory;
import org.apache.uima.fit.factory.CollectionReaderFactory;
import org.apache.uima.fit.factory.ExternalResourceFactory;
import org.apache.uima.resource.ExternalResourceDescription;
import org.apache.uima.resource.ResourceInitializationException;
import org.junit.Test;
import uk.gov.dstl.baleen.resources.SharedActiveMQResource;
import uk.gov.dstl.baleen.transports.util.FakeBaleenContentExtractor;
import uk.gov.dstl.baleen.transports.util.TransportTester;
import uk.gov.dstl.baleen.uima.BaleenCollectionReader;
import uk.gov.dstl.baleen.uima.utils.TypeSystemSingleton;
public class ActiveMQTransportsTest {
private static String PROTOCOL_VALUE = "vm";
private static String BROKERARGS_VALUE = "broker.persistent=false";
private final ExternalResourceDescription mqerd =
ExternalResourceFactory.createNamedResourceDescription(
SharedActiveMQResource.RESOURCE_KEY,
SharedActiveMQResource.class,
SharedActiveMQResource.PARAM_PROTOCOL,
PROTOCOL_VALUE,
SharedActiveMQResource.PARAM_BROKERARGS,
BROKERARGS_VALUE);
private final ExternalResourceDescription ceerd =
ExternalResourceFactory.createNamedResourceDescription(
KEY_CONTENT_EXTRACTOR, FakeBaleenContentExtractor.class);
@Test
public void testTransportCanSendAndRecieve() throws UIMAException, IOException {
AnalysisEngine sender = createAnalysisEngine();
BaleenCollectionReader receiver = createReciever();
TransportTester tester = new TransportTester(sender, receiver);
tester.run();
tester.assertCompleteMatch();
}
@Test
public void testTransportCanClose() throws UIMAException, IOException {
AnalysisEngine sender = createAnalysisEngine();
BaleenCollectionReader receiver = createReciever();
TransportTester tester = new TransportTester(sender, receiver);
tester.close();
}
private BaleenCollectionReader createReciever() throws ResourceInitializationException {
return (BaleenCollectionReader)
CollectionReaderFactory.createReader(
ActiveMQTransportReceiver.class,
TypeSystemSingleton.getTypeSystemDescriptionInstance(),
SharedActiveMQResource.RESOURCE_KEY,
mqerd,
KEY_CONTENT_EXTRACTOR,
ceerd);
}
private AnalysisEngine createAnalysisEngine() throws ResourceInitializationException {
return createAnalysisEngine(SharedActiveMQResource.RESOURCE_KEY, mqerd);
}
private AnalysisEngine createAnalysisEngine(Object... args)
throws ResourceInitializationException {
return AnalysisEngineFactory.createEngine(
ActiveMQTransportSender.class,
TypeSystemSingleton.getTypeSystemDescriptionInstance(),
args);
}
}