|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185 |
- /*
- * Copyright (C) 2012-present 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.pf4j;
-
- import com.google.testing.compile.Compilation;
- import com.google.testing.compile.Compiler;
- import com.google.testing.compile.JavaFileObjects;
- import org.junit.jupiter.api.BeforeEach;
- import org.junit.jupiter.api.Test;
- import org.pf4j.processor.ExtensionAnnotationProcessor;
- import org.pf4j.processor.LegacyExtensionStorage;
- import org.pf4j.test.JavaSources;
-
- import javax.tools.JavaFileObject;
- import java.io.IOException;
- import java.util.Collections;
- import java.util.HashMap;
- import java.util.HashSet;
- import java.util.Map;
- import java.util.Set;
-
- import static com.google.testing.compile.CompilationSubject.assertThat;
- import static com.google.testing.compile.Compiler.javac;
- import static org.junit.jupiter.api.Assertions.assertEquals;
-
- /**
- * @author Mario Franco
- * @author Decebal Suiu
- */
- public class ExtensionAnnotationProcessorTest {
-
- public static final JavaFileObject WhazzupGreeting_NoExtensionPoint = JavaFileObjects.forSourceLines("WhazzupGreeting",
- "package test;",
- "import org.pf4j.Extension;",
- "",
- "@Extension",
- "public class WhazzupGreeting {",
- " @Override",
- " public String getGreeting() {",
- " return \"Whazzup\";",
- " }",
- "}");
-
- public static final JavaFileObject SpinnakerExtension = JavaFileObjects.forSourceLines("SpinnakerExtension",
- "package test;",
- "",
- "import org.pf4j.Extension;",
- "import java.lang.annotation.Documented;",
- "import java.lang.annotation.ElementType;",
- "import java.lang.annotation.Retention;",
- "import java.lang.annotation.RetentionPolicy;",
- "import java.lang.annotation.Target;",
- "",
- "@Extension",
- "@Retention(RetentionPolicy.RUNTIME)",
- "@Target(ElementType.TYPE)",
- "@Documented",
- "public @interface SpinnakerExtension {",
- "}");
-
- public static final JavaFileObject WhazzupGreeting_SpinnakerExtension = JavaFileObjects.forSourceLines("WhazzupGreeting",
- "package test;",
- "",
- "@SpinnakerExtension",
- "public class WhazzupGreeting implements Greeting {",
- " @Override",
- " public String getGreeting() {",
- " return \"Whazzup\";",
- " }",
- "}");
-
- /**
- * The same as {@link #SpinnakerExtension} but without {@code Extension} annotation.
- */
- public static final JavaFileObject SpinnakerExtension_NoExtension = JavaFileObjects.forSourceLines("SpinnakerExtension",
- "package test;",
- "",
- "import org.pf4j.Extension;",
- "import java.lang.annotation.Documented;",
- "import java.lang.annotation.ElementType;",
- "import java.lang.annotation.Retention;",
- "import java.lang.annotation.RetentionPolicy;",
- "import java.lang.annotation.Target;",
- "",
- // "@Extension",
- "@Retention(RetentionPolicy.RUNTIME)",
- "@Target(ElementType.TYPE)",
- "@Documented",
- "public @interface SpinnakerExtension {",
- "}");
-
- private ExtensionAnnotationProcessor annotationProcessor;
-
- @BeforeEach
- public void setUp() throws IOException {
- annotationProcessor = new ExtensionAnnotationProcessor();
- }
-
- @Test
- public void getSupportedAnnotationTypes() {
- Set<String> result = annotationProcessor.getSupportedAnnotationTypes();
- assertEquals(1, result.size());
- assertEquals("*", result.iterator().next());
- }
-
- @Test
- public void getSupportedOptions() {
- Set<String> result = annotationProcessor.getSupportedOptions();
- assertEquals(2, result.size());
- }
-
- @Test
- public void options() {
- Compilation compilation = compiler().withOptions("-Ab=2", "-Ac=3")
- .compile(JavaSources.Greeting, JavaSources.WhazzupGreeting);
- assertEquals(Compilation.Status.SUCCESS, compilation.status());
- Map<String, String> options = new HashMap<>();
- options.put("b", "2");
- options.put("c", "3");
- assertEquals(options, annotationProcessor.getProcessingEnvironment().getOptions());
- }
-
- @Test
- public void storage() {
- Compilation compilation = compile(JavaSources.Greeting, JavaSources.WhazzupGreeting);
- assertEquals(Compilation.Status.SUCCESS, compilation.status());
- assertEquals(annotationProcessor.getStorage().getClass(), LegacyExtensionStorage.class);
- }
-
- @Test
- public void compileWithoutError() {
- Compilation compilation = compile(JavaSources.Greeting, JavaSources.WhazzupGreeting);
- assertThat(compilation).succeededWithoutWarnings();
- }
-
- @Test
- public void compileWithError() {
- Compilation compilation = compile(JavaSources.Greeting, WhazzupGreeting_NoExtensionPoint);
- assertThat(compilation).failed();
- assertThat(compilation).hadErrorContaining("it doesn't implement ExtensionPoint")
- .inFile(WhazzupGreeting_NoExtensionPoint)
- .onLine(5)
- .atColumn(8);
- }
-
- @Test
- public void getExtensions() {
- Compilation compilation = compile(JavaSources.Greeting, JavaSources.WhazzupGreeting);
- assertThat(compilation).succeededWithoutWarnings();
- Map<String, Set<String>> extensions = new HashMap<>();
- extensions.put("test.Greeting", new HashSet<>(Collections.singletonList("test.WhazzupGreeting")));
- assertEquals(extensions, annotationProcessor.getExtensions());
- }
-
- @Test
- public void compileNestedExtensionAnnotation() {
- Compilation compilation = compile(JavaSources.Greeting, SpinnakerExtension, WhazzupGreeting_SpinnakerExtension);
- assertThat(compilation).succeededWithoutWarnings();
- Map<String, Set<String>> extensions = new HashMap<>();
- extensions.put("test.Greeting", new HashSet<>(Collections.singletonList("test.WhazzupGreeting")));
- assertEquals(extensions, annotationProcessor.getExtensions());
- }
-
- private Compiler compiler() {
- return javac().withProcessors(annotationProcessor);
- }
-
- private Compilation compile(JavaFileObject... sources) {
- return compiler().compile(sources);
- }
-
- }
|