core/runtime/src/test/java/org/apache/zest/runtime/activation/ServiceActivationTest.java

/*
 * Copyright (c) 2012, Paul Merlin.
 *
 * 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.
 *
 */
package org.apache.zest.runtime.activation;

import org.junit.Assert;
import org.junit.Test;

import org.apache.zest.api.activation.Activator;
import org.apache.zest.api.mixin.Mixins;
import org.apache.zest.api.service.ServiceComposite;
import org.apache.zest.api.service.ServiceReference;
import org.apache.zest.api.structure.Application;
import org.apache.zest.bootstrap.AssemblyException;
import org.apache.zest.bootstrap.ModuleAssembly;
import org.apache.zest.bootstrap.SingletonAssembler;

public class ServiceActivationTest
{

    private static int activationLevel = 0;

    private static int passivationLevel = 0;

    public static class TestedActivator
            implements Activator<ServiceReference<TestedService>>
    {

        public void beforeActivation( ServiceReference<TestedService> activating )
        {
            Assert.assertFalse( "Service should not be active before activation", activating.isActive() );
            try {
                activating.get();
                Assert.fail( "Service is not activated yet, the reference get method should throw IllegalStateException." );
            } catch ( IllegalStateException expected ) {
            }
            activationLevel++;
        }

        public void afterActivation( ServiceReference<TestedService> activated )
        {
            Assert.assertTrue( "Service should be active after activation", activated.isActive() );
            Assert.assertEquals( "After activation", "bar", activated.get().foo() );
            activationLevel++;
        }

        public void beforePassivation( ServiceReference<TestedService> passivating )
        {
            Assert.assertTrue( "Service should be active before passivation", passivating.isActive() );
            Assert.assertEquals( "Before passivation", "bar", passivating.get().foo() );
            passivationLevel++;
        }

        public void afterPassivation( ServiceReference<TestedService> passivated )
        {
            Assert.assertFalse( "Service should not be active after passivation", passivated.isActive() );
            try {
                passivated.get();
                Assert.fail( "Service is passivated, the reference get method should throw IllegalStateException." );
            } catch ( IllegalStateException expected ) {
            }
            passivationLevel++;
        }

    }

    @Mixins( TestedServiceMixin.class )
    public static interface TestedServiceComposite
            extends TestedService, ServiceComposite
    {
    }

    @Mixins( TestedServiceMixin.class )
    public static interface TestedServiceComposite2
            extends TestedService, ServiceComposite
    {
    }

    public static interface TestedService
    {

        String foo();

    }

    public static class TestedServiceMixin
            implements TestedService
    {

        public String foo()
        {
            return "bar";
        }

    }

    @Test
    public void testServicesActivators()
            throws Exception
    {
        SingletonAssembler assembly = new SingletonAssembler()
        {

            public void assemble( ModuleAssembly module )
                    throws AssemblyException
            {
                module.addServices( TestedServiceComposite.class ).
                        withActivators( TestedActivator.class ).
                        instantiateOnStartup();
                module.addServices( TestedServiceComposite2.class ).
                        withActivators( TestedActivator.class ).
                        instantiateOnStartup();
            }

        };
        // Activate
        Application application = assembly.application();

        // Assert activated
        Assert.assertEquals( "Activation Level", 4, activationLevel );

        // Passivate
        application.passivate();

        // Assert passivated
        Assert.assertEquals( "Passivation Level", 4, passivationLevel );
    }

}