Add some performance tests for InitCommand

Fixed gh-212 again

spring-boot-cli/src/test/java/org/springframework/boot/cli/command/InitCommandPerformanceTests.java

@@ -0,0 +1,116 @@
+/*
+ * Copyright 2012-2013 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.boot.cli.command;
+
+import java.util.Random;
+import java.util.ServiceLoader;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.springframework.boot.OutputCapture;
+import org.springframework.boot.cli.Command;
+import org.springframework.boot.cli.CommandFactory;
+import org.springframework.boot.cli.SpringCli;
+
+import static org.mockito.Mockito.mock;
+
+/**
+ * @author Dave Syer
+ */
+public class InitCommandPerformanceTests {
+
+	@Rule
+	public OutputCapture output = new OutputCapture();
+	private SpringCli cli = mock(SpringCli.class);
+	private ClassLoader classLoader;
+	private Random random = new Random();
+
+	@Before
+	public void init() {
+		this.classLoader = Thread.currentThread().getContextClassLoader();
+	}
+
+	@After
+	public void close() {
+		Thread.currentThread().setContextClassLoader(this.classLoader);
+	}
+
+	@Test
+	public void initDefault() throws Exception {
+		for (int i = 0; i < 100; i++) {
+			InitCommand command = new InitCommand(this.cli);
+			command.run();
+			close();
+		}
+	}
+
+	@Test
+	// Fast...
+	public void initNonExistent() throws Exception {
+		for (int i = 0; i < 100; i++) {
+			InitCommand command = new InitCommand(this.cli);
+			command.run("--init=" + this.random.nextInt() + ".groovy");
+			close();
+		}
+	}
+
+	@Test
+	// Fast...
+	public void fakeCommand() throws Exception {
+		for (int i = 0; i < 100; i++) {
+			Command command = new AbstractCommand("fake", "") {
+				@Override
+				public void run(String... args) throws Exception {
+					for (CommandFactory factory : ServiceLoader.load(
+							CommandFactory.class, Thread.currentThread()
+									.getContextClassLoader())) {
+						for (Command command : factory
+								.getCommands(InitCommandPerformanceTests.this.cli)) {
+							InitCommandPerformanceTests.this.cli.register(command);
+						}
+					}
+				}
+			};
+			command.run("--init=" + this.random.nextInt() + ".groovy");
+			close();
+		}
+	}
+
+	@Test
+	// Fast...
+	public void initNonExistentWithPrefix() throws Exception {
+		for (int i = 0; i < 100; i++) {
+			InitCommand command = new InitCommand(this.cli);
+			command.run("--init=file:" + this.random.nextInt() + ".groovy");
+			close();
+		}
+	}
+
+	@Test
+	// There is an init.groovy on the test classpath so this succeeds
+	// Slow...
+	public void initFromClasspath() throws Exception {
+		for (int i = 0; i < 5; i++) {
+			InitCommand command = new InitCommand(this.cli);
+			command.run("--init=init.groovy");
+			close();
+		}
+	}
+
+}

spring-boot-cli/src/test/java/org/springframework/boot/cli/command/InitCommandTests.java

@@ -18,6 +18,8 @@ package org.springframework.boot.cli.command;
 
 import groovy.lang.GroovyClassLoader;
 
+import java.util.Random;
+
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Rule;
@@ -26,6 +28,7 @@ import org.springframework.boot.OutputCapture;
 import org.springframework.boot.cli.Command;
 import org.springframework.boot.cli.SpringCli;
 
+import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
 import static org.mockito.Matchers.any;
 import static org.mockito.Mockito.mock;
@@ -107,11 +110,28 @@ public class InitCommandTests {
 		this.command.run("nonexistent.groovy");
 	}
 
-	// There is an init.groovy on the test classpath so this succeeds
 	@Test
 	public void initDefault() throws Exception {
 		this.command.run();
+		assertFalse(this.output.toString().contains("Hello Init"));
+	}
+
+	@Test
+	public void initWithCommandline() throws Exception {
+		this.command.run("--init=init.groovy");
 		assertTrue(this.output.toString().contains("Hello Init"));
 	}
 
+	public static void main(String[] args) throws Exception {
+		Random random = new Random();
+		InitCommandTests test = new InitCommandTests();
+		test.init();
+		SpringCli cli = new SpringCli();
+		while (true) {
+			InitCommand command = new InitCommand(cli);
+			command.run("--init=file:" + random.nextInt() + ".groovy");
+			test.close();
+		}
+	}
+
 }