calculate default schemas and provide getter; handle test env.

PhilippSalvisberg · PhilippSalvisberg · commit c2e7b12dc804 · 2020-05-30T21:27:40.000+02:00
diff --git a/sqldev/src/main/java/org/utplsql/sqldev/coverage/CodeCoverageReporter.java b/sqldev/src/main/java/org/utplsql/sqldev/coverage/CodeCoverageReporter.java
@@ -22,7 +22,11 @@
 import java.sql.Connection;
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Optional;
 import java.util.logging.Logger;
 
 import org.utplsql.sqldev.dal.RealtimeReporterDao;
@@ -53,6 +57,7 @@ public CodeCoverageReporter(final List<String> pathList, final List<String> incl
             final String connectionName) {
         this.pathList = pathList;
         this.includeObjectList = includeObjectList;
+        setDefaultSchema();
         setConnection(connectionName);
     }
 
@@ -62,6 +67,7 @@ public CodeCoverageReporter(final List<String> pathList, final List<String> incl
         this.pathList = pathList;
         this.includeObjectList = includeObjectList;
         this.conn = conn;
+        setDefaultSchema();
     }
 
     private void setConnection(final String connectionName) {
@@ -75,6 +81,31 @@ private void setConnection(final String connectionName) {
             this.conn = DatabaseTools.getConnection(connectionName);
         }
     }
+    
+    private void setDefaultSchema() {
+        if (includeObjectList != null && !includeObjectList.isEmpty()) {
+            // use the owner with the most hits in includeObjectList
+            HashMap<String, Integer> owners = new HashMap<>();
+            for (String entry : includeObjectList) {
+                String[] obj = entry.toUpperCase().split("\\.");
+                if (obj.length == 2) {
+                    // only if objectOwner and objectName are available
+                    Integer count = owners.get(obj[0]);
+                    if (count == null) {
+                        count = 1;
+                    } else {
+                        count++;
+                    }
+                    owners.put(obj[0], count);
+                }
+            }
+            Optional<Entry<String, Integer>> top = owners.entrySet().stream()
+                    .sorted(Map.Entry.<String, Integer>comparingByValue().reversed()).findFirst();
+            if (top.isPresent()) {
+                schemas = top.get().getKey();
+            }
+        }
+    }
 
     private ArrayList<String> toStringList(final String s) {
         final ArrayList<String> list = new ArrayList<>();
@@ -92,8 +123,14 @@ private void run() {
         logger.fine(() -> "Running code coverage reporter for " + pathList + "...");
         try {
             final RealtimeReporterDao dao = new RealtimeReporterDao(conn);
-            final PreferenceModel preferences = PreferenceModel.getInstance(Preferences.getPreferences());
-            if (preferences.isUseRealtimeReporter() && dao.isSupported()) {
+            PreferenceModel preferences;
+            try {
+                preferences = PreferenceModel.getInstance(Preferences.getPreferences());
+            } catch (NoClassDefFoundError error) {
+                // not running in SQL Developer (in tests)
+                preferences = PreferenceModel.getInstance(null);
+            }
+            if (preferences.isUseRealtimeReporter() && dao.isSupported() && connectionName != null) {
                 runCodeCoverageWithRealtimeReporter();
             } else {
                 runCodeCoverageStandalone();
@@ -181,6 +218,10 @@ public void setSchemas(final String schemas) {
         this.schemas = schemas;
     }
 
+    public String getSchemas() {
+        return schemas;
+    }
+
     public void setIncludeObjects(final String includeObjects) {
         this.includeObjects = includeObjects;
     }
