applications/notification-supplier/src/test/java/org/opendaylight/openflowplugin/applications/notification/supplier/NotificationProviderImplTest.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.openflowplugin.applications.notification.supplier;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.mockito.Mockito.mock;
  13 import java.util.List;
  14 import org.junit.Before;
  15 import org.junit.Test;
  16 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  17 import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
  18 import org.opendaylight.openflowplugin.applications.notification.supplier.tools.NotificationProviderConfig;
  19 import org.opendaylight.openflowplugin.applications.notification.supplier.tools.NotificationProviderConfig.NotificationProviderConfigBuilder;
  20
  21 public class NotificationProviderImplTest {
  22
  23     private NotificationProviderService notificationProviderService;
  24     private NotificationProviderConfig config;
  25     private DataBroker dataBroker;
  26
  27     @Before
  28     public void initialization() {
  29         dataBroker = mock(DataBroker.class);
  30         notificationProviderService = mock(NotificationProviderService.class);
  31     }
  32
  33     @Test
  34     public void testCreateAllSuppliers() {
  35         final NotificationProviderConfig config = createAllConfigSupplier();
  36         final NotificationProviderImpl provider = new NotificationProviderImpl(config, notificationProviderService, dataBroker);
  37         provider.start();
  38         final List<NotificationSupplierDefinition<?>> listSuppliers = provider.getSupplierList();
  39         int nrOfSuppliers = 0;
  40         for (final NotificationSupplierDefinition<?> supplier : listSuppliers) {
  41             if (supplier != null) {
  42                 nrOfSuppliers++;
  43             }
  44         }
  45         assertEquals(11, nrOfSuppliers);
  46     }
  47
  48     @Test
  49     public void testCreateRootSuppliersOnly() {
  50         final NotificationProviderConfig config = createNonConfigSupplier();
  51         final NotificationProviderImpl provider = new NotificationProviderImpl(config, notificationProviderService, dataBroker);
  52         provider.start();
  53         final List<NotificationSupplierDefinition<?>> listSuppliers = provider.getSupplierList();
  54         int nrOfSuppliers = 0;
  55         for (final NotificationSupplierDefinition<?> supplier : listSuppliers) {
  56             if (supplier != null) {
  57                 nrOfSuppliers++;
  58             }
  59         }
  60         assertEquals(2, nrOfSuppliers);
  61     }
  62
  63     private NotificationProviderConfig createAllConfigSupplier() {
  64         final NotificationProviderConfigBuilder builder = new NotificationProviderConfigBuilder();
  65         builder.setFlowStatSupport(true);
  66         builder.setFlowSupport(true);
  67         builder.setFlowTableStatSupport(true);
  68         builder.setGroupStatSupport(true);
  69         builder.setGroupSupport(true);
  70         builder.setMeterStatSupport(true);
  71         builder.setMeterSupport(true);
  72         builder.setNodeConnectorStatSupport(true);
  73         builder.setQueueStatSupport(true);
  74         return builder.build();
  75     }
  76
  77     private NotificationProviderConfig createNonConfigSupplier() {
  78         final NotificationProviderConfigBuilder builder = new NotificationProviderConfigBuilder();
  79         builder.setFlowStatSupport(false);
  80         builder.setFlowSupport(false);
  81         builder.setFlowTableStatSupport(false);
  82         builder.setGroupStatSupport(false);
  83         builder.setGroupSupport(false);
  84         builder.setMeterStatSupport(false);
  85         builder.setMeterSupport(false);
  86         builder.setNodeConnectorStatSupport(false);
  87         builder.setQueueStatSupport(false);
  88         return builder.build();
  89     }
  90 }