X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=webapp-backend%2Fsrc%2Fmain%2Fjava%2Forg%2Foransc%2Fric%2Fportal%2Fdashboard%2Fconfig%2FE2ManagerMockConfiguration.java;h=d2921482140b8812c72c2ccbb6dd63fdb19501d7;hb=refs%2Fchanges%2F13%2F113%2F1;hp=3df6dc70d188b18ce0321b24e9c969b0689a2196;hpb=f660cae7a447b60d84ef75f7c2bcbf62412d4579;p=portal%2Fric-dashboard.git

diff --git a/webapp-backend/src/main/java/org/oransc/ric/portal/dashboard/config/E2ManagerMockConfiguration.java b/webapp-backend/src/main/java/org/oransc/ric/portal/dashboard/config/E2ManagerMockConfiguration.java
index 3df6dc70..d2921482 100644
--- a/webapp-backend/src/main/java/org/oransc/ric/portal/dashboard/config/E2ManagerMockConfiguration.java
+++ b/webapp-backend/src/main/java/org/oransc/ric/portal/dashboard/config/E2ManagerMockConfiguration.java
@@ -1,6 +1,6 @@
 /*-
  * ========================LICENSE_START=================================
- * ORAN-OSC
+ * O-RAN-SC
  * %%
  * Copyright (C) 2019 AT&T Intellectual Property and Nokia
  * %%
@@ -26,9 +26,7 @@ import static org.mockito.Mockito.when;
 
 import java.lang.invoke.MethodHandles;
 
-import org.oransc.ric.e2mgr.client.api.EndcSetupRequestApi;
-import org.oransc.ric.e2mgr.client.api.HealthCheckApi;
-import org.oransc.ric.e2mgr.client.api.X2SetupRequestApi;
+import org.oransc.ric.e2mgr.client.api.E2ManagerApi;
 import org.oransc.ric.e2mgr.client.invoker.ApiClient;
 import org.oransc.ric.e2mgr.client.model.SetupRequest;
 import org.slf4j.Logger;
@@ -39,8 +37,7 @@ import org.springframework.context.annotation.Profile;
 import org.springframework.http.HttpStatus;
 
 /**
- * Creates an implementation of the E2 manager client that answers requests with
- * mock data.
+ * Creates a mock implementation of the E2 manager client API.
  */
 @Profile("mock")
 @Configuration
@@ -59,32 +56,23 @@ public class E2ManagerMockConfiguration {
 	}
 
 	@Bean
-	public EndcSetupRequestApi endcSetupRequestApi() {
+	public E2ManagerApi e2ManagerApi() {
 		ApiClient apiClient = apiClient();
-		EndcSetupRequestApi mockApi = mock(EndcSetupRequestApi.class);
+		E2ManagerApi mockApi = mock(E2ManagerApi.class);
 		when(mockApi.getApiClient()).thenReturn(apiClient);
-		return mockApi;
-	}
 
-	@Bean
-	public HealthCheckApi healthCheckApi() {
-		ApiClient apiClient = apiClient();
-		HealthCheckApi mockApi = mock(HealthCheckApi.class);
-		when(mockApi.getApiClient()).thenReturn(apiClient);
 		doAnswer(i -> {
 			return null;
 		}).when(mockApi).healthCheck();
-		return mockApi;
-	}
 
-	@Bean
-	public X2SetupRequestApi x2SetupRequestApi() {
-		ApiClient apiClient = apiClient();
-		X2SetupRequestApi mockApi = mock(X2SetupRequestApi.class);
-		when(mockApi.getApiClient()).thenReturn(apiClient);
+		doAnswer(i -> {
+			return null;
+		}).when(mockApi).endcSetup(any(SetupRequest.class));
+
 		doAnswer(i -> {
 			return null;
 		}).when(mockApi).setup(any(SetupRequest.class));
+
 		return mockApi;
 	}