gerrit.fd Code Review - hc2vpp.git/blob

   1 /*
   2  * Copyright (c) 2018 Bell Canada, Pantheon Technologies 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.hc2vpp.fib.management;
  18
  19 import static org.hamcrest.CoreMatchers.is;
  20 import static org.hamcrest.CoreMatchers.not;
  21 import static org.hamcrest.Matchers.empty;
  22 import static org.junit.Assert.assertEquals;
  23 import static org.junit.Assert.assertNotNull;
  24 import static org.junit.Assert.assertThat;
  25 import static org.junit.Assert.assertTrue;
  26 import static org.mockito.MockitoAnnotations.initMocks;
  27
  28 import com.google.inject.Guice;
  29 import com.google.inject.Inject;
  30 import com.google.inject.name.Named;
  31 import com.google.inject.testing.fieldbinder.Bind;
  32 import com.google.inject.testing.fieldbinder.BoundFieldModule;
  33 import io.fd.hc2vpp.fib.management.read.FibManagementReaderFactory;
  34 import io.fd.hc2vpp.fib.management.write.FibManagementWriterFactory;
  35 import io.fd.honeycomb.translate.impl.read.registry.CompositeReaderRegistryBuilder;
  36 import io.fd.honeycomb.translate.impl.write.registry.FlatWriterRegistryBuilder;
  37 import io.fd.honeycomb.translate.read.ReaderFactory;
  38 import io.fd.honeycomb.translate.util.YangDAG;
  39 import io.fd.honeycomb.translate.write.WriterFactory;
  40 import io.fd.vpp.jvpp.core.future.FutureJVppCore;
  41 import java.util.HashSet;
  42 import java.util.Set;
  43 import org.junit.Before;
  44 import org.junit.Test;
  45 import org.mockito.Mock;
  46 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  47
  48 public class FibManagementModuleTest {
  49
  50     @Named("honeycomb-context")
  51     @Bind
  52     @Mock
  53     private DataBroker honeycombContext;
  54
  55     @Named("honeycomb-initializer")
  56     @Bind
  57     @Mock
  58     private DataBroker honeycombInitializer;
  59
  60     @Bind
  61     @Mock
  62     private FutureJVppCore futureJVppCore;
  63
  64     @Inject
  65     private Set<ReaderFactory> readerFactories = new HashSet<>();
  66
  67     @Inject
  68     private Set<WriterFactory> writerFactories = new HashSet<>();
  69
  70     @Before
  71     public void setUp() {
  72         initMocks(this);
  73         Guice.createInjector(new FibManagementModule(), BoundFieldModule.of(this)).injectMembers(this);
  74     }
  75
  76     @Test
  77     public void testReaderFactories() {
  78         assertThat(readerFactories, is(not(empty())));
  79
  80         // Test registration process (all dependencies present, topological order of readers does exist, etc.)
  81         final CompositeReaderRegistryBuilder registryBuilder = new CompositeReaderRegistryBuilder(new YangDAG());
  82         readerFactories.forEach(factory -> factory.init(registryBuilder));
  83         assertNotNull(registryBuilder.build());
  84         assertEquals(1, readerFactories.size());
  85         assertTrue(readerFactories.iterator().next() instanceof FibManagementReaderFactory);
  86     }
  87
  88     @Test
  89     public void testWriterFactories() {
  90         assertThat(writerFactories, is(not(empty())));
  91
  92         // Test registration process (all dependencies present, topological order of writers does exist, etc.)
  93         final FlatWriterRegistryBuilder registryBuilder = new FlatWriterRegistryBuilder(new YangDAG());
  94         writerFactories.forEach(factory -> factory.init(registryBuilder));
  95         assertNotNull(registryBuilder.build());
  96         assertEquals(1, writerFactories.size());
  97         assertTrue(writerFactories.iterator().next() instanceof FibManagementWriterFactory);
  98     }
  99 }