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 static org.mockito.Matchers.any;
11 import static org.mockito.Matchers.anyString;
12 import static org.mockito.Matchers.eq;
13 import static org.mockito.Mockito.doNothing;
14 import static org.mockito.Mockito.doReturn;
15 import static org.mockito.Mockito.mock;
17 import com.google.common.collect.Lists;
18 import com.google.common.collect.Sets;
19 import java.io.Closeable;
20 import java.io.IOException;
21 import java.util.Collections;
22 import java.util.Dictionary;
23 import java.util.List;
24 import java.util.SortedSet;
25 import java.util.TreeSet;
26 import org.mockito.Mock;
27 import org.mockito.MockitoAnnotations;
28 import org.opendaylight.controller.config.persist.api.ConfigPusher;
29 import org.opendaylight.controller.config.persist.api.ConfigSnapshotHolder;
30 import org.opendaylight.controller.config.persist.api.Persister;
31 import org.opendaylight.controller.config.persist.api.PropertiesProvider;
32 import org.opendaylight.controller.netconf.api.Capability;
33 import org.opendaylight.controller.netconf.mapping.api.NetconfOperationService;
34 import org.opendaylight.controller.netconf.mapping.api.NetconfOperationServiceFactory;
35 import org.opendaylight.controller.netconf.persist.impl.DummyAdapter;
36 import org.osgi.framework.Bundle;
37 import org.osgi.framework.BundleContext;
38 import org.osgi.framework.Filter;
39 import org.osgi.framework.ServiceListener;
40 import org.osgi.framework.ServiceReference;
41 import org.osgi.framework.ServiceRegistration;
43 final class MockedBundleContext {
45 private BundleContext context;
47 private Filter outerFilter, innerFilter;
49 private ServiceReference<?> serviceReference;
51 private Bundle bundle;
53 NetconfOperationServiceFactory serviceFactory;
55 private NetconfOperationService service;
57 private ServiceRegistration<?> registration;
59 MockedBundleContext(long maxWaitForCapabilitiesMillis, long conflictingVersionTimeoutMillis) throws Exception {
60 MockitoAnnotations.initMocks(this);
61 doReturn(null).when(context).getProperty(anyString());
62 initContext(maxWaitForCapabilitiesMillis, conflictingVersionTimeoutMillis);
64 String outerFilterString = "(objectClass=org.opendaylight.controller.netconf.mapping.api.NetconfOperationServiceFactory)";
65 doReturn(outerFilter).when(context).createFilter(outerFilterString);
66 doNothing().when(context).addServiceListener(any(ServiceListener.class), eq(outerFilterString));
67 ServiceReference<?>[] toBeReturned = {serviceReference};
68 doReturn(toBeReturned).when(context).getServiceReferences(NetconfOperationServiceFactory.class.getName(), null);
70 String innerFilterString = "innerfilter";
71 doReturn(innerFilterString).when(outerFilter).toString();
73 doReturn(innerFilter).when(context).createFilter(ConfigPersisterActivator.getFilterString());
74 doReturn(innerFilterString).when(innerFilter).toString();
75 doNothing().when(context).addServiceListener(any(ServiceListener.class), eq(innerFilterString));
77 doReturn(toBeReturned).when(context).getServiceReferences((String) null, innerFilterString);
78 doReturn(bundle).when(serviceReference).getBundle();
79 doReturn(context).when(bundle).getBundleContext();
80 doReturn("").when(serviceReference).toString();
81 doReturn("context").when(context).toString();
82 doReturn(serviceFactory).when(context).getService(any(ServiceReference.class));
83 doReturn(service).when(serviceFactory).createService(anyString());
84 final Capability cap = mock(Capability.class);
85 doReturn("cap1").when(cap).getCapabilityUri();
86 doReturn(Collections.singleton(cap)).when(serviceFactory).getCapabilities();
87 doNothing().when(service).close();
88 doReturn("serviceFactoryMock").when(serviceFactory).toString();
90 doNothing().when(registration).unregister();
91 doReturn(registration).when(context).registerService(
92 eq(ConfigPusher.class.getName()), any(Closeable.class),
93 any(Dictionary.class));
96 public BundleContext getBundleContext() {
100 private void initContext(long maxWaitForCapabilitiesMillis, long conflictingVersionTimeoutMillis) {
101 initProp(context, "active", "1");
102 initProp(context, "1." + ConfigPersisterActivator.STORAGE_ADAPTER_CLASS_PROP_SUFFIX, DummyAdapterWithInitialSnapshot.class.getName());
103 initProp(context, "1." + "readonly", "false");
104 initProp(context, "1." + ".properties.fileStorage", "target/configuration-persister-test/initial/");
105 initProp(context, ConfigPersisterActivator.MAX_WAIT_FOR_CAPABILITIES_MILLIS_PROPERTY, String.valueOf(maxWaitForCapabilitiesMillis));
106 initProp(context, ConfigPersisterActivator.CONFLICTING_VERSION_TIMEOUT_MILLIS_PROPERTY, String.valueOf(conflictingVersionTimeoutMillis));
109 private void initProp(BundleContext context, String key, String value) {
110 initPropNoPrefix(context, ConfigPersisterActivator.NETCONF_CONFIG_PERSISTER + "." + key, value);
113 private void initPropNoPrefix(BundleContext context, String key, String value) {
114 doReturn(value).when(context).getProperty(key);
117 public static class DummyAdapterWithInitialSnapshot extends DummyAdapter {
119 public static final String CONFIG_SNAPSHOT = "config-snapshot";
120 public static String expectedCapability = "cap2";
123 public List<ConfigSnapshotHolder> loadLastConfigs() throws IOException {
124 return Lists.newArrayList(getConfigSnapshot());
128 public Persister instantiate(PropertiesProvider propertiesProvider) {
132 public ConfigSnapshotHolder getConfigSnapshot() {
133 return new ConfigSnapshotHolder() {
135 public String getConfigSnapshot() {
136 return "<data><" + CONFIG_SNAPSHOT + "/></data>";
140 public SortedSet<String> getCapabilities() {
141 TreeSet<String> strings = Sets.newTreeSet();
142 strings.add(expectedCapability);
147 public String toString() {
148 return getConfigSnapshot();