openflowplugin/src/test/java/org/opendaylight/openflowplugin/openflow/md/core/sal/OpenflowPluginProviderTest.java

   1 /*\r
   2  * Copyright (c) 2014 Pantheon Technologies s.r.o. and others. All rights reserved.\r
   3  *\r
   4  * This program and the accompanying materials are made available under the\r
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html\r
   7  */\r
   8 \r
   9 package org.opendaylight.openflowplugin.openflow.md.core.sal;\r
  10 \r
  11 \r
  12 import com.google.common.util.concurrent.Futures;
  13 import java.util.ArrayList;\r
  14 import org.junit.Assert;\r
  15 import org.junit.Before;\r
  16 import org.junit.Test;\r
  17 import org.mockito.Mock;\r
  18 import org.mockito.Mockito;
  19 import org.mockito.MockitoAnnotations;\r
  20 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  21 import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
  22 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
  23 import org.opendaylight.openflowjava.protocol.spi.connection.SwitchConnectionProvider;\r
  24 \r
  25 /**\r
  26  * @author michal.polkorab\r
  27  *\r
  28  */\r
  29 public class OpenflowPluginProviderTest {\r
  30 \r
  31     @Mock
  32     DataBroker dataBroker;
  33     @Mock
  34     NotificationProviderService notificationProviderService;
  35     @Mock
  36     RpcProviderRegistry rpcProviderRegistry;
  37     @Mock
  38     SwitchConnectionProvider switchProvider;
  39 \r
  40     OpenflowPluginProvider provider = new OpenflowPluginProvider();\r
  41 \r
  42     /**\r
  43      * Initializes mocks\r
  44      */\r
  45     @Before\r
  46     public void startUp() {\r
  47         MockitoAnnotations.initMocks(this);\r
  48         provider = new OpenflowPluginProvider();\r
  49     }\r
  50 \r
  51     /**\r
  52      * Test {@link OpenflowPluginProvider#initialization()}\r
  53      */\r
  54     @Test\r
  55     public void testInitialization() {\r
  56         Mockito.when(switchProvider.startup()).thenReturn(Futures.immediateFuture(Boolean.TRUE));
  57         ArrayList<SwitchConnectionProvider> switchProviders = new ArrayList<>();\r
  58         switchProviders.add(switchProvider);\r
  59         provider.setSwitchConnectionProviders(switchProviders);\r
  60         provider.setDataBroker(dataBroker);
  61         provider.setNotificationService(notificationProviderService);
  62         provider.setRpcRegistry(rpcProviderRegistry);
  63         provider.initialization();\r
  64 \r
  65         Assert.assertNotNull("Wrong message count dumper", provider.getMessageCountDumper());\r
  66         Assert.assertNotNull("Wrong extension converter registrator", provider.getExtensionConverterRegistrator());\r
  67         Assert.assertNotNull("Wrong data broker", provider.getDataBroker());
  68         Assert.assertNotNull("Wrong notification service", provider.getNotificationService());
  69         Assert.assertNotNull("Wrong rpc registry", provider.getRpcRegistry());
  70     }\r
  71 }