/*******************************************************************************
* Mission Control Technologies, Copyright (c) 2009-2012, United States Government
* as represented by the Administrator of the National Aeronautics and Space
* Administration. All rights reserved.
*
* The MCT platform is licensed under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* http://www.apache.org/licenses/LICENSE-2.0.
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*
* MCT includes source code licensed under additional open source licenses. See
* the MCT Open Source Licenses file included with this distribution or the About
* MCT Licenses dialog available at runtime from the MCT Help menu for additional
* information.
*******************************************************************************/
package gov.nasa.arc.mct.subscribe.manager;
import gov.nasa.arc.mct.event.services.EventProvider;
import gov.nasa.arc.mct.subscribe.manager.config.ConfigurationService;
import java.util.Collections;
import java.util.Dictionary;
import java.util.Hashtable;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceReference;
import org.osgi.service.cm.ConfigurationException;
import org.osgi.service.component.ComponentContext;
import org.osgi.service.log.LogService;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
public class SubscriptionManagerServiceTest {
private SubscriptionManagerService subManager;
@Mock
private LogService logService;
@Mock
private ComponentContext compContext;
@Mock
private BundleContext bc;
@Mock
private ServiceReference providerSR;
@Mock
private EventProvider eventProvider;
private String feedID = "testFeed";
@BeforeMethod
public void setup() throws ConfigurationException {
ConfigurationService cs = ConfigurationService.getInstance();
Dictionary<String, Integer> dict = new Hashtable<String, Integer>();
dict.put(ConfigurationService.UNSUBSCRIPTION_GRACE_PERIOD, 0);
cs.updated(dict);
subManager = new SubscriptionManagerService();
MockitoAnnotations.initMocks(this);
subManager.unsetLogger(null);
subManager.setLogger(logService);
Mockito.when(compContext.getBundleContext()).thenReturn(bc);
Mockito.when(eventProvider.subscribeTopics(feedID)).thenReturn(Collections.<String>emptyList());
}
/**
* Verify the subscription and unsubscriptions that make it to ISP work also
*/
@Test
public void checkSubscriptionLifeCycleThroughService() throws Exception {
final String feedID2 = "testFeedId2";
subManager.activate(compContext);
subManager.addProvider(eventProvider);
Mockito.when(eventProvider.subscribeTopics(feedID)).thenReturn(Collections.singleton(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID));
subManager.subscribe(feedID);
Thread.sleep(100);
Mockito.when(eventProvider.subscribeTopics(feedID2)).thenReturn(Collections.singleton(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID2));
subManager.subscribe(feedID2);
Mockito.verify(eventProvider, Mockito.times(1)).subscribeTopics(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID);
Mockito.verify(eventProvider, Mockito.times(1)).subscribeTopics(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID2);
// should be added to the list, now verify that this will get removed once the cycle has completed
subManager.unsubscribe(feedID);
subManager.unsubscribe(feedID2);
// wait until the unsubscription timer event fires to verify that the unsubscription event was sent
Thread.yield();
Thread.sleep(1400);
Mockito.verify(eventProvider, Mockito.times(1)).unsubscribeTopics(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID);
Mockito.verify(eventProvider, Mockito.times(1)).unsubscribeTopics(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID2);
}
@Test
public void addProviderTest() {
subManager.subscribe(feedID);
Mockito.verify(eventProvider, Mockito.never()).subscribeTopics(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID);
subManager.activate(compContext);
subManager.addProvider(eventProvider);
Mockito.verify(eventProvider, Mockito.times(1)).subscribeTopics(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID);
subManager.unsubscribe(feedID);
Mockito.verify(eventProvider, Mockito.never()).unsubscribeTopics(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID);
}
@Test
public void removeProviderTest() {
subManager.addProvider(eventProvider);
subManager.removeProvider(eventProvider);
subManager.subscribe(feedID);
Mockito.verify(eventProvider, Mockito.never()).subscribeTopics(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID);
subManager.activate(compContext);
Mockito.verify(eventProvider, Mockito.never()).subscribeTopics(EventProvider.TELEMETRY_TOPIC_PREFIX+feedID);
}
}