v3po/vpp-cfg-init/src/test/java/io/fd/honeycomb/v3po/vpp/data/init/InitializerRegistryImplTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.honeycomb.v3po.vpp.data.init;
  18
  19 import static org.mockito.Mockito.verify;
  20 import static org.mockito.MockitoAnnotations.initMocks;
  21
  22 import java.util.Arrays;
  23 import org.junit.Before;
  24 import org.junit.Test;
  25 import org.mockito.Mock;
  26
  27 public class InitializerRegistryImplTest {
  28
  29     @Mock(name="dti1")
  30     private DataTreeInitializer dti1;
  31     @Mock(name="dti2")
  32     private DataTreeInitializer dti2;
  33     @Mock(name="dti3")
  34     private DataTreeInitializer dti3;
  35
  36     private InitializerRegistryImpl initializerRegistry;
  37
  38     @Before
  39     public void setUp() throws Exception {
  40         initMocks(this);
  41         initializerRegistry = new InitializerRegistryImpl(Arrays.asList(dti1, dti2, dti3));
  42     }
  43
  44     @Test(expected = IllegalArgumentException.class)
  45     public void testConstructorFailed() throws Exception {
  46         new InitializerRegistryImpl(Arrays.asList(dti1, null));
  47     }
  48
  49     @Test
  50     public void testInitialize() throws Exception {
  51         initializerRegistry.initialize();
  52
  53         verify(dti1).initialize();
  54         verify(dti2).initialize();
  55         verify(dti3).initialize();
  56     }
  57
  58     @Test
  59     public void testClose() throws Exception {
  60         initializerRegistry.close();
  61
  62         verify(dti1).close();
  63         verify(dti2).close();
  64         verify(dti3).close();
  65     }
  66 }