diff options
author | 2015-02-25 16:45:20 +0100 | |
---|---|---|
committer | 2015-02-25 16:45:20 +0100 | |
commit | d08b27fa9701fecfdb69e1b0d1ac2459efc2129b (patch) | |
tree | 5d50963026239ca5aebfb47ea5b8db7e814e57c8 /src/test/java/com/google/devtools/build/lib/testutil/CustomSuite.java |
Update from Google.
--
MOE_MIGRATED_REVID=85702957
Diffstat (limited to 'src/test/java/com/google/devtools/build/lib/testutil/CustomSuite.java')
-rw-r--r-- | src/test/java/com/google/devtools/build/lib/testutil/CustomSuite.java | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/testutil/CustomSuite.java b/src/test/java/com/google/devtools/build/lib/testutil/CustomSuite.java new file mode 100644 index 0000000000..6e3b6c56ab --- /dev/null +++ b/src/test/java/com/google/devtools/build/lib/testutil/CustomSuite.java @@ -0,0 +1,53 @@ +// Copyright 2014 Google Inc. All rights reserved. +// +// 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 com.google.devtools.build.lib.testutil; + +import org.junit.runners.Suite; +import org.junit.runners.model.RunnerBuilder; + +import java.lang.reflect.Method; +import java.lang.reflect.Modifier; +import java.util.Set; + +/** + * A JUnit4 suite implementation that delegates the class finding to a {@code suite()} method on the + * annotated class. To be used in combination with {@link TestSuiteBuilder}. + */ +public final class CustomSuite extends Suite { + + /** + * Only called reflectively. Do not use programmatically. + */ + public CustomSuite(Class<?> klass, RunnerBuilder builder) throws Throwable { + super(builder, klass, getClasses(klass)); + } + + private static Class<?>[] getClasses(Class<?> klass) { + Set<Class<?>> result = evalSuite(klass); + return result.toArray(new Class<?>[result.size()]); + } + + @SuppressWarnings("unchecked") // unchecked cast to a generic type + private static Set<Class<?>> evalSuite(Class<?> klass) { + try { + Method m = klass.getMethod("suite"); + if (!Modifier.isStatic(m.getModifiers())) { + throw new IllegalStateException("suite() must be static"); + } + return (Set<Class<?>>) m.invoke(null); + } catch (Exception e) { + throw new IllegalStateException(e); + } + } +} |