98a8eeaef979fecd976f42b2a59bf6c1cfab7f58
[ccsdk/features.git] /
1 /*
2  * ============LICENSE_START=======================================================
3  * ONAP : ccsdk feature sdnr wt sdnr-wt-devicemanager-provider
4  *  ================================================================================
5  * Copyright (C) 2019 highstreet technologies GmbH Intellectual Property.
6  * All rights reserved.
7  * ================================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *      http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END=========================================================
20  */
21
22 package org.onap.ccsdk.features.sdnr.wt.netconfnodestateservice.test.mock;
23
24 import java.util.Optional;
25 import org.mockito.Mockito;
26 import org.opendaylight.mdsal.binding.api.BindingService;
27 import org.opendaylight.mdsal.binding.api.DataBroker;
28 import org.opendaylight.mdsal.binding.api.MountPoint;
29 import org.opendaylight.mdsal.binding.api.NotificationService;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.network.topology.topology.topology.types.TopologyNetconf;
31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
34 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
36
37 /**
38  * @author herbert
39  *
40  */
41 public class MountPointMock extends Mockito implements MountPoint {
42
43     private boolean databrokerAbsent = true;
44     private final RpcConsumerRegistryMock rpcConsumerRegistry = new RpcConsumerRegistryMock();
45     private NotificationService setReadTransaction;
46     private DataBroker dataBroker;
47
48     private static final InstanceIdentifier<Topology> NETCONF_TOPO_IID =
49             InstanceIdentifier.create(NetworkTopology.class).child(Topology.class,
50                     new TopologyKey(new TopologyId(TopologyNetconf.QNAME.getLocalName())));
51
52     public MountPointMock() {
53         this.dataBroker = mock(DataBroker.class);
54     }
55
56     @Override
57     public InstanceIdentifier<?> getIdentifier() {
58         return NETCONF_TOPO_IID;
59     }
60
61     @SuppressWarnings("unchecked")
62     @Override
63     public <T extends BindingService> Optional<T> getService(Class<T> service) {
64
65         System.out.println("Requested mountpoint service: " + service.getSimpleName() + " databrokerAbsent state: "
66                 + databrokerAbsent);
67
68         Optional<? extends BindingService> res;
69         if (service.isInstance(dataBroker)) {
70             System.out.println("Delivering databroker");
71             res = databrokerAbsent ? Optional.empty() : Optional.of(dataBroker);
72         } else if (service.isInstance(rpcConsumerRegistry)) {
73             System.out.println("Delivering RpcConsumerRegistryMock");
74             res = Optional.of(rpcConsumerRegistry);
75         } else if (service.isInstance(setReadTransaction)) {
76             System.out.println("Delivering notificationService");
77             res = Optional.of(setReadTransaction);
78         } else {
79             System.out.println("Delivering no service");
80             res = Optional.empty();
81         }
82         return (Optional<T>) res;
83     }
84
85     public void setDatabrokerAbsent(boolean state) {
86         this.databrokerAbsent = state;
87     }
88
89     public DataBroker getDataBroker() {
90         return dataBroker;
91     }
92
93 }