src/test/java/org/onap/dmaap/mr/test/dme2/DME2ProducerTest.java

   1 /*******************************************************************************
   2  *  ============LICENSE_START=======================================================
   3  *  org.onap.dmaap
   4  *  ================================================================================
   5  *  Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  *  ================================================================================
   7  *  Licensed under the Apache License, Version 2.0 (the "License");
   8  *  you may not use this file except in compliance with the License.
   9  *  You may obtain a copy of the License at
  10  *        http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  *  Unless required by applicable law or agreed to in writing, software
  13  *  distributed under the License is distributed on an "AS IS" BASIS,
  14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  *  See the License for the specific language governing permissions and
  16  *  limitations under the License.
  17  *  ============LICENSE_END=========================================================
  18  *
  19  *  ECOMP is a trademark and service mark of AT&T Intellectual Property.
  20  *
  21  *******************************************************************************/
  22 package org.onap.dmaap.mr.test.dme2;
  23
  24 import com.att.aft.dme2.api.DME2Client;
  25 import com.att.aft.dme2.api.DME2Exception;
  26 import com.att.aft.dme2.internal.jackson.map.ObjectMapper;
  27 import java.net.URI;
  28 import java.net.URISyntaxException;
  29 import java.util.HashMap;
  30 import java.util.Properties;
  31 import junit.framework.TestCase;
  32 import org.apache.logging.log4j.LogManager;
  33 import org.apache.logging.log4j.Logger;
  34
  35 public class DME2ProducerTest extends TestCase {
  36
  37     private static final Logger LOGGER = LogManager.getLogger(DME2ProducerTest.class);
  38
  39     public void testProducer() {
  40         DME2TopicTest topicTestObj = new DME2TopicTest();
  41         Properties props = LoadPropertyFile.getPropertyFileDataProducer();
  42         String latitude = props.getProperty("Latitude");
  43         String longitude = props.getProperty("Longitude");
  44         String version = props.getProperty("Version");
  45         String serviceName = props.getProperty("ServiceName");
  46         String env = props.getProperty("Environment");
  47         String partner = props.getProperty("Partner");
  48         String protocol = props.getProperty("Protocol");
  49         String url =
  50             protocol + "://DME2SEARCH/" + "service=" + serviceName + "/" + "version=" + version
  51                 + "/"
  52                 + "envContext=" + env + "/" + "partner=" + partner;
  53         LoadPropertyFile.loadAFTProperties(latitude, longitude);
  54         HashMap<String, String> hm = new HashMap<String, String>();
  55         hm.put("AFT_DME2_EP_READ_TIMEOUT_MS", "50000");
  56         hm.put("AFT_DME2_ROUNDTRIP_TIMEOUT_MS", "240000");
  57         hm.put("AFT_DME2_EP_CONN_TIMEOUT", "5000");
  58         // checking whether topic exist or not
  59         if (!topicTestObj.topicExist(url, props, hm)) {
  60             // if topic doesn't exist then create the topic
  61             topicTestObj.createTopic(url, props, hm);
  62             // after creating the topic publish on that topic
  63             publishMessage(url, props, hm);
  64         } else {
  65             // if topic already exist start publishing on the topic
  66             publishMessage(url, props, hm);
  67         }
  68
  69     }
  70
  71     public void publishMessage(String url, Properties props, HashMap<String, String> mapData) {
  72         try {
  73             LOGGER.info("Call to publish message ");
  74             DME2Client sender = new DME2Client(new URI(url), 5000L);
  75             sender.setAllowAllHttpReturnCodes(true);
  76             sender.setMethod(props.getProperty("MethodTypePost"));
  77             String subcontextpathPublish =
  78                 props.getProperty("SubContextPathproducer") + props.getProperty("newTopic");
  79             sender.setSubContext(subcontextpathPublish);
  80             String jsonStringApiBean = new ObjectMapper()
  81                 .writeValueAsString(new ApiKeyBean("example@att.com",
  82                     "description"));
  83             sender.setPayload(jsonStringApiBean);
  84
  85             sender.setCredentials(props.getProperty("user"), props.getProperty("password"));
  86             sender.addHeader("content-type", props.getProperty("contenttype"));
  87             LOGGER.info("Publishing message");
  88             String reply = sender.sendAndWait(5000L);
  89             // assertTrue(LoadPropertyFile.isValidJsonString(reply));
  90             assertNotNull(reply);
  91             LOGGER.info("response =" + reply);
  92         } catch (DME2Exception e) {
  93             e.printStackTrace();
  94         } catch (URISyntaxException e) {
  95             e.printStackTrace();
  96         } catch (Exception e) {
  97             e.printStackTrace();
  98         }
  99     }
 100 }