1 files changed, 8 insertions, 5 deletions
diff --git a/src/test/java/org/onap/a1pesimulator/service/VesBrokerServiceImplTest.java b/src/test/java/org/onap/a1pesimulator/service/VesBrokerServiceImplTest.java
index 5b27529..2c2d9ff 100644
--- a/src/test/java/org/onap/a1pesimulator/service/VesBrokerServiceImplTest.java
+++ b/src/test/java/org/onap/a1pesimulator/service/VesBrokerServiceImplTest.java
@@ -15,19 +15,20 @@ package org.onap.a1pesimulator.service;
 
 import static org.mockito.Mockito.when;
 
-import com.fasterxml.jackson.databind.ObjectMapper;
 import java.io.IOException;
 import java.net.URISyntaxException;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.ArgumentMatchers;
 import org.mockito.Mock;
-import org.onap.a1pesimulator.data.ves.Event;
+import org.onap.a1pesimulator.data.ReportingMethodEnum;
+import org.onap.a1pesimulator.data.ves.VesEvent;
 import org.onap.a1pesimulator.service.ves.RanVesBrokerService;
 import org.onap.a1pesimulator.service.ves.RanVesSender;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -40,6 +41,8 @@ import org.springframework.test.context.junit4.SpringRunner;
 import org.springframework.test.util.ReflectionTestUtils;
 import org.springframework.web.client.RestTemplate;
 
+import com.fasterxml.jackson.databind.ObjectMapper;
+
 @RunWith(SpringRunner.class)
 @SpringBootTest
 public class VesBrokerServiceImplTest {
@@ -68,13 +71,13 @@ public class VesBrokerServiceImplTest {
                 ArgumentMatchers.any(HttpEntity.class), ArgumentMatchers.eq(String.class))).thenReturn(responseEntity);
 
         ResponseEntity<String> response = vesBrokerService.startSendingVesEvents("CustomIdentifier",
-                loadEventFromFile("VesBrokerControllerTest_pm_ves.json"), 10);
+                loadEventFromFile("VesBrokerControllerTest_pm_ves.json"), 10, ReportingMethodEnum.VES);
 
         Assert.assertEquals(HttpStatus.ACCEPTED, response.getStatusCode());
     }
 
-    private Event loadEventFromFile(String fileName) throws Exception {
-        return mapper.readValue(loadFileContent(fileName), Event.class);
+    private VesEvent loadEventFromFile(String fileName) throws Exception {
+        return mapper.readValue(loadFileContent(fileName), VesEvent.class);
     }
 
     private String loadFileContent(String fileName) throws IOException, URISyntaxException {