85 lines
2.7 KiB
Java
85 lines
2.7 KiB
Java
|
|
package eu.dnetlib.dhp.broker.oa;
|
|
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.List;
|
|
|
|
import org.junit.jupiter.api.BeforeEach;
|
|
import org.junit.jupiter.api.Test;
|
|
|
|
import eu.dnetlib.dhp.broker.model.Event;
|
|
import eu.dnetlib.dhp.broker.model.MappedFields;
|
|
import eu.dnetlib.dhp.broker.model.Subscription;
|
|
import eu.dnetlib.dhp.broker.oa.util.NotificationGroup;
|
|
|
|
class IndexNotificationsJobTest {
|
|
|
|
private List<Subscription> subscriptions;
|
|
|
|
@BeforeEach
|
|
void setUp() throws Exception {
|
|
final Subscription s = new Subscription();
|
|
s.setTopic("ENRICH/MISSING/PID");
|
|
s.setConditions("[{\"field\":\"targetDatasourceName\",\"fieldType\":\"STRING\",\"operator\":\"EXACT\",\"listParams\":[{\"value\":\"reposiTUm\"}]},{\"field\":\"trust\",\"fieldType\":\"FLOAT\",\"operator\":\"RANGE\",\"listParams\":[{\"value\":\"0\",\"otherValue\":\"1\"}]}]");
|
|
subscriptions = Arrays.asList(s);
|
|
IndexNotificationsJob.initConditionsForSubscriptions(subscriptions);
|
|
}
|
|
|
|
@Test
|
|
void testGenerateNotifications_invalid_topic() {
|
|
final Event event = new Event();
|
|
event.setTopic("ENRICH/MISSING/PROJECT");
|
|
|
|
for (int i = 0; i < 10; i++) {
|
|
final long start = System.currentTimeMillis();
|
|
final NotificationGroup res = IndexNotificationsJob.generateNotifications(event, subscriptions, 0);
|
|
final long end = System.currentTimeMillis();
|
|
|
|
System.out.println("no topic - execution time (ms): " + (end - start));
|
|
|
|
assertEquals(0, res.getData().size());
|
|
}
|
|
}
|
|
|
|
@Test
|
|
void testGenerateNotifications_topic_match() {
|
|
final Event event = new Event();
|
|
event.setTopic("ENRICH/MISSING/PID");
|
|
event.setMap(new MappedFields());
|
|
event.getMap().setTargetDatasourceName("reposiTUm");
|
|
event.getMap().setTrust(0.8f);
|
|
|
|
for (int i = 0; i < 10; i++) {
|
|
final long start = System.currentTimeMillis();
|
|
final NotificationGroup res = IndexNotificationsJob.generateNotifications(event, subscriptions, 0);
|
|
final long end = System.currentTimeMillis();
|
|
|
|
System.out.println("topic match - execution time (ms): " + (end - start));
|
|
|
|
assertEquals(1, res.getData().size());
|
|
}
|
|
}
|
|
|
|
@Test
|
|
void testGenerateNotifications_topic_no_match() {
|
|
final Event event = new Event();
|
|
event.setTopic("ENRICH/MISSING/PID");
|
|
event.setMap(new MappedFields());
|
|
event.getMap().setTargetDatasourceName("Puma");
|
|
event.getMap().setTrust(0.8f);
|
|
|
|
for (int i = 0; i < 10; i++) {
|
|
final long start = System.currentTimeMillis();
|
|
final NotificationGroup res = IndexNotificationsJob.generateNotifications(event, subscriptions, 0);
|
|
final long end = System.currentTimeMillis();
|
|
|
|
System.out.println("topic no match - execution time (ms): " + (end - start));
|
|
|
|
assertEquals(0, res.getData().size());
|
|
}
|
|
}
|
|
|
|
}
|