catalog-be/src/test/java/org/openecomp/sdc/be/components/distribution/engine/AuditDistributionNotificationBuilderTest.java

   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 org.junit.Test;
  23 import org.mockito.invocation.InvocationOnMock;
  24 import org.mockito.stubbing.Answer;
  25 import org.openecomp.sdc.be.model.Service;
  26 import org.openecomp.sdc.be.model.User;
  27
  28 import static org.mockito.Mockito.RETURNS_DEFAULTS;
  29 import static org.mockito.Mockito.mock;
  30 import static org.mockito.Mockito.when;
  31
  32 public class AuditDistributionNotificationBuilderTest {
  33
  34
  35      public class SelfReturningAnswer implements Answer<Object>{
  36
  37         public Object answer(InvocationOnMock invocation) throws Throwable {
  38             Object mock = invocation.getMock();
  39             if( invocation.getMethod().getReturnType().isInstance( mock )){
  40                 return mock;
  41             }
  42             else{
  43                 return RETURNS_DEFAULTS.answer(invocation);
  44             }
  45         }
  46     }
  47
  48
  49      private CambriaErrorResponse status;
  50      private Service service;
  51      private User modifier;
  52
  53     @Test
  54     public void testBuilder() {
  55
  56         AuditDistributionNotificationBuilder mockBuilder =
  57                 mock(AuditDistributionNotificationBuilder.class, new SelfReturningAnswer());
  58
  59         when(mockBuilder.setTopicName("topicName").setDistributionId("distributionId").setStatus(status)
  60                 .setService(service).setEnvId("envId").setModifier(modifier).setWorkloadContext("workloadContext")
  61                 .setTenant("tenant")).thenReturn(mockBuilder);
  62         assert mockBuilder.setTopicName("topicName").setDistributionId("distributionId").setStatus(status)
  63                 .setService(service).setEnvId("envId").setModifier(modifier).setWorkloadContext("workloadContext")
  64                 .setTenant("tenant") == mockBuilder;
  65     }
  66 }