artifactbroker/plugins/reception-plugins/src/test/java/org/onap/policy/distribution/reception/handling/sdc/TestComponentDoneStatusMessage.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2018 Intel. 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
  21 package org.onap.policy.distribution.reception.handling.sdc;
  22
  23 import static org.junit.Assert.assertEquals;
  24
  25 import org.junit.Test;
  26 import org.onap.sdc.utils.DistributionStatusEnum;
  27
  28 public class TestComponentDoneStatusMessage {
  29
  30     @Test
  31     public void testComponentDoneStatusMessage_Success() {
  32         final String consumerId = "dummyId";
  33         final String distributionId = "dummyDistribution";
  34         final long timestamp = System.currentTimeMillis();
  35         final ComponentDoneStatusMessageBuilder messageBuilder =
  36                 new ComponentDoneStatusMessageBuilder().setConsumerId(consumerId).setDistributionId(distributionId)
  37                         .setDistributionStatus(DistributionStatusEnum.COMPONENT_DONE_OK).setTimestamp(timestamp);
  38         final ComponentDoneStatusMessage message = new ComponentDoneStatusMessage(messageBuilder);
  39         assertEquals("POLICY", message.getComponentName());
  40         assertEquals(consumerId, message.getConsumerID());
  41         assertEquals(distributionId, message.getDistributionID());
  42         assertEquals(DistributionStatusEnum.COMPONENT_DONE_OK, message.getStatus());
  43         assertEquals(timestamp, message.getTimestamp());
  44     }
  45
  46     @Test
  47     public void testComponentDoneStatusMessage_Failure() {
  48         final String consumerId = "dummyId";
  49         final String distributionId = "dummyDistribution";
  50         final long timestamp = System.currentTimeMillis();
  51         final ComponentDoneStatusMessageBuilder messageBuilder =
  52                 new ComponentDoneStatusMessageBuilder().setConsumerId(consumerId).setDistributionId(distributionId)
  53                         .setDistributionStatus(DistributionStatusEnum.COMPONENT_DONE_ERROR).setTimestamp(timestamp);
  54         final ComponentDoneStatusMessage message = new ComponentDoneStatusMessage(messageBuilder);
  55         assertEquals("POLICY", message.getComponentName());
  56         assertEquals(consumerId, message.getConsumerID());
  57         assertEquals(distributionId, message.getDistributionID());
  58         assertEquals(DistributionStatusEnum.COMPONENT_DONE_ERROR, message.getStatus());
  59         assertEquals(timestamp, message.getTimestamp());
  60     }
  61 }