2 * Copyright (c) 2013 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
8 package org.opendaylight.controller.netconf.persist.impl.osgi;
10 import com.google.common.collect.Lists;
11 import com.google.common.collect.Sets;
12 import org.mockito.Mock;
13 import org.mockito.MockitoAnnotations;
14 import org.opendaylight.controller.config.persist.api.ConfigSnapshotHolder;
15 import org.opendaylight.controller.config.persist.api.Persister;
16 import org.opendaylight.controller.config.persist.api.PropertiesProvider;
17 import org.opendaylight.controller.netconf.mapping.api.NetconfOperationProvider;
18 import org.opendaylight.controller.netconf.mapping.api.NetconfOperationService;
19 import org.opendaylight.controller.netconf.mapping.api.NetconfOperationServiceFactory;
20 import org.opendaylight.controller.netconf.persist.impl.DummyAdapter;
21 import org.osgi.framework.Bundle;
22 import org.osgi.framework.BundleContext;
23 import org.osgi.framework.Filter;
24 import org.osgi.framework.ServiceListener;
25 import org.osgi.framework.ServiceReference;
27 import java.io.IOException;
28 import java.util.Collections;
29 import java.util.List;
30 import java.util.SortedSet;
31 import java.util.TreeSet;
33 import static org.mockito.Matchers.any;
34 import static org.mockito.Matchers.anyString;
35 import static org.mockito.Matchers.eq;
36 import static org.mockito.Mockito.doNothing;
37 import static org.mockito.Mockito.doReturn;
39 final class MockedBundleContext {
41 private BundleContext context;
43 private Filter outerFilter, innerFilter;
45 private ServiceReference<?> serviceReference;
47 private Bundle bundle;
49 NetconfOperationServiceFactory serviceFactory;
51 private NetconfOperationService service;
53 MockedBundleContext(long maxWaitForCapabilitiesMillis, long conflictingVersionTimeoutMillis) throws Exception {
54 MockitoAnnotations.initMocks(this);
55 doReturn(null).when(context).getProperty(anyString());
56 initContext(maxWaitForCapabilitiesMillis, conflictingVersionTimeoutMillis);
58 String outerFilterString = "(objectClass=org.opendaylight.controller.netconf.mapping.api.NetconfOperationProvider)";
59 doReturn(outerFilter).when(context).createFilter(outerFilterString);
60 doNothing().when(context).addServiceListener(any(ServiceListener.class), eq(outerFilterString));
61 ServiceReference<?>[] toBeReturned = {serviceReference};
62 doReturn(toBeReturned).when(context).getServiceReferences(NetconfOperationProvider.class.getName(), null);
64 String innerFilterString = "innerfilter";
65 doReturn(innerFilterString).when(outerFilter).toString();
67 doReturn(innerFilter).when(context).createFilter(ConfigPersisterActivator.getFilterString());
68 doReturn(innerFilterString).when(innerFilter).toString();
69 doNothing().when(context).addServiceListener(any(ServiceListener.class), eq(innerFilterString));
71 doReturn(toBeReturned).when(context).getServiceReferences((String) null, innerFilterString);
72 doReturn(bundle).when(serviceReference).getBundle();
73 doReturn(context).when(bundle).getBundleContext();
74 doReturn("").when(serviceReference).toString();
75 doReturn(serviceFactory).when(context).getService(any(ServiceReference.class));
76 doReturn(service).when(serviceFactory).createService(anyString());
77 doReturn(Collections.emptySet()).when(service).getCapabilities();
78 doNothing().when(service).close();
79 doReturn("serviceFactoryMock").when(serviceFactory).toString();
82 public BundleContext getBundleContext() {
86 private void initContext(long maxWaitForCapabilitiesMillis, long conflictingVersionTimeoutMillis) {
87 initProp(context, "active", "1");
88 initProp(context, "1." + ConfigPersisterActivator.STORAGE_ADAPTER_CLASS_PROP_SUFFIX, DummyAdapterWithInitialSnapshot.class.getName());
89 initProp(context, "1." + "readonly", "false");
90 initProp(context, "1." + ".properties.fileStorage", "target/configuration-persister-test/initial/");
91 initProp(context, ConfigPersisterActivator.MAX_WAIT_FOR_CAPABILITIES_MILLIS_PROPERTY, String.valueOf(maxWaitForCapabilitiesMillis));
92 initProp(context, ConfigPersisterActivator.CONFLICTING_VERSION_TIMEOUT_MILLIS_PROPERTY, String.valueOf(conflictingVersionTimeoutMillis));
95 private void initProp(BundleContext context, String key, String value) {
96 initPropNoPrefix(context, ConfigPersisterActivator.NETCONF_CONFIG_PERSISTER + "." + key, value);
99 private void initPropNoPrefix(BundleContext context, String key, String value) {
100 doReturn(value).when(context).getProperty(key);
103 public static class DummyAdapterWithInitialSnapshot extends DummyAdapter {
105 public static final String CONFIG_SNAPSHOT = "config-snapshot";
106 public static String expectedCapability = "cap2";
109 public List<ConfigSnapshotHolder> loadLastConfigs() throws IOException {
110 return Lists.newArrayList(getConfigSnapshot());
114 public Persister instantiate(PropertiesProvider propertiesProvider) {
118 public ConfigSnapshotHolder getConfigSnapshot() {
119 return new ConfigSnapshotHolder() {
121 public String getConfigSnapshot() {
122 return "<data><" + CONFIG_SNAPSHOT + "/></data>";
126 public SortedSet<String> getCapabilities() {
127 TreeSet<String> strings = Sets.newTreeSet();
128 strings.add(expectedCapability);
133 public String toString() {
134 return getConfigSnapshot();