2 * Copyright (c) 2014 Brocade Communications Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.controller.cluster.datastore.utils;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertTrue;
12 import static org.junit.Assert.fail;
13 import com.google.common.collect.Lists;
14 import com.google.common.util.concurrent.Uninterruptibles;
15 import java.util.Collections;
16 import java.util.List;
17 import java.util.concurrent.CountDownLatch;
18 import java.util.concurrent.TimeUnit;
19 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
20 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener;
21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
22 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
25 * A mock DataChangeListener implementation.
27 * @author Thomas Pantelis
29 public class MockDataChangeListener implements
30 AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> {
32 private final List<AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>>> changeList =
33 Collections.synchronizedList(Lists.<AsyncDataChangeEvent<YangInstanceIdentifier,
34 NormalizedNode<?, ?>>>newArrayList());
36 private volatile CountDownLatch changeLatch;
37 private int expChangeEventCount;
39 public MockDataChangeListener(int expChangeEventCount) {
40 reset(expChangeEventCount);
43 public void reset(int expChangeEventCount) {
44 changeLatch = new CountDownLatch(expChangeEventCount);
45 this.expChangeEventCount = expChangeEventCount;
50 public void onDataChanged(AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>> change) {
51 changeList.add(change);
52 changeLatch.countDown();
55 public void waitForChangeEvents(YangInstanceIdentifier... expPaths) {
56 boolean done = Uninterruptibles.awaitUninterruptibly(changeLatch, 5, TimeUnit.SECONDS);
58 fail(String.format("Missing change notifications. Expected: %d. Actual: %d",
59 expChangeEventCount, (expChangeEventCount - changeLatch.getCount())));
62 for(int i = 0; i < expPaths.length; i++) {
63 assertTrue(String.format("Change %d does not contain %s", (i+1), expPaths[i]),
64 changeList.get(i).getCreatedData().containsKey(expPaths[i]));
68 public void expectNoMoreChanges(String assertMsg) {
69 Uninterruptibles.sleepUninterruptibly(1, TimeUnit.SECONDS);
70 assertEquals(assertMsg, expChangeEventCount, changeList.size());