renderers/faas/src/test/java/org/opendaylight/groupbasedpolicy/renderer/faas/test/DataChangeListenerTester.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.groupbasedpolicy.renderer.faas.test;
   9
  10 import static org.mockito.Mockito.mock;
  11 import static org.mockito.Mockito.when;
  12
  13 import java.util.HashMap;
  14 import java.util.HashSet;
  15 import java.util.Map;
  16 import java.util.Set;
  17
  18 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  19 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  20 import org.opendaylight.yangtools.yang.binding.DataObject;
  21 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  22
  23 public class DataChangeListenerTester {
  24
  25     private AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> changeMock;
  26     private DataChangeListener listener;
  27
  28     private Map<InstanceIdentifier<?>, DataObject> testData;
  29     private Set<InstanceIdentifier<?>> removedPaths;
  30
  31     @SuppressWarnings("unchecked")
  32     public DataChangeListenerTester(DataChangeListener listener) {
  33         changeMock = mock(AsyncDataChangeEvent.class);
  34         testData = new HashMap<>();
  35         removedPaths = new HashSet<>();
  36
  37         this.listener = listener;
  38
  39         when(changeMock.getCreatedData()).thenReturn(testData);
  40         when(changeMock.getOriginalData()).thenReturn(testData);
  41         when(changeMock.getUpdatedData()).thenReturn(testData);
  42         when(changeMock.getRemovedPaths()).thenReturn(removedPaths);
  43     }
  44
  45     public AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> getChangeMock() {
  46         return changeMock;
  47     }
  48
  49     public DataChangeListenerTester setDataObject(InstanceIdentifier<? extends DataObject> iid, DataObject dataObject){
  50         testData.clear();
  51         return addDataObject(iid, dataObject);
  52     }
  53
  54     public DataChangeListenerTester addDataObject(InstanceIdentifier<? extends DataObject> iid, DataObject dataObject){
  55         testData.put(iid, dataObject);
  56         return this;
  57     }
  58
  59     public DataChangeListenerTester setRemovedPath(InstanceIdentifier<? extends DataObject> iid){
  60         removedPaths.clear();
  61         return addRemovedPath(iid);
  62     }
  63
  64     public DataChangeListenerTester addRemovedPath(InstanceIdentifier<? extends DataObject> iid){
  65         removedPaths.add(iid);
  66         return this;
  67     }
  68
  69     public void callOnDataChanged(){
  70         listener.onDataChanged(changeMock);
  71     }
  72
  73 }