Remove ListenerRegistration protobuff messages

[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / datastore / DataChangeListenerRegistrationProxyTest.java

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/DataChangeListenerRegistrationProxyTest.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/DataChangeListenerRegistrationProxyTest.java
index b338329740cbef8e8737b1e1fbed18ce97c6e60c..b5567fa1d59b247d2f28c4804989e0c6b95a495b 100644 (file)
--- a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/DataChangeListenerRegistrationProxyTest.java
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/DataChangeListenerRegistrationProxyTest.java
@@ -113,7 +113,7 @@ public class DataChangeListenerRegistrationProxyTest extends AbstractActorTest {
             proxy.close();
 
             // The listener registration actor should get a Close message
-            expectMsgClass(timeout, CloseDataChangeListenerRegistration.SERIALIZABLE_CLASS);
+            expectMsgClass(timeout, CloseDataChangeListenerRegistration.class);
 
             // The DataChangeListener actor should be terminated
             expectMsgClass(timeout, Terminated.class);
@@ -171,7 +171,7 @@ public class DataChangeListenerRegistrationProxyTest extends AbstractActorTest {
             proxy.close();
 
             // The listener registration actor should get a Close message
-            expectMsgClass(timeout, CloseDataChangeListenerRegistration.SERIALIZABLE_CLASS);
+            expectMsgClass(timeout, CloseDataChangeListenerRegistration.class);
 
             // The DataChangeListener actor should be terminated
             expectMsgClass(timeout, Terminated.class);
@@ -313,7 +313,7 @@ public class DataChangeListenerRegistrationProxyTest extends AbstractActorTest {
             proxy.init(YangInstanceIdentifier.of(TestModel.TEST_QNAME),
                     AsyncDataBroker.DataChangeScope.ONE);
 
-            expectMsgClass(duration("5 seconds"), CloseDataChangeListenerRegistration.SERIALIZABLE_CLASS);
+            expectMsgClass(duration("5 seconds"), CloseDataChangeListenerRegistration.class);
 
             Assert.assertEquals("getListenerRegistrationActor", null,
                     proxy.getListenerRegistrationActor());