Support PropertySource annotation analysis

Author: EgorkaKulikov

utbot-spring-analyzer/src/main/java/analyzer/ConfigurationManager.java

@@ -0,0 +1,31 @@
+package analyzer;
+
+import java.lang.reflect.Field;
+import java.lang.reflect.InvocationHandler;
+import java.util.Map;
+
+public class ConfigurationManager {
+
+    private final ClassLoader classLoader;
+    private final Class userConfigurationClass;
+
+    public ConfigurationManager(ClassLoader classLoader, Class userConfigurationClass) {
+        this.classLoader = classLoader;
+        this.userConfigurationClass = userConfigurationClass;
+    }
+
+    public void patchPropertySourceAnnotation() throws IllegalAccessException, NoSuchFieldException, ClassNotFoundException {
+
+        Class<?> proxyClass = classLoader.loadClass("java.lang.reflect.Proxy");
+        Field hField = proxyClass.getDeclaredField("h");
+        hField.setAccessible(true);
+
+        InvocationHandler o = (InvocationHandler) (hField.get(userConfigurationClass.getAnnotations()[2]));
+
+        Class<?> annotationInvocationHandlerClass = classLoader.loadClass("sun.reflect.annotation.AnnotationInvocationHandler");
+        Field memberValuesField = annotationInvocationHandlerClass.getDeclaredField("memberValues");
+        memberValuesField.setAccessible(true);
+        Map<String, Object> memberValues = (Map<String, Object>) (memberValuesField.get(o));
+        memberValues.put("value", "classpath:fakeapplication.properties");
+    }
+}

utbot-spring-analyzer/src/main/java/analyzer/PropertiesAnalyzer.java

@@ -0,0 +1,30 @@
+package analyzer;
+
+import java.io.BufferedReader;
+import java.io.FileReader;
+import java.io.IOException;
+import java.util.ArrayList;
+
+public class PropertiesAnalyzer {
+
+    private final String propertiesFilePath;
+
+    public PropertiesAnalyzer(String propertiesFilePath) {
+        this.propertiesFilePath = propertiesFilePath;
+    }
+
+    public ArrayList<String> readProperties() throws IOException {
+        ArrayList<String> props = new ArrayList<>();
+
+        BufferedReader reader = new BufferedReader(new FileReader(propertiesFilePath));
+        String line = reader.readLine();
+        while (line != null) {
+            props.add(line);
+            line = reader.readLine();
+        }
+
+        reader.close();
+
+        return props;
+    }
+}

utbot-spring-analyzer/src/main/java/analyzer/SpringAnalysisRunner.java

@@ -0,0 +1,43 @@
+package analyzer;
+
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.builder.SpringApplicationBuilder;
+import org.springframework.context.ApplicationContextException;
+
+import java.io.IOException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.nio.file.Path;
+
+@SpringBootApplication
+public class SpringAnalysisRunner {
+
+    public static void main(String[] args) throws ClassNotFoundException, IOException, NoSuchFieldException, IllegalAccessException {
+        //String arg0 = "D:\\Projects\\spring-starter-lesson-28\\build\\classes\\java\\main";
+        //String arg1 = "com.dmdev.spring.config.ApplicationConfiguration";
+        //String arg2 = "D:\\Projects\\spring-starter-lesson-28\\build\\resources\\main\\application.properties";
+
+        ClassLoader classLoader = new URLClassLoader(new URL[]{Path.of(args[0]).toUri().toURL()});
+        Class<?> userConfigurationClass = classLoader.loadClass(args[1]);
+
+        ConfigurationManager configurationManager = new ConfigurationManager(classLoader, userConfigurationClass);
+        PropertiesAnalyzer propertiesAnalyzer = new PropertiesAnalyzer(args[2]);
+
+        configurationManager.patchPropertySourceAnnotation();
+
+        SpringApplicationBuilder app = new SpringApplicationBuilder(SpringAnalysisRunner.class)
+                .sources(TestApplicationConfiguration.class, userConfigurationClass);
+
+        for (String prop : propertiesAnalyzer.readProperties()) {
+            app.properties(prop);
+        }
+
+        try {
+            app.build();
+            app.run();
+        } catch (ApplicationContextException e) {
+            System.out.println("Bean analysis finished successfully");
+        }
+    }
+
+}