Map identities to proper objects

[mdsal.git] / binding / mdsal-binding-dom-adapter / src / test / java / org / opendaylight / mdsal / binding / dom / adapter / BindingDOMAdapterLoaderTest.java

diff --git a/binding/mdsal-binding-dom-adapter/src/test/java/org/opendaylight/mdsal/binding/dom/adapter/BindingDOMAdapterLoaderTest.java b/binding/mdsal-binding-dom-adapter/src/test/java/org/opendaylight/mdsal/binding/dom/adapter/BindingDOMAdapterLoaderTest.java
index 71f2f5644e122f0e13e39f829dcd9b387db880b3..49b7c807b87a46efd5dda9d3619549127213fb54 100644 (file)
--- a/binding/mdsal-binding-dom-adapter/src/test/java/org/opendaylight/mdsal/binding/dom/adapter/BindingDOMAdapterLoaderTest.java
+++ b/binding/mdsal-binding-dom-adapter/src/test/java/org/opendaylight/mdsal/binding/dom/adapter/BindingDOMAdapterLoaderTest.java
@@ -10,61 +10,60 @@ package org.opendaylight.mdsal.binding.dom.adapter;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;
-import static org.mockito.MockitoAnnotations.initMocks;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.mock;
 
-import javax.annotation.Nullable;
+import com.google.common.collect.ImmutableClassToInstanceMap;
 import org.junit.Before;
 import org.junit.Test;
+import org.junit.runner.RunWith;
 import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnitRunner;
 import org.opendaylight.mdsal.binding.api.DataBroker;
-import org.opendaylight.mdsal.binding.generator.impl.GeneratedClassLoadingStrategy;
+import org.opendaylight.mdsal.binding.api.TransactionChainListener;
 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
 import org.opendaylight.mdsal.dom.api.DOMService;
-import org.opendaylight.yangtools.binding.data.codec.impl.BindingNormalizedNodeCodecRegistry;
 
+@RunWith(MockitoJUnitRunner.StrictStubs.class)
 public class BindingDOMAdapterLoaderTest {
 
     @Mock
     private DOMDataBroker domService;
-
     @Mock
-    private BindingNormalizedNodeCodecRegistry mockCodecRegistry;
+    private AdapterContext mockContext;
 
     private BindingDOMAdapterLoader bindingDOMAdapterLoader;
     private BindingDOMDataBrokerAdapter bindingDOMDataBrokerAdapter;
 
     @Before
-    public void setUp() throws Exception {
-        initMocks(this);
-        bindingDOMAdapterLoader = new BindingDOMAdapterLoader(
-                new BindingToNormalizedNodeCodec(GeneratedClassLoadingStrategy.getTCCLClassLoadingStrategy(),
-                        mockCodecRegistry)) {
-            @Nullable
+    public void setUp() {
+        doReturn(ImmutableClassToInstanceMap.of()).when(domService).getExtensions();
+        bindingDOMAdapterLoader = new BindingDOMAdapterLoader(mockContext) {
             @Override
-            protected DOMService getDelegate(Class<? extends DOMService> reqDeleg) {
+            protected DOMService getDelegate(final Class<? extends DOMService> reqDeleg) {
                 return domService;
             }
         };
     }
 
     @Test
-    public void createBuilderTest() throws Exception {
+    public void createBuilderTest() {
         assertTrue(bindingDOMAdapterLoader.load(DataBroker.class).get() instanceof BindingDOMDataBrokerAdapter);
         domService = null;
         assertFalse(bindingDOMAdapterLoader.load(DataBroker.class).isPresent());
     }
 
     @Test
-    public void createChainTest() throws Exception {
+    public void createChainTest() {
         bindingDOMDataBrokerAdapter
                 = (BindingDOMDataBrokerAdapter) bindingDOMAdapterLoader.load(DataBroker.class).get();
-        assertNotNull(bindingDOMDataBrokerAdapter.createTransactionChain(null));
+        assertNotNull(bindingDOMDataBrokerAdapter.createTransactionChain(mock(TransactionChainListener.class)));
     }
 
     @Test(expected = UnsupportedOperationException.class)
-    public void registerWithException() throws Exception {
+    public void registerWithException() {
         bindingDOMDataBrokerAdapter
                 = (BindingDOMDataBrokerAdapter) bindingDOMAdapterLoader.load(DataBroker.class).get();
-        bindingDOMDataBrokerAdapter.registerDataTreeChangeListener(null,null);
+        bindingDOMDataBrokerAdapter.registerDataTreeChangeListener(null, null);
     }
 }
\ No newline at end of file