Add unit tests for router-core
[aai/router-core.git] / src / test / java / org / onap / aai / rest / RestClientTest.java
diff --git a/src/test/java/org/onap/aai/rest/RestClientTest.java b/src/test/java/org/onap/aai/rest/RestClientTest.java
new file mode 100644 (file)
index 0000000..4ee7e58
--- /dev/null
@@ -0,0 +1,99 @@
+/**
+ * ============LICENSE_START=======================================================
+ * org.onap.aai
+ * ================================================================================
+ * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
+ * Copyright © 2017 Amdocs
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ *
+ * ECOMP is a trademark and service mark of AT&T Intellectual Property.
+ */
+package org.onap.aai.rest;
+
+import static org.junit.Assert.assertTrue;
+
+import java.io.PrintWriter;
+import java.io.StringWriter;
+import java.lang.reflect.Method;
+import java.util.List;
+import java.util.Map;
+
+import org.apache.camel.Exchange;
+import org.apache.camel.Message;
+import org.junit.Before;
+import org.junit.Test;
+import org.onap.aai.restclient.client.Headers;
+
+public class RestClientTest {
+
+  /**
+   * Test case initialization
+   * 
+   * @throws Exception the exception
+   */
+  @Before
+  public void init() throws Exception {
+  }
+  
+  @Test
+  public void validate() throws Exception {
+
+    try {
+      RestClientComponent rc = new RestClientComponent();
+      RestClientEndpoint endpoint = new RestClientEndpoint("http://host.com:8443/endpoint", rc);
+
+      endpoint.setEcompClientCert("client-cert");
+      endpoint.setEcompKeystore("keystore");
+      endpoint.setEcompKeystorePassword("pwd");
+      endpoint.setOp("operation");
+
+      assertTrue(endpoint.getEcompClientCert().compareTo("client-cert") == 0);
+      assertTrue(endpoint.getEcompKeystore().compareTo("keystore") == 0);
+      assertTrue(endpoint.getEcompKeystorePassword().compareTo("pwd") == 0);
+      assertTrue(endpoint.getOp().compareTo("operation") == 0);
+      assertTrue(endpoint.isSingleton());
+      
+      RestClientProducer producer = (RestClientProducer)endpoint.createProducer();
+      assertTrue(producer != null);
+      
+      Method method = RestClientProducer.class.getDeclaredMethod("populateRestHeaders", Exchange.class);
+      method.setAccessible(true);
+      
+      Exchange exchange = endpoint.createExchange();
+      Message in = exchange.getIn();
+      in.setHeader(RestClientEndpoint.IN_HEADER_URL, "svc/endpoint");
+      in.setHeader(Headers.FROM_APP_ID, "val1");
+      in.setHeader(Headers.TRANSACTION_ID, "val2");
+      in.setHeader(Headers.RESOURCE_VERSION, "val2");
+      in.setHeader(Headers.ETAG, "val2");
+      in.setHeader(Headers.IF_MATCH, "val2");
+      in.setHeader(Headers.IF_NONE_MATCH, "val2");
+      in.setHeader(Headers.ACCEPT, "val2");
+      in.setHeader("Content-Type", "val2");
+      
+      Map<String, List<String>> headers = (Map<String, List<String>>)method.invoke(producer, exchange); 
+      assertTrue(headers.size() == 8);  
+    }
+    catch (Exception ex) {
+      StringWriter writer = new StringWriter();
+      PrintWriter printWriter = new PrintWriter( writer );
+      ex.printStackTrace( printWriter );
+      printWriter.flush();
+      System.out.println(writer.toString());
+      throw ex;
+    }
+  }
+    
+}