Added MockStrategiesHelper

test/src/main/java/org/springframework/ws/test/support/MockStrategiesHelper.java

@@ -0,0 +1,74 @@
+/*
+ * Copyright 2005-2010 the original author or authors.
+ *
+ * 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.springframework.ws.test.support;
+
+import java.util.Map;
+
+import org.springframework.beans.factory.BeanInitializationException;
+import org.springframework.context.ApplicationContext;
+import org.springframework.util.ClassUtils;
+
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+
+/**
+ * Helper class for for loading default implementations of an interface.
+ *
+ * @author Arjen Poutsma
+ * @since 2.0
+ */
+public class MockStrategiesHelper {
+
+    private static final Log logger = LogFactory.getLog(MockStrategiesHelper.class);
+
+    private final ApplicationContext applicationContext;
+
+    /**
+     * Creates a new instance of the {@code MockStrategiesHelper} with the given application context.
+     *
+     * @param applicationContext the application context
+     */
+    public MockStrategiesHelper(ApplicationContext applicationContext) {
+        this.applicationContext = applicationContext;
+    }
+
+    /**
+     * Returns a single strategy found in the given application context.
+     *
+     * @param type the type of bean to be found in the application context
+     * @return the bean, or {@code null} if no bean of the given type can be found
+     * @throws BeanInitializationException if there is more than 1 beans of the given type
+     */
+    public <T> T getStrategy(Class<T> type) {
+        Map<String, T> map = applicationContext.getBeansOfType(type);
+        if (map.isEmpty()) {
+            return null;
+        }
+        else if (map.size() == 1) {
+            Map.Entry<String, T> entry = map.entrySet().iterator().next();
+            if (logger.isDebugEnabled()) {
+                logger.debug("Using " + ClassUtils.getShortName(type) + " [" + entry.getKey() + "]");
+            }
+            return entry.getValue();
+        }
+        else {
+            throw new BeanInitializationException(
+                    "Could not find exactly 1 " + ClassUtils.getShortName(type) + " in application context");
+        }
+    }
+
+}

test/src/test/java/org/springframework/ws/test/support/MockStrategiesHelperTest.java

@@ -0,0 +1,60 @@
+/*
+ * Copyright 2005-2010 the original author or authors.
+ *
+ * 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.springframework.ws.test.support;
+
+import org.springframework.beans.factory.BeanInitializationException;
+import org.springframework.context.support.StaticApplicationContext;
+
+import org.junit.Test;
+
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+
+public class MockStrategiesHelperTest {
+
+    @Test
+    public void none() {
+        StaticApplicationContext applicationContext = new StaticApplicationContext();
+
+        MockStrategiesHelper helper = new MockStrategiesHelper(applicationContext);
+        assertNull(helper.getStrategy(MyBean.class));
+    }
+
+    @Test
+    public void one() {
+        StaticApplicationContext applicationContext = new StaticApplicationContext();
+        applicationContext.registerSingleton("myBean", MyBean.class);
+
+        MockStrategiesHelper helper = new MockStrategiesHelper(applicationContext);
+        assertNotNull(helper.getStrategy(MyBean.class));
+    }
+
+    @Test(expected = BeanInitializationException.class)
+    public void many() {
+        StaticApplicationContext applicationContext = new StaticApplicationContext();
+        applicationContext.registerSingleton("myBean1", MyBean.class);
+        applicationContext.registerSingleton("myBean2", MyBean.class);
+
+        MockStrategiesHelper helper = new MockStrategiesHelper(applicationContext);
+        helper.getStrategy(MyBean.class);
+    }
+
+    public static class MyBean {
+
+    }
+
+}