Change the import test to use 2 clusters if available

Author: ScottDugas

fdb-record-layer-core/src/test/java/com/apple/foundationdb/record/provider/foundationdb/keyspace/KeySpacePathImportDataTest.java

@@ -55,10 +55,12 @@ class KeySpacePathImportDataTest {
     @RegisterExtension
     final FDBDatabaseExtension dbExtension = new FDBDatabaseExtension();
     private FDBDatabase database;
+    private List<FDBDatabase> databases;
 
     @BeforeEach
     void setUp() {
-        database = dbExtension.getDatabase();
+        databases = dbExtension.getDatabases(2);
+        database = databases.get(0);
     }
 
     @Test
@@ -395,8 +397,12 @@ private void copyData(final KeySpacePath path) {
         // Export the data
         final List<DataInKeySpacePath> exportedData = getExportedData(path);
 
-        // Clear the data and import it back
-        clearPath(database, path);
+        if (databases.size() > 1) {
+            database = databases.get(1);
+        } else {
+            // Clear the data and import it back
+            clearPath(database, path);
+        }
 
         // Import the data
         importData(database, path, exportedData);

fdb-record-layer-core/src/test/java/com/apple/foundationdb/record/test/FDBDatabaseExtension.java

@@ -36,10 +36,14 @@
 
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
+import java.util.ArrayList;
+import java.util.Collections;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 import java.util.Objects;
 import java.util.concurrent.Executor;
+import java.util.stream.Collectors;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
@@ -161,6 +165,12 @@ public FDBDatabase getDatabase(@Nullable String clusterFile) {
                 });
     }
 
+    public List<FDBDatabase> getDatabases(int count) {
+        List<String> clusterFiles = new ArrayList<>(FDBTestEnvironment.allClusterFiles());
+        Collections.shuffle(clusterFiles);
+        return clusterFiles.stream().limit(count).map(this::getDatabase).collect(Collectors.toList());
+    }
+
     public void checkForOpenContexts() {
         for (final Map.Entry<String, FDBDatabase> clusterFileToDatabase : databases.entrySet()) {
             assertEquals(0, clusterFileToDatabase.getValue().warnAndCloseOldTrackedOpenContexts(0),