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

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP : APPC
   4  * ================================================================================
   5  * Copyright (C) 2017 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  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  22  * ============LICENSE_END=========================================================
  23  */
  24
  25 package org.onap.appc.oam.messageadapter;
  26
  27 import org.junit.Assert;
  28 import org.junit.Before;
  29 import org.junit.Test;
  30 import org.junit.runner.RunWith;
  31 import org.onap.appc.adapter.message.MessageAdapterFactory;
  32 import org.onap.appc.adapter.message.Producer;
  33
  34 import org.mockito.Mockito;
  35 import org.osgi.framework.Bundle;
  36 import org.osgi.framework.BundleContext;
  37 import org.osgi.framework.FrameworkUtil;
  38 import org.osgi.framework.ServiceReference;
  39 import org.powermock.api.mockito.PowerMockito;
  40 import org.powermock.core.classloader.annotations.PrepareForTest;
  41 import org.powermock.modules.junit4.PowerMockRunner;
  42 import org.powermock.reflect.Whitebox;
  43
  44 import java.util.HashSet;
  45
  46 import static org.mockito.Mockito.mock;
  47 import static org.powermock.api.mockito.PowerMockito.mockStatic;
  48
  49
  50 @RunWith(PowerMockRunner.class)
  51 @PrepareForTest({MessageAdapter.class, FrameworkUtil.class})
  52 public class MessageAdapterTest {
  53
  54     private Producer fakeProducer;
  55
  56     private MessageAdapter messageAdapter;
  57
  58
  59     @Before
  60     public final void setup() throws Exception {
  61         fakeProducer = mock(Producer.class);
  62         messageAdapter = new MessageAdapter();
  63     }
  64
  65     @Test
  66     public void testGetProducerReturnsNull() throws Exception {
  67         MessageAdapter maSpy = Mockito.spy(messageAdapter);
  68         Mockito.doNothing().when(maSpy).createProducer();
  69
  70         Producer producer = maSpy.getProducer();
  71         Assert.assertTrue("getProducer() did not return null", producer == null);
  72         Producer mySpyProducer = Whitebox.getInternalState(maSpy, "producer");
  73         Assert.assertTrue("MessageAdapter producer is not null", mySpyProducer == null);
  74         Mockito.verify(maSpy, Mockito.times(1)).createProducer();
  75     }
  76
  77     @Test
  78     public void testGetProducerWithExistingProducer() throws Exception {
  79         MessageAdapter maSpy = Mockito.spy(messageAdapter);
  80         Whitebox.setInternalState(maSpy, "producer", fakeProducer);
  81
  82         Producer producer = maSpy.getProducer();
  83         Assert.assertTrue("getProducer() returned null", producer == fakeProducer);
  84         Mockito.verify(maSpy, Mockito.times(0)).createProducer();
  85     }
  86
  87     @Test
  88     public void testGetProducerWithCreateProducer() throws Exception {
  89         MessageAdapter maSpy = Mockito.spy(messageAdapter);
  90         Whitebox.setInternalState(maSpy, "producer", (Object) null);
  91         HashSet<String> pool = new HashSet<>();
  92         Whitebox.setInternalState(maSpy, "pool", pool);
  93
  94         // Prepare all mocks
  95         mockStatic(FrameworkUtil.class);
  96         Bundle maBundle = mock(Bundle.class);
  97         PowerMockito.when(FrameworkUtil.getBundle(MessageAdapter.class)).thenReturn(maBundle);
  98
  99         BundleContext maBundleContext = mock(BundleContext.class);
 100         Mockito.when(maBundle.getBundleContext()).thenReturn(maBundleContext);
 101
 102         ServiceReference svcRef = mock(ServiceReference.class);
 103         Mockito.when(maBundleContext.getServiceReference(MessageAdapterFactory.class.getName())).thenReturn(svcRef);
 104
 105         MessageAdapterFactory maFactory = mock(MessageAdapterFactory.class);
 106         Mockito.when(maBundleContext.getService(svcRef)).thenReturn(maFactory);
 107         Mockito.when(maFactory.createProducer(pool, (String) null, null, null)).thenReturn(fakeProducer);
 108
 109         Producer producer = maSpy.getProducer();
 110         Assert.assertTrue("getProducer() result does not match", producer == fakeProducer);
 111         Producer mySpyProducer = Whitebox.getInternalState(maSpy, "producer");
 112         Assert.assertTrue("MessageAdapter producer does not match",mySpyProducer == fakeProducer);
 113         Mockito.verify(maSpy, Mockito.times(1)).createProducer();
 114     }
 115 }