JAL-3829 mocking for PDBFTS for related query to UniprotFTS mock.

[jalview.git] / src / jalview / fts / core / FTSRestClient.java

diff --git a/src/jalview/fts/core/FTSRestClient.java b/src/jalview/fts/core/FTSRestClient.java
index f651707..b262f64 100644 (file)
--- a/src/jalview/fts/core/FTSRestClient.java
+++ b/src/jalview/fts/core/FTSRestClient.java
@@ -31,6 +31,7 @@ import java.util.Objects;
 import jalview.fts.api.FTSDataColumnI;
 import jalview.fts.api.FTSDataColumnI.FTSDataColumnGroupI;
 import jalview.fts.core.FTSDataColumnPreferences.PreferenceSource;
+import jalview.fts.service.threedbeacons.TDBeaconsFTSRestClient;
 import jalview.fts.api.FTSRestClientI;
 
 /**
@@ -57,6 +58,10 @@ public abstract class FTSRestClient implements FTSRestClientI
 
   private int defaultResponsePageSize = 100;
 
+  protected String mockQuery = null;
+
+  protected String mockResponse = null;
+
   protected FTSRestClient()
   {
 
@@ -428,6 +433,12 @@ public abstract class FTSRestClient implements FTSRestClientI
             "Couldn't find data column with name : " + nameOrCode);
   }
 
+  public static void createMockFTSRestClient(FTSRestClient instance,String workingQuery,
+          String jsonResponse)
+  {
+    instance.setMock(workingQuery,jsonResponse);
+  }
+
   @Override
   public FTSDataColumnGroupI getDataColumnGroupById(String id)
           throws Exception
@@ -481,6 +492,12 @@ public abstract class FTSRestClient implements FTSRestClientI
     return String.valueOf(code) + " " + message;
   }
 
+  public static void unMock(FTSRestClient instance)
+  {
+    instance.mockQuery =null;
+    instance.mockResponse =null;
+  }
+
   protected String getResourceFile(String fileName)
   {
     String result = "";
@@ -505,6 +522,17 @@ public abstract class FTSRestClient implements FTSRestClientI
     return defaultResponsePageSize;
   }
 
+  protected void setMock(String workingQuery, String jsonResponse)
+  {
+    mockQuery=workingQuery;
+    mockResponse = jsonResponse;
+  }
+
+  protected boolean isMocked()
+  {
+    return mockQuery != null && mockResponse!=null;
+  }
+
   @Override
   public String[] getPreferencesColumnsFor(PreferenceSource source)
   {