appc-oam/appc-oam-bundle/src/test/java/org/onap/appc/oam/messageadapter/MessageAdapterTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP : APPC
   4  * ================================================================================
   5  * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Copyright (C) 2017 Amdocs
   8  * =============================================================================
   9  * Licensed under the Apache License, Version 2.0 (the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *      http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  *
  21  * ============LICENSE_END=========================================================
  22  */
  23
  24 package org.onap.appc.oam.messageadapter;
  25
  26 import org.junit.Assert;
  27 import org.junit.Before;
  28 import org.junit.Test;
  29 import org.junit.runner.RunWith;
  30 import org.onap.appc.adapter.message.MessageAdapterFactory;
  31 import org.onap.appc.adapter.message.Producer;
  32
  33 import org.mockito.Mockito;
  34 import org.osgi.framework.Bundle;
  35 import org.osgi.framework.BundleContext;
  36 import org.osgi.framework.FrameworkUtil;
  37 import org.osgi.framework.ServiceReference;
  38 import org.powermock.api.mockito.PowerMockito;
  39 import org.powermock.core.classloader.annotations.PrepareForTest;
  40 import org.powermock.modules.junit4.PowerMockRunner;
  41 import org.powermock.reflect.Whitebox;
  42
  43 import java.util.HashSet;
  44
  45 import static org.mockito.Mockito.mock;
  46 import static org.powermock.api.mockito.PowerMockito.mockStatic;
  47
  48
  49 @RunWith(PowerMockRunner.class)
  50 @PrepareForTest({MessageAdapter.class, FrameworkUtil.class})
  51 public class MessageAdapterTest {
  52
  53     private Producer fakeProducer;
  54
  55     private MessageAdapter messageAdapter;
  56
  57
  58     @Before
  59     public final void setup() throws Exception {
  60         fakeProducer = mock(Producer.class);
  61         messageAdapter = new MessageAdapter();
  62     }
  63
  64     @Test
  65     public void testGetProducerReturnsNull() throws Exception {
  66         MessageAdapter maSpy = Mockito.spy(messageAdapter);
  67         Mockito.doNothing().when(maSpy).createProducer();
  68
  69         Producer producer = maSpy.getProducer();
  70         Assert.assertTrue("getProducer() did not return null", producer == null);
  71         Producer mySpyProducer = Whitebox.getInternalState(maSpy, "producer");
  72         Assert.assertTrue("MessageAdapter producer is not null", mySpyProducer == null);
  73         Mockito.verify(maSpy, Mockito.times(1)).createProducer();
  74     }
  75
  76     @Test
  77     public void testGetProducerWithExistingProducer() throws Exception {
  78         MessageAdapter maSpy = Mockito.spy(messageAdapter);
  79         Whitebox.setInternalState(maSpy, "producer", fakeProducer);
  80
  81         Producer producer = maSpy.getProducer();
  82         Assert.assertTrue("getProducer() returned null", producer == fakeProducer);
  83         Mockito.verify(maSpy, Mockito.times(0)).createProducer();
  84     }
  85
  86     @Test
  87     public void testGetProducerWithCreateProducer() throws Exception {
  88         MessageAdapter maSpy = Mockito.spy(messageAdapter);
  89         Whitebox.setInternalState(maSpy, "producer", (Object) null);
  90         HashSet<String> pool = new HashSet<>();
  91         Whitebox.setInternalState(maSpy, "pool", pool);
  92
  93         // Prepare all mocks
  94         mockStatic(FrameworkUtil.class);
  95         Bundle maBundle = mock(Bundle.class);
  96         PowerMockito.when(FrameworkUtil.getBundle(MessageAdapter.class)).thenReturn(maBundle);
  97
  98         BundleContext maBundleContext = mock(BundleContext.class);
  99         Mockito.when(maBundle.getBundleContext()).thenReturn(maBundleContext);
 100
 101         ServiceReference svcRef = mock(ServiceReference.class);
 102         Mockito.when(maBundleContext.getServiceReference(MessageAdapterFactory.class.getName())).thenReturn(svcRef);
 103
 104         MessageAdapterFactory maFactory = mock(MessageAdapterFactory.class);
 105         Mockito.when(maBundleContext.getService(svcRef)).thenReturn(maFactory);
 106         Mockito.when(maFactory.createProducer(pool, (String) null, null, null)).thenReturn(fakeProducer);
 107
 108         Producer producer = maSpy.getProducer();
 109         Assert.assertTrue("getProducer() result does not match", producer == fakeProducer);
 110         Producer mySpyProducer = Whitebox.getInternalState(maSpy, "producer");
 111         Assert.assertTrue("MessageAdapter producer does not match",mySpyProducer == fakeProducer);
 112         Mockito.verify(maSpy, Mockito.times(1)).createProducer();
 113     }
 114 }