gerrit.onap Code Review - policy/common.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * policy-endpoints
   4  * ================================================================================
   5  * Copyright (C) 2018-2020 AT&T Intellectual Property. All rights reserved.
   6  * Modifications Copyright (C) 2024 Nordix Foundation.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *      http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.policy.common.endpoints.event.comm.bus.internal;
  23
  24 import static org.assertj.core.api.Assertions.assertThatCode;
  25 import static org.junit.Assert.assertFalse;
  26 import static org.junit.Assert.assertTrue;
  27 import static org.mockito.ArgumentMatchers.anyString;
  28 import static org.mockito.Mockito.doThrow;
  29 import static org.mockito.Mockito.mock;
  30 import static org.mockito.Mockito.verify;
  31 import static org.mockito.Mockito.when;
  32
  33 import com.att.nsa.cambria.client.CambriaBatchingPublisher;
  34 import java.io.IOException;
  35 import org.junit.Before;
  36 import org.junit.Test;
  37 import org.onap.policy.common.endpoints.event.comm.bus.TopicTestBase;
  38 import org.onap.policy.common.endpoints.event.comm.bus.internal.BusPublisher.CambriaPublisherWrapper;
  39
  40
  41 public class BusPublisherTest extends TopicTestBase {
  42
  43     @Before
  44     @Override
  45     public void setUp() {
  46         super.setUp();
  47     }
  48
  49     @Test
  50     public void testCambriaPublisherWrapper() {
  51         // verify that different wrappers can be built
  52         new CambriaPublisherWrapper(makeBuilder().build());
  53         new CambriaPublisherWrapper(makeBuilder().useHttps(false).build());
  54         new CambriaPublisherWrapper(makeBuilder().useHttps(true).build());
  55         new CambriaPublisherWrapper(makeBuilder().useHttps(true).allowSelfSignedCerts(false).build());
  56         new CambriaPublisherWrapper(makeBuilder().useHttps(true).allowSelfSignedCerts(true).build());
  57         new CambriaPublisherWrapper(makeBuilder().apiKey(null).build());
  58         new CambriaPublisherWrapper(makeBuilder().apiSecret(null).build());
  59         new CambriaPublisherWrapper(makeBuilder().apiKey(null).apiSecret(null).build());
  60         new CambriaPublisherWrapper(makeBuilder().userName(null).build());
  61         new CambriaPublisherWrapper(makeBuilder().password(null).build());
  62         assertThatCode(() -> new CambriaPublisherWrapper(makeBuilder().userName(null).password(null).build()))
  63                         .doesNotThrowAnyException();
  64     }
  65
  66     @Test
  67     public void testCambriaPublisherWrapperSend() throws Exception {
  68         CambriaBatchingPublisher pub = mock(CambriaBatchingPublisher.class);
  69         CambriaPublisherWrapper cambria = new CambriaPublisherWrapper(makeBuilder().build());
  70         cambria.publisher = pub;
  71
  72         assertTrue(cambria.send(MY_PARTITION, MY_MESSAGE));
  73
  74         // publisher exception
  75         when(pub.send(anyString(), anyString())).thenThrow(new IOException(EXPECTED));
  76         assertFalse(cambria.send(MY_PARTITION2, MY_MESSAGE2));
  77     }
  78
  79     @Test(expected = IllegalArgumentException.class)
  80     public void testCambriaPublisherWrapperSend_InvalidMsg() {
  81         CambriaPublisherWrapper cambria = new CambriaPublisherWrapper(makeBuilder().build());
  82         cambria.publisher = mock(CambriaBatchingPublisher.class);
  83
  84         cambria.send(MY_PARTITION, null);
  85     }
  86
  87     @Test
  88     public void testCambriaPublisherWrapperClose() {
  89         CambriaBatchingPublisher pub = mock(CambriaBatchingPublisher.class);
  90         CambriaPublisherWrapper cambria = new CambriaPublisherWrapper(makeBuilder().build());
  91         cambria.publisher = pub;
  92
  93         cambria.close();
  94         verify(pub).close();
  95
  96         // try again, this time with an exception
  97         doThrow(new RuntimeException(EXPECTED)).when(pub).close();
  98         cambria.close();
  99     }
 100 }