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;
18 import java.util.concurrent.CountDownLatch;
19 import java.util.concurrent.TimeUnit;
20 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
21 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener;
22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
23 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
26 * A mock DataChangeListener implementation.
28 * @author Thomas Pantelis
30 public class MockDataChangeListener implements
31 AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> {
33 private final List<AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>>> changeList =
34 Collections.synchronizedList(Lists.<AsyncDataChangeEvent<YangInstanceIdentifier,
35 NormalizedNode<?, ?>>>newArrayList());
37 private volatile CountDownLatch changeLatch;
38 private int expChangeEventCount;
40 public MockDataChangeListener(int expChangeEventCount) {
41 reset(expChangeEventCount);
44 public void reset(int expChangeEventCount) {
45 changeLatch = new CountDownLatch(expChangeEventCount);
46 this.expChangeEventCount = expChangeEventCount;
51 public void onDataChanged(AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>> change) {
52 changeList.add(change);
53 changeLatch.countDown();
56 public void waitForChangeEvents(YangInstanceIdentifier... expPaths) {
57 boolean done = Uninterruptibles.awaitUninterruptibly(changeLatch, 5, TimeUnit.SECONDS);
59 fail(String.format("Missing change notifications. Expected: %d. Actual: %d",
60 expChangeEventCount, (expChangeEventCount - changeLatch.getCount())));
63 for(int i = 0; i < expPaths.length; i++) {
64 Map<YangInstanceIdentifier, NormalizedNode<?, ?>> createdData = changeList.get(i).getCreatedData();
65 assertTrue(String.format("Change %d does not contain %s. Actual: %s", (i+1), expPaths[i], createdData),
66 createdData.containsKey(expPaths[i]));
70 public NormalizedNode<?, ?> getCreatedData(int i, YangInstanceIdentifier path) {
71 return changeList.get(i).getCreatedData().get(path);
74 public void verifyCreatedData(int i, YangInstanceIdentifier path) {
75 Map<YangInstanceIdentifier, NormalizedNode<?, ?>> createdData = changeList.get(i).getCreatedData();
76 assertTrue(path + " not present in " + createdData.keySet(), createdData.get(path) != null);
79 public void expectNoMoreChanges(String assertMsg) {
80 Uninterruptibles.sleepUninterruptibly(500, TimeUnit.MILLISECONDS);
81 assertEquals(assertMsg, expChangeEventCount, changeList.size());
84 public void verifyNoCreatedData(int i, YangInstanceIdentifier path) {
85 Map<YangInstanceIdentifier, NormalizedNode<?, ?>> createdData = changeList.get(i).getCreatedData();
86 assertTrue("Unexpected " + path + " present in createdData", createdData.get(path) == null);