Update published event to include header and body
[aai/gizmo.git] / src / test / java / org / onap / crud / event / GraphEventEnvelopeTest.java
1 package org.onap.crud.event;
2
3 import static org.hamcrest.Matchers.is;
4 import static org.junit.Assert.assertThat;
5 import org.junit.Test;
6 import org.onap.crud.entity.Vertex;
7 import org.onap.crud.event.GraphEvent.GraphEventOperation;
8 import org.onap.crud.event.envelope.GraphEventEnvelope;
9 import org.onap.crud.test.util.TestUtil;
10 import org.skyscreamer.jsonassert.Customization;
11 import org.skyscreamer.jsonassert.JSONAssert;
12 import org.skyscreamer.jsonassert.JSONCompareMode;
13 import org.skyscreamer.jsonassert.comparator.CustomComparator;
14 import com.google.gson.Gson;
15 import com.google.gson.JsonElement;
16
17 public class GraphEventEnvelopeTest {
18
19     @Test
20     public void testPublishedEventFormat() throws Exception {
21         String expectedEnvelope = TestUtil.getFileAsString("event/event-envelope.json");
22
23         GraphEvent body = GraphEvent.builder(GraphEventOperation.CREATE)
24                 .vertex(GraphEventVertex.fromVertex(new Vertex.Builder("pserver").build(), "v13")).build();
25         String graphEventEnvelope = new GraphEventEnvelope(body).toJson();
26
27         JSONAssert.assertEquals(expectedEnvelope, graphEventEnvelope,
28                 new CustomComparator(JSONCompareMode.STRICT, new Customization("header.request-id", (o1, o2) -> true),
29                         new Customization("header.timestamp", (o1, o2) -> true),
30                         new Customization("body.timestamp", (o1, o2) -> true),
31                         new Customization("body.transaction-id", (o1, o2) -> true)));
32
33         Gson gson = new Gson();
34         GraphEventEnvelope envelope = gson.fromJson(graphEventEnvelope, GraphEventEnvelope.class);
35         assertThat(envelope.getHeader().getRequestId(), is(envelope.getBody().getTransactionId()));
36     }
37
38     @Test
39     public void testConsumedEventFormat() throws Exception {
40         String expectedEnvelope = TestUtil.getFileAsString("event/event-envelope-sentinel.json");
41         Gson gson = new Gson();
42         GraphEventEnvelope envelope = gson.fromJson(expectedEnvelope, GraphEventEnvelope.class);
43         JsonElement jsonElement = envelope.getPolicyViolations().getAsJsonArray().get(0);
44
45         assertThat(jsonElement.getAsJsonObject().get("summary").getAsString(), is("a summary"));
46         assertThat(jsonElement.getAsJsonObject().get("policyName").getAsString(), is("a policy name"));
47     }
48 }