Convert DataChangeListeners to DataTreeChangeListeners

[nemo.git] / nemo-renderers / openflow-renderer / src / test / java / org / opendaylight / nemo / renderer / openflow / physicalnetwork / OFLinkListenerTest.java

diff --git a/nemo-renderers/openflow-renderer/src/test/java/org/opendaylight/nemo/renderer/openflow/physicalnetwork/OFLinkListenerTest.java b/nemo-renderers/openflow-renderer/src/test/java/org/opendaylight/nemo/renderer/openflow/physicalnetwork/OFLinkListenerTest.java
index fcb69c4a7f52513f7403e55844e454c213d16052..532232d752861181ffc3452b9dfa42e3d88b2eb7 100644 (file)
--- a/nemo-renderers/openflow-renderer/src/test/java/org/opendaylight/nemo/renderer/openflow/physicalnetwork/OFLinkListenerTest.java
+++ b/nemo-renderers/openflow-renderer/src/test/java/org/opendaylight/nemo/renderer/openflow/physicalnetwork/OFLinkListenerTest.java
@@ -7,46 +7,41 @@
  */
 package org.opendaylight.nemo.renderer.openflow.physicalnetwork;
 
-import org.junit.Assert;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+
+import java.util.Collections;
 import org.junit.Before;
 import org.junit.Test;
-import org.opendaylight.nemo.renderer.openflow.physicalnetwork.OFLinkListener;
-
-import static org.junit.Assert.*;
-import static org.mockito.Mockito.*;
-import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
-import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
-import org.opendaylight.nemo.renderer.openflow.physicalnetwork.PhysicalNetworkAdapter;
+import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
+import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Link;
-import org.opendaylight.yangtools.yang.binding.DataObject;
-import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
-import java.util.Map;
-import java.util.Set;
 /**
  * Created by zhangmeng on 2015/11/19.
  */
-public class OFLinkListenerTest extends OFLinkListener {
-    public OFLinkListenerTest(){
-        super(mock(PhysicalNetworkAdapter.class));
-    }
-    private AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> asyncDataChangeEvent;
+public class OFLinkListenerTest {
+    private final PhysicalNetworkAdapter mockAdapter = mock(PhysicalNetworkAdapter.class);
+    private OFLinkListener listener;
 
     @Before
     public void setUp() throws Exception {
-        asyncDataChangeEvent = mock(AsyncDataChangeEvent.class);
+        listener = new OFLinkListener(mockAdapter);
     }
 
+    @SuppressWarnings("unchecked")
     @Test
-    public void testOnDataChanged() throws Exception {
-        this.onDataChanged(null);
-        this.onDataChanged(asyncDataChangeEvent);
-        verify(asyncDataChangeEvent).getCreatedData();
-        verify(asyncDataChangeEvent).getUpdatedData();
-        verify(asyncDataChangeEvent).getOriginalData();
-        verify(asyncDataChangeEvent).getRemovedPaths();
-        Assert.assertNotNull(this);
+    public void testOnDataTreeChanged() throws Exception {
+        DataTreeModification<Link> mockDataTreeModification = mock(DataTreeModification.class);
+        DataObjectModification<Link> mockModification = mock(DataObjectModification.class);
+        doReturn(mockModification).when(mockDataTreeModification).getRootNode();
+
+        Link mockLink = mock(Link.class);
+        doReturn(mockLink).when(mockModification).getDataAfter();
+        doReturn(DataObjectModification.ModificationType.WRITE).when(mockModification).getModificationType();
+
+        listener.onDataTreeChanged(Collections.singletonList(mockDataTreeModification));
+
+        verify(mockAdapter).ofLinkAdded(mockLink);
     }
-}
\ No newline at end of file
+}