Initial source code
[oam/tr069-adapter.git] / acs / application-booter / src / test / java / org / commscope / tr069adapter / acs / cpe / test / inform / TransferCompleteTest.java
diff --git a/acs/application-booter/src/test/java/org/commscope/tr069adapter/acs/cpe/test/inform/TransferCompleteTest.java b/acs/application-booter/src/test/java/org/commscope/tr069adapter/acs/cpe/test/inform/TransferCompleteTest.java
new file mode 100644 (file)
index 0000000..1ccd285
--- /dev/null
@@ -0,0 +1,158 @@
+/*\r
+ * ============LICENSE_START========================================================================\r
+ * ONAP : tr-069-adapter\r
+ * =================================================================================================\r
+ * Copyright (C) 2020 CommScope Inc Intellectual Property.\r
+ * =================================================================================================\r
+ * This tr-069-adapter software file is distributed by CommScope Inc under the Apache License,\r
+ * Version 2.0 (the "License"); you may not use this file except in compliance with the License. You\r
+ * may obtain a copy of the License at\r
+ *\r
+ * http://www.apache.org/licenses/LICENSE-2.0\r
+ *\r
+ * This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,\r
+ * either express or implied. See the License for the specific language governing permissions and\r
+ * limitations under the License.\r
+ * ===============LICENSE_END=======================================================================\r
+ */\r
+\r
+package org.commscope.tr069adapter.acs.cpe.test.inform;\r
+\r
+import static org.junit.Assert.assertEquals;\r
+import static org.junit.Assert.fail;\r
+\r
+import javax.servlet.http.Cookie;\r
+\r
+import org.apache.activemq.broker.BrokerService;\r
+import org.apache.commons.httpclient.HttpStatus;\r
+import org.commscope.tr069adapter.acs.booter.ACSServiceBooter;\r
+import org.commscope.tr069adapter.acs.common.DeviceInform;\r
+import org.commscope.tr069adapter.acs.common.dto.TR069DeviceDetails;\r
+import org.commscope.tr069adapter.acs.common.utils.AcsConstants;\r
+import org.commscope.tr069adapter.acs.cpe.handler.DeviceValidator;\r
+import org.commscope.tr069adapter.acs.cpe.rpc.Inform;\r
+import org.commscope.tr069adapter.acs.requestprocessor.DeviceOperationInterface;\r
+import org.commscope.tr069adapter.acs.requestprocessor.dao.DeviceRPCRequestRepository;\r
+import org.commscope.tr069adapter.acs.requestprocessor.dto.SessionDTO;\r
+import org.commscope.tr069adapter.acs.requestprocessor.impl.SessionManager;\r
+import org.commscope.tr069adapter.acs.requestprocessor.impl.TR069EventNotificationService;\r
+import org.junit.After;\r
+import org.junit.Test;\r
+import org.junit.runner.RunWith;\r
+import org.mockito.Mockito;\r
+import org.springframework.beans.factory.annotation.Autowired;\r
+import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;\r
+import org.springframework.boot.test.context.SpringBootTest;\r
+import org.springframework.boot.test.mock.mockito.MockBean;\r
+import org.springframework.http.MediaType;\r
+import org.springframework.mock.web.MockHttpServletResponse;\r
+import org.springframework.test.context.ContextConfiguration;\r
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;\r
+import org.springframework.test.web.servlet.MockMvc;\r
+import org.springframework.test.web.servlet.MvcResult;\r
+import org.springframework.test.web.servlet.RequestBuilder;\r
+import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;\r
+\r
+@SpringBootTest(classes = {ACSServiceBooter.class})\r
+@RunWith(SpringJUnit4ClassRunner.class)\r
+@AutoConfigureMockMvc\r
+@ContextConfiguration\r
+public class TransferCompleteTest {\r
+\r
+  @Autowired\r
+  private MockMvc mockMvc;\r
+\r
+  @MockBean\r
+  private DeviceValidator deviceValidator;\r
+\r
+  @MockBean\r
+  protected DeviceRPCRequestRepository deviceRPCRequestRepository;\r
+\r
+  @MockBean\r
+  private TR069EventNotificationService tr069EventNotificationService;\r
+\r
+  @MockBean\r
+  private DeviceOperationInterface deviceOperationInterface;\r
+\r
+  @MockBean\r
+  private SessionManager sessionManager;\r
+\r
+  @Autowired\r
+  BrokerService broker;\r
+\r
+  @Test\r
+  public void processInformPnPTest() throws Exception {\r
+\r
+    String exampleInform =\r
+        "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?><ns3:Envelope xmlns:ns1=\"http://schemas.xmlsoap.org/soap/encoding/\" xmlns:ns4=\"urn:dslforum-org:cwmp-1-0\" xmlns:ns3=\"http://schemas.xmlsoap.org/soap/envelope/\"><ns3:Header><ns4:ID ns3:mustUnderstand=\"1\">0</ns4:ID></ns3:Header><ns3:Body><ns4:TransferComplete><CommandKey>0005B9AAAA22</CommandKey><FaultStruct><FaultCode>0</FaultCode><FaultString></FaultString></FaultStruct><StartTime>2019-07-08T18:15:43.032Z</StartTime><CompleteTime>2019-07-08T18:15:43.032Z</CompleteTime></ns4:TransferComplete></ns3:Body></ns3:Envelope>";\r
+\r
+    try {\r
+\r
+      SessionDTO session = new SessionDTO();\r
+      session.setDeviceId("0005B9AAAA22");\r
+      session.setSessionId("sessionId");\r
+      session.setCurrentOperationId(10000L);\r
+\r
+      TR069DeviceDetails tr069DeviceDetails = new TR069DeviceDetails();\r
+      tr069DeviceDetails.setDeviceId("0005B9AAAA22");\r
+      tr069DeviceDetails.setConnectionRequestURL("dummyURL");\r
+      tr069DeviceDetails.setSoftwareVersion("10.2.1");\r
+      tr069DeviceDetails.setSoftwareVersion("xxx");\r
+\r
+      Mockito.when(sessionManager.getSessionBySessionId(Mockito.anyString())).thenReturn(session);\r
+      Mockito.when(sessionManager.getLockedSession(Mockito.anyString())).thenReturn(session);\r
+      Mockito.when(sessionManager.updateSession(Mockito.any(SessionDTO.class))).thenReturn(session);\r
+      Mockito.when(sessionManager.generateUniqueSessionID()).thenReturn("newSessionId");\r
+\r
+      Mockito.when(deviceOperationInterface.getDeviceDetails(Mockito.anyString()))\r
+          .thenReturn(tr069DeviceDetails);\r
+\r
+      Mockito.doNothing().when(tr069EventNotificationService)\r
+          .sendDeviceInformToNBI(Mockito.any(DeviceInform.class));\r
+\r
+      Mockito\r
+          .when(deviceValidator.isDeviceAuthorized(Mockito.any(Inform.class), Mockito.anyString()))\r
+          .thenReturn(new Boolean(true));\r
+\r
+      Mockito.when(deviceValidator.validateDevice(Mockito.anyString(), Mockito.anyString(),\r
+          Mockito.anyString())).thenReturn(new Boolean(true));\r
+\r
+      Mockito.when(deviceRPCRequestRepository.findByDeviceIdAndIsProcessed(Mockito.anyString(),\r
+          Mockito.anyInt())).thenReturn(null);\r
+\r
+      RequestBuilder requestBuilder = MockMvcRequestBuilders.post("/CPEMgmt/acs")\r
+          .accept(MediaType.TEXT_PLAIN).content(exampleInform).header("Authorization", "basic")\r
+          .contentType(MediaType.TEXT_PLAIN);\r
+\r
+      MvcResult result = mockMvc.perform(requestBuilder).andReturn();\r
+\r
+      MockHttpServletResponse response = result.getResponse();\r
+      assertEquals(HttpStatus.SC_OK, response.getStatus());\r
+\r
+      Cookie cookie = response.getCookie(AcsConstants.ACS_SESSIONID);\r
+      requestBuilder =\r
+          MockMvcRequestBuilders.post("/CPEMgmt/acs").accept(MediaType.TEXT_PLAIN).content("")\r
+              .header("Authorization", "basic").cookie(cookie).contentType(MediaType.TEXT_PLAIN);\r
+\r
+      result = mockMvc.perform(requestBuilder).andReturn();\r
+\r
+      response = result.getResponse();\r
+      assertEquals(HttpStatus.SC_NO_CONTENT, response.getStatus());\r
+    } catch (Exception e) {\r
+      fail(e.getMessage());\r
+    }\r
+  }\r
+\r
+  @After\r
+  public void stopBroker() throws Exception {\r
+    try {\r
+      System.out.println("Tearing down the broker");\r
+      broker.stop();\r
+      broker.waitUntilStopped();\r
+      broker = null;\r
+    } catch (Exception e) {\r
+      e.printStackTrace();\r
+    }\r
+  }\r
+\r
+}\r