gerrit.onap Code Review - sdc.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2019 Samsung Electronics Co., Ltd. All rights reserved.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.openecomp.sdc.be.components.distribution.engine;
  21
  22 import static org.mockito.Mockito.RETURNS_DEFAULTS;
  23 import static org.mockito.Mockito.mock;
  24 import static org.mockito.Mockito.when;
  25 import org.junit.Test;
  26 import org.mockito.invocation.InvocationOnMock;
  27 import org.mockito.stubbing.Answer;
  28 import org.openecomp.sdc.be.model.Service;
  29 import org.openecomp.sdc.be.model.User;
  30
  31 public class AuditDistributionNotificationBuilderTest {
  32
  33
  34      public class SelfReturningAnswer implements Answer<Object>{
  35
  36         public Object answer(InvocationOnMock invocation) throws Throwable {
  37             Object mock = invocation.getMock();
  38             if( invocation.getMethod().getReturnType().isInstance( mock )){
  39                 return mock;
  40             }
  41             else{
  42                 return RETURNS_DEFAULTS.answer(invocation);
  43             }
  44         }
  45     }
  46
  47
  48      private CambriaErrorResponse status;
  49      private Service service;
  50      private User modifier;
  51
  52     @Test
  53     public void testBuilder() {
  54
  55         AuditDistributionNotificationBuilder mockBuilder =
  56                 mock(AuditDistributionNotificationBuilder.class, new SelfReturningAnswer());
  57
  58         when(mockBuilder.setTopicName("topicName").setDistributionId("distributionId").setStatus(status)
  59                 .setService(service).setEnvId("envId").setModifier(modifier).setWorkloadContext("workloadContext")
  60                 .setTenant("tenant")).thenReturn(mockBuilder);
  61         assert mockBuilder.setTopicName("topicName").setDistributionId("distributionId").setStatus(status)
  62                 .setService(service).setEnvId("envId").setModifier(modifier).setWorkloadContext("workloadContext")
  63                 .setTenant("tenant") == mockBuilder;
  64     }
  65 }