forked from D-Net/dnet-hadoop
52 lines
1.4 KiB
Java
52 lines
1.4 KiB
Java
|
|
package eu.dnetlib.message;
|
|
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
|
|
import java.io.IOException;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
import org.junit.jupiter.api.Test;
|
|
|
|
public class MessageTest {
|
|
|
|
@Test
|
|
public void fromJsonTest() throws IOException {
|
|
Message m = new Message();
|
|
m.setWorkflowId("wId");
|
|
m.setType(MessageType.ONGOING);
|
|
m.setJobName("Collection");
|
|
Map<String, String> body = new HashMap<>();
|
|
body.put("parsedItem", "300");
|
|
body.put("ExecutionTime", "30s");
|
|
|
|
m.setBody(body);
|
|
System.out.println("m = " + m);
|
|
Message m1 = Message.fromJson(m.toString());
|
|
assertEquals(m1.getWorkflowId(), m.getWorkflowId());
|
|
assertEquals(m1.getType(), m.getType());
|
|
assertEquals(m1.getJobName(), m.getJobName());
|
|
|
|
assertNotNull(m1.getBody());
|
|
m1.getBody().keySet().forEach(it -> assertEquals(m1.getBody().get(it), m.getBody().get(it)));
|
|
assertEquals(m1.getJobName(), m.getJobName());
|
|
}
|
|
|
|
@Test
|
|
public void toStringTest() {
|
|
final String expectedJson = "{\"workflowId\":\"wId\",\"jobName\":\"Collection\",\"type\":\"ONGOING\",\"body\":{\"ExecutionTime\":\"30s\",\"parsedItem\":\"300\"}}";
|
|
Message m = new Message();
|
|
m.setWorkflowId("wId");
|
|
m.setType(MessageType.ONGOING);
|
|
m.setJobName("Collection");
|
|
Map<String, String> body = new HashMap<>();
|
|
body.put("parsedItem", "300");
|
|
body.put("ExecutionTime", "30s");
|
|
|
|
m.setBody(body);
|
|
|
|
assertEquals(expectedJson, m.toString());
|
|
}
|
|
}
|