001    /*****************************************************************************
002     * Copyright (C) PicoContainer Organization. All rights reserved.            *
003     * ------------------------------------------------------------------------- *
004     * The software in this package is published under the terms of the BSD      *
005     * style license a copy of which has been included with this distribution in *
006     * the LICENSE.txt file.                                                     *
007     *                                                                           *
008     * Idea by Rachel Davies, Original code by Aslak Hellesoy and Paul Hammant   *
009     *****************************************************************************/
010    package org.picocontainer.injectors;
011    
012    import org.picocontainer.PicoCompositionException;
013    import org.picocontainer.ComponentFactory;
014    import org.picocontainer.injectors.SetterInjection;
015    import org.picocontainer.monitors.NullComponentMonitor;
016    import org.picocontainer.DefaultPicoContainer;
017    import org.picocontainer.injectors.SetterInjector;
018    import org.picocontainer.tck.AbstractComponentFactoryTestCase;
019    import org.picocontainer.tck.AbstractComponentAdapterTestCase.RecordingLifecycleStrategy;
020    import org.picocontainer.testmodel.NullLifecycle;
021    import org.picocontainer.testmodel.RecordingLifecycle;
022    import org.picocontainer.testmodel.RecordingLifecycle.One;
023    
024    /**
025     * @author Jörg Schaible
026     */
027    public class SetterInjectionTestCase extends AbstractComponentFactoryTestCase {
028        protected void setUp() throws Exception {
029            picoContainer = new DefaultPicoContainer(createComponentFactory());
030        }
031    
032        protected ComponentFactory createComponentFactory() {
033            return new SetterInjection();
034        }
035    
036        public static interface Bean {
037        }
038    
039        public static class NamedBean implements Bean {
040            private String name;
041    
042            public String getName() {
043                return name;
044            }
045    
046            public void setName(String name) {
047                this.name = name;
048            }
049        }
050    
051        public static class NamedBeanWithPossibleDefault extends NamedBean {
052            private boolean byDefault;
053    
054            public NamedBeanWithPossibleDefault() {
055            }
056    
057            public NamedBeanWithPossibleDefault(String name) {
058                setName(name);
059                byDefault = true;
060            }
061    
062            public boolean getByDefault() {
063                return byDefault;
064            }
065        }
066    
067        public static class NoBean extends NamedBean {
068            public NoBean(String name) {
069                setName(name);
070            }
071        }
072    
073        public void testContainerUsesStandardConstructor() {
074            picoContainer.addComponent(Bean.class, NamedBeanWithPossibleDefault.class);
075            picoContainer.addComponent("Tom");
076            NamedBeanWithPossibleDefault bean = (NamedBeanWithPossibleDefault) picoContainer.getComponent(Bean.class);
077            assertFalse(bean.getByDefault());
078        }
079    
080        public void testContainerUsesOnlyStandardConstructor() {
081            picoContainer.addComponent(Bean.class, NoBean.class);
082            picoContainer.addComponent("Tom");
083            try {
084                picoContainer.getComponent(Bean.class);
085                fail("Instantiation should have failed.");
086            } catch (PicoCompositionException e) {
087            }
088        }
089    
090        public void testCustomLifecycleCanBeInjected() throws NoSuchMethodException {
091            RecordingLifecycleStrategy strategy = new RecordingLifecycleStrategy(new StringBuffer());
092            SetterInjection componentFactory = new SetterInjection();
093            SetterInjector sica = (SetterInjector)componentFactory.createComponentAdapter(new NullComponentMonitor(), strategy, null, NullLifecycle.class, NullLifecycle.class);
094            One one = new RecordingLifecycle.One(new StringBuffer());
095            sica.start(one);
096            sica.stop(one);        
097            sica.dispose(one);
098            assertEquals("<start<stop<dispose", strategy.recording());
099        }
100    
101        public static class AnotherNamedBean implements Bean {
102            private String name;
103    
104            public String getName() {
105                return name;
106            }
107    
108            public void initName(String name) {
109                this.name = name;
110            }
111        }
112    
113        public void testAlternatePrefixWorks() {
114            picoContainer = new DefaultPicoContainer(new SetterInjection("init"));
115            picoContainer.addComponent(Bean.class, AnotherNamedBean.class);
116            picoContainer.addComponent("Tom");
117            AnotherNamedBean bean = picoContainer.getComponent(AnotherNamedBean.class);
118            assertEquals("Tom", bean.getName());
119        }
120    
121    
122    }