1 package org.opendaylight.controller.cluster.datastore;
3 import akka.actor.ActorRef;
4 import akka.actor.Props;
5 import akka.testkit.JavaTestKit;
6 import com.google.common.util.concurrent.ListeningExecutorService;
7 import com.google.common.util.concurrent.MoreExecutors;
9 import org.opendaylight.controller.cluster.datastore.messages.CloseDataChangeListenerRegistration;
10 import org.opendaylight.controller.cluster.datastore.messages.CloseDataChangeListenerRegistrationReply;
11 import org.opendaylight.controller.md.cluster.datastore.model.TestModel;
12 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker;
13 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
14 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener;
15 import org.opendaylight.controller.md.sal.dom.store.impl.InMemoryDOMDataStore;
16 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
17 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
19 import static org.junit.Assert.assertEquals;
21 public class DataChangeListenerRegistrationTest extends AbstractActorTest {
22 private static ListeningExecutorService storeExecutor = MoreExecutors.listeningDecorator(MoreExecutors.sameThreadExecutor());
24 private static final InMemoryDOMDataStore store = new InMemoryDOMDataStore("OPER", storeExecutor,
25 MoreExecutors.sameThreadExecutor());
28 store.onGlobalContextUpdated(TestModel.createTestContext());
33 public void testOnReceiveCloseListenerRegistration() throws Exception {
34 new JavaTestKit(getSystem()) {{
35 final Props props = DataChangeListenerRegistration.props(store
36 .registerChangeListener(TestModel.TEST_PATH, noOpDataChangeListener(),
37 AsyncDataBroker.DataChangeScope.BASE));
38 final ActorRef subject = getSystem().actorOf(props, "testCloseListenerRegistration");
40 new Within(duration("1 seconds")) {
42 protected void run() {
44 subject.tell(new CloseDataChangeListenerRegistration().toSerializable(), getRef());
46 final String out = new ExpectMsg<String>(duration("1 seconds"), "match hint") {
47 // do not put code outside this method, will run afterwards
49 protected String match(Object in) {
50 if (in.getClass().equals(CloseDataChangeListenerRegistrationReply.SERIALIZABLE_CLASS)) {
56 }.get(); // this extracts the received message
58 assertEquals("match", out);
68 private AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> noOpDataChangeListener(){
69 return new AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>>() {
71 public void onDataChanged(AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>> change) {