extension/openflowplugin-extension-onf/src/test/java/org/opendaylight/openflowplugin/extension/onf/OnfExtensionProviderTest.java

   1 /*
   2  * Copyright (c) 2017 Pantheon Technologies s.r.o. 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 package org.opendaylight.openflowplugin.extension.onf;
   9
  10 import org.junit.Before;
  11 import org.junit.Test;
  12 import org.junit.runner.RunWith;
  13 import org.mockito.Mock;
  14 import org.mockito.Mockito;
  15 import org.mockito.runners.MockitoJUnitRunner;
  16 import org.opendaylight.openflowjava.protocol.spi.connection.SwitchConnectionProvider;
  17 import org.opendaylight.openflowplugin.extension.api.ExtensionConverterRegistrator;
  18 import org.opendaylight.openflowplugin.extension.api.OpenFlowPluginExtensionRegistratorProvider;
  19 import org.opendaylight.openflowplugin.extension.api.TypeVersionKey;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.experimenter.types.rev151020.experimenter.core.message.ExperimenterMessageOfChoice;
  21
  22 @RunWith(MockitoJUnitRunner.class)
  23 public class OnfExtensionProviderTest {
  24
  25     @Mock
  26     private SwitchConnectionProvider switchConnectionProvider;
  27     @Mock
  28     private OpenFlowPluginExtensionRegistratorProvider openFlowPluginExtensionRegistratorProvider;
  29     @Mock
  30     private ExtensionConverterRegistrator extensionConverterRegistrator;
  31
  32     private OnfExtensionProvider onfExtensionProvider;
  33
  34     @Before
  35     public void setUp() throws Exception {
  36         Mockito
  37                 .when(openFlowPluginExtensionRegistratorProvider.getExtensionConverterRegistrator())
  38                 .thenReturn(extensionConverterRegistrator);
  39
  40         onfExtensionProvider =
  41                 new OnfExtensionProvider(switchConnectionProvider, openFlowPluginExtensionRegistratorProvider);
  42     }
  43
  44     @Test
  45     public void init() throws Exception {
  46         onfExtensionProvider.init();
  47         Mockito.verify(switchConnectionProvider, Mockito.times(2))
  48                 .registerExperimenterMessageSerializer(Mockito.any(), Mockito.any());
  49         Mockito.verify(switchConnectionProvider)
  50                 .registerExperimenterMessageDeserializer(Mockito.any(), Mockito.any());
  51         Mockito.verify(switchConnectionProvider)
  52                 .registerErrorDeserializer(Mockito.any(), Mockito.any());
  53         Mockito.verify(extensionConverterRegistrator, Mockito.times(2))
  54                 .registerMessageConvertor(Mockito.<TypeVersionKey<? extends ExperimenterMessageOfChoice>>any(),
  55                         Mockito.any());
  56     }
  57
  58 }