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;
14 import com.google.common.collect.Lists;
15 import com.google.common.util.concurrent.Uninterruptibles;
16 import java.util.Collections;
17 import java.util.List;
19 import java.util.concurrent.CountDownLatch;
20 import java.util.concurrent.TimeUnit;
21 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
22 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener;
23 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
24 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
27 * A mock DataChangeListener implementation.
29 * @author Thomas Pantelis
31 public class MockDataChangeListener implements
32 AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> {
34 private final List<AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>>> changeList =
35 Collections.synchronizedList(Lists.<AsyncDataChangeEvent<YangInstanceIdentifier,
36 NormalizedNode<?, ?>>>newArrayList());
38 private volatile CountDownLatch changeLatch;
39 private int expChangeEventCount;
41 public MockDataChangeListener(int expChangeEventCount) {
42 reset(expChangeEventCount);
45 public void reset(int newExpChangeEventCount) {
46 changeLatch = new CountDownLatch(newExpChangeEventCount);
47 this.expChangeEventCount = newExpChangeEventCount;
52 public void onDataChanged(AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>> change) {
53 changeList.add(change);
54 changeLatch.countDown();
57 public void waitForChangeEvents(YangInstanceIdentifier... expPaths) {
58 boolean done = Uninterruptibles.awaitUninterruptibly(changeLatch, 5, TimeUnit.SECONDS);
60 fail(String.format("Missing change notifications. Expected: %d. Actual: %d",
61 expChangeEventCount, expChangeEventCount - changeLatch.getCount()));
64 for (int i = 0; i < expPaths.length; i++) {
65 Map<YangInstanceIdentifier, NormalizedNode<?, ?>> createdData = changeList.get(i).getCreatedData();
66 assertTrue(String.format("Change %d does not contain %s. Actual: %s", i + 1, expPaths[i], createdData),
67 createdData.containsKey(expPaths[i]));
71 public NormalizedNode<?, ?> getCreatedData(int num, YangInstanceIdentifier path) {
72 return changeList.get(num).getCreatedData().get(path);
75 public void verifyCreatedData(int num, YangInstanceIdentifier path) {
76 Map<YangInstanceIdentifier, NormalizedNode<?, ?>> createdData = changeList.get(num).getCreatedData();
77 assertTrue(path + " not present in " + createdData.keySet(), createdData.get(path) != null);
80 public void expectNoMoreChanges(String assertMsg) {
81 Uninterruptibles.sleepUninterruptibly(500, TimeUnit.MILLISECONDS);
82 assertEquals(assertMsg, expChangeEventCount, changeList.size());
85 public void verifyNoCreatedData(int num, YangInstanceIdentifier path) {
86 Map<YangInstanceIdentifier, NormalizedNode<?, ?>> createdData = changeList.get(num).getCreatedData();
87 assertTrue("Unexpected " + path + " present in createdData", createdData.get(path) == null);