feature-pooling-dmaap/src/test/java/org/onap/policy/drools/pooling/message/LeaderTest.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * ONAP
   4  * ================================================================================
   5  * Copyright (C) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.drools.pooling.message;
  22
  23 import org.junit.Before;
  24 import org.junit.Test;
  25
  26 public class LeaderTest extends SupportMessageWithAssignmentsTester<Leader> {
  27
  28     public LeaderTest() {
  29         super(Leader.class);
  30     }
  31
  32     @Before
  33     public void setUp() {
  34         setNullAssignments(false);
  35     }
  36
  37     /**
  38      * The superclass will already invoke testCheckValidity_InvalidFields() to
  39      * verify that things work with a fully populated message. This verifies
  40      * that it also works if the assignments are null.
  41      *
  42      * @throws Exception if an error occurs
  43      */
  44     @Test
  45     public void testCheckValidity_InvalidFields_NullAssignments() throws Exception {
  46         // null assignments are invalid
  47         expectCheckValidityFailure(msg -> msg.setAssignments(null));
  48     }
  49
  50     @Test
  51     public void testCheckValidity_SourceIsNotLeader() throws Exception {
  52         Leader ldr = makeValidMessage();
  53
  54         ldr.setSource("xyz");
  55
  56         // the source does not have an assignment
  57         BucketAssignments asgnUnassigned = new BucketAssignments(new String[] {"abc", "def"});
  58         expectCheckValidityFailure(msg -> msg.setAssignments(asgnUnassigned));
  59
  60         // the source is not the smallest UUID in this assignment
  61         BucketAssignments asgnNotSmallest = new BucketAssignments(new String[] {VALID_HOST_PREDECESSOR, VALID_HOST});
  62         expectCheckValidityFailure(msg -> msg.setAssignments(asgnNotSmallest));
  63     }
  64
  65     @Override
  66     public Leader makeValidMessage() {
  67         Leader msg = new Leader(VALID_HOST, (isNullAssignments() ? null : VALID_ASGN));
  68         msg.setChannel(VALID_CHANNEL);
  69
  70         return msg;
  71     }
  72
  73 }