aries-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From csie...@apache.org
Subject svn commit: r1796110 - /aries/trunk/component-dsl/itests/src/main/java/org/apache/aries/osgi/functional/test/ComponentTest.java
Date Wed, 24 May 2017 21:44:05 GMT
Author: csierra
Date: Wed May 24 21:44:05 2017
New Revision: 1796110

URL: http://svn.apache.org/viewvc?rev=1796110&view=rev
Log:
Test definition of component-like "programs"

Added:
    aries/trunk/component-dsl/itests/src/main/java/org/apache/aries/osgi/functional/test/ComponentTest.java

Added: aries/trunk/component-dsl/itests/src/main/java/org/apache/aries/osgi/functional/test/ComponentTest.java
URL: http://svn.apache.org/viewvc/aries/trunk/component-dsl/itests/src/main/java/org/apache/aries/osgi/functional/test/ComponentTest.java?rev=1796110&view=auto
==============================================================================
--- aries/trunk/component-dsl/itests/src/main/java/org/apache/aries/osgi/functional/test/ComponentTest.java
(added)
+++ aries/trunk/component-dsl/itests/src/main/java/org/apache/aries/osgi/functional/test/ComponentTest.java
Wed May 24 21:44:05 2017
@@ -0,0 +1,253 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You 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.
+ */
+
+package org.apache.aries.osgi.functional.test;
+
+import org.apache.aries.osgi.functional.OSGi;
+import org.apache.aries.osgi.functional.OSGiResult;
+import org.junit.AfterClass;
+import org.junit.Assert;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.osgi.framework.BundleContext;
+import org.osgi.framework.FrameworkUtil;
+import org.osgi.framework.ServiceReference;
+import org.osgi.framework.ServiceRegistration;
+import org.osgi.service.cm.Configuration;
+import org.osgi.service.cm.ConfigurationAdmin;
+import org.osgi.service.cm.ManagedService;
+import org.osgi.util.tracker.ServiceTracker;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Dictionary;
+import java.util.HashMap;
+import java.util.Hashtable;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
+import java.util.function.Consumer;
+
+import static org.apache.aries.osgi.functional.OSGi.bundleContext;
+import static org.apache.aries.osgi.functional.OSGi.configurations;
+import static org.apache.aries.osgi.functional.OSGi.just;
+import static org.apache.aries.osgi.functional.OSGi.onClose;
+import static org.apache.aries.osgi.functional.OSGi.register;
+import static org.apache.aries.osgi.functional.OSGi.services;
+import static org.apache.aries.osgi.functional.test.HighestRankingRouter.highest;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+
+/**
+ * @author Carlos Sierra Andrés
+ */
+public class ComponentTest {
+
+    static BundleContext _bundleContext =
+        FrameworkUtil.getBundle(DSLTest.class).getBundleContext();
+    private static ConfigurationAdmin _configurationAdmin;
+    private static ServiceReference<ConfigurationAdmin> _configAdminServiceReference;
+
+    @BeforeClass
+    public static void setupClass() {
+        _configAdminServiceReference = _bundleContext.getServiceReference(
+            ConfigurationAdmin.class);
+
+        _configurationAdmin = _bundleContext.getService(
+            _configAdminServiceReference);
+    }
+
+    @AfterClass
+    public static void tearDownClass() {
+        _bundleContext.ungetService(_configAdminServiceReference);
+    }
+
+    @Test
+    public void testComponent() {
+        OSGi<?> program =
+            configurations("org.components.MyComponent").flatMap(props ->
+            services(Service.class).flatMap(ms ->
+            just(new Component(props, ms)).flatMap(component ->
+            register(Component.class, component, new HashMap<>()).distribute(
+            ign -> dynamic(
+                highestService(ServiceOptional.class),
+                component::setOptional, c -> component.setOptional(null)),
+            ign -> dynamic(
+                services(ServiceForList.class),
+                component::addService, component::removeService)
+        ))));
+
+        ServiceTracker<Component, Component> serviceTracker =
+            new ServiceTracker<>(_bundleContext, Component.class, null);
+
+        serviceTracker.open();
+
+        CountDownLatch countDownLatch = new CountDownLatch(1);
+
+        _bundleContext.registerService(
+            ManagedService.class, dictionary -> countDownLatch.countDown(),
+            new Hashtable<String, Object>() {{
+                put("service.pid", "org.components.MyComponent");
+            }});
+
+        try (OSGiResult<?> run = program.run(_bundleContext)) {
+            Configuration factoryConfiguration =
+                _configurationAdmin.createFactoryConfiguration(
+                    "org.components.MyComponent");
+            factoryConfiguration.update(new Hashtable<>());
+
+            countDownLatch.await(10, TimeUnit.SECONDS);
+
+            assertNull(serviceTracker.getService());
+
+            ServiceRegistration<Service> serviceRegistration =
+                _bundleContext.registerService(
+                    Service.class, new Service(), new Hashtable<>());
+
+            Component component = serviceTracker.waitForService(10 * 1000);
+
+            assertNotNull(component);
+
+            assertNull(component.getOptional());
+
+            ServiceRegistration<ServiceOptional> serviceRegistration2 =
+                _bundleContext.registerService(
+                    ServiceOptional.class, new ServiceOptional(),
+                    new Hashtable<>());
+
+            Thread.sleep(1000L);
+
+            assertNotNull(component.getOptional());
+
+            ServiceOptional serviceOptional = new ServiceOptional();
+
+            ServiceRegistration<ServiceOptional> serviceRegistration3 =
+                _bundleContext.registerService(
+                    ServiceOptional.class, serviceOptional,
+                    new Hashtable<String, Object>() {{
+                        put("service.ranking", 1);
+                    }});
+
+            assertEquals(serviceOptional, component.getOptional());
+
+            serviceRegistration3.unregister();
+
+            assertNotNull(component.getOptional());
+
+            serviceRegistration2.unregister();
+
+            assertNull(component.getOptional());
+
+            ServiceRegistration<ServiceForList> serviceRegistration4 =
+                _bundleContext.registerService(
+                    ServiceForList.class, new ServiceForList(),
+                    new Hashtable<>());
+
+            ServiceRegistration<ServiceForList> serviceRegistration5 =
+                _bundleContext.registerService(
+                    ServiceForList.class, new ServiceForList(),
+                    new Hashtable<>());
+
+            assertEquals(2, component.getServiceForLists().size());
+
+            serviceRegistration4.unregister();
+
+            assertEquals(1, component.getServiceForLists().size());
+
+            serviceRegistration5.unregister();
+
+            assertEquals(0, component.getServiceForLists().size());
+
+            serviceRegistration.unregister();
+
+            assertNull(serviceTracker.getService());
+        }
+        catch (IOException ioe) {
+
+        }
+        catch (InterruptedException e) {
+            Assert.fail("Timeout waiting for configuration");
+        }
+        finally {
+            serviceTracker.close();
+        }
+    }
+
+    private static <T> OSGi<T> highestService(Class<T> clazz) {
+        return highest(clazz).flatMap(sr ->
+        bundleContext().flatMap(bc ->
+        onClose(() -> bc.ungetService(sr)).then(
+        just(bc.getService(sr))
+        )));
+    }
+
+    public static <T> OSGi<Void> dynamic(
+        OSGi<T> program, Consumer<T> bind, Consumer<T> unbind) {
+
+        return program.foreach(bind, unbind);
+    }
+
+    private class Component {
+
+        final Dictionary<String, ?> configuration;
+        final Service mandatory;
+        private final ArrayList<ServiceForList> _serviceForLists;
+        ServiceOptional optional = null;
+
+        public Component(Dictionary<String, ?> configuration, Service mandatory) {
+            this.configuration = configuration;
+            this.mandatory = mandatory;
+
+            _serviceForLists = new ArrayList<>();
+        }
+
+        public Dictionary<String, ?> getConfiguration() {
+            return configuration;
+        }
+
+        public Service getMandatory() {
+            return mandatory;
+        }
+
+        public ServiceOptional getOptional() {
+            return optional;
+        }
+
+        public void setOptional(ServiceOptional optional) {
+            this.optional = optional;
+        }
+
+        public void addService(ServiceForList serviceForList) {
+            _serviceForLists.add(serviceForList);
+        }
+
+        public void removeService(ServiceForList serviceForList) {
+            _serviceForLists.remove(serviceForList);
+        }
+
+        public ArrayList<ServiceForList> getServiceForLists() {
+            return _serviceForLists;
+        }
+
+    }
+
+    private class Service {}
+
+    private class ServiceOptional {}
+
+    private class ServiceForList {}
+}



Mime
View raw message