2 * Copyright (c) 2016 Cisco 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
9 package org.opendaylight.groupbasedpolicy.ne.location.provider;
11 import static org.mockito.Matchers.any;
12 import static org.mockito.Mockito.mock;
13 import static org.mockito.Mockito.verify;
14 import static org.mockito.Mockito.when;
16 import org.junit.Before;
17 import org.junit.Test;
18 import org.mockito.Matchers;
19 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
20 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
21 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.Endpoints;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.endpoints.AddressEndpoints;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.endpoints.address.endpoints.AddressEndpoint;
25 import org.opendaylight.yangtools.concepts.ListenerRegistration;
26 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
28 public class EndpointsListenerTest {
30 private EndpointsListener endpointsListener;
31 private NeLocationProvider listener;
32 private DataBroker dataBroker;
33 private ListenerRegistration<EndpointsListener> listenerRegistration;
37 dataBroker = mock(DataBroker.class);
38 listenerRegistration = mock(ListenerRegistration.class);
39 when(dataBroker.registerDataTreeChangeListener(any(), Matchers.any(EndpointsListener.class)))
40 .thenReturn(listenerRegistration);
41 listener = mock(NeLocationProvider.class);
42 endpointsListener = new EndpointsListener(dataBroker, listener);
46 public void test_RegistrationDone() {
47 verify(dataBroker).registerDataTreeChangeListener(new DataTreeIdentifier<>(LogicalDatastoreType.OPERATIONAL,
48 InstanceIdentifier.builder(Endpoints.class).child(AddressEndpoints.class)
49 .child(AddressEndpoint.class).build()), endpointsListener);
53 public void test_ListenerNotification() {
54 endpointsListener.onDataTreeChanged(null);
55 verify(listener).onEndpointsChange(any());
59 public void test_UnregiserOnClose() {
60 endpointsListener.close();
61 verify(listenerRegistration).close();