JAL-1601 Patch JPred4 service tests.

[jalview.git] / src / jalview / ws2 / client / jpred4 / JPred4WSClient.java

diff --git a/src/jalview/ws2/client/jpred4/JPred4WSClient.java b/src/jalview/ws2/client/jpred4/JPred4WSClient.java
index 36cc6c2..3782723 100644 (file)
--- a/src/jalview/ws2/client/jpred4/JPred4WSClient.java
+++ b/src/jalview/ws2/client/jpred4/JPred4WSClient.java
@@ -23,8 +23,14 @@ public class JPred4WSClient implements SecStructPredWebServiceClientI
 {
   final JPredClient client;
 
-  JPred4WSClient(JPredClient client)
+  private final JPredRequest.Mode mode;
+
+  private final JPredRequest.Format format;
+
+  JPred4WSClient(JPredClient client, boolean msaMode)
   {
+    this.mode = msaMode ? JPredRequest.Mode.MSA : JPredRequest.Mode.SINGLE;
+    this.format = msaMode ? JPredRequest.Format.FASTA : JPredRequest.Format.SEQ;
     this.client = client;
   }
 
@@ -44,13 +50,16 @@ public class JPred4WSClient implements SecStructPredWebServiceClientI
   public WebServiceJobHandle submit(List<SequenceI> sequences,
       List<ArgumentI> args, Credentials credentials) throws IOException
   {
-    var input = FileFormat.Fasta
-        .getWriter(null)
-        .print(sequences.toArray(SequenceI[]::new), false);
+    var input = mode == JPredRequest.Mode.MSA
+        ? FileFormat.Fasta
+            .getWriter(null)
+            .print(sequences.toArray(new SequenceI[sequences.size()]), false)
+        : sequences.get(0).getSequenceAsString();
+
     JPredRequest request = JPredRequest
         .newBuilder()
-        .mode(JPredRequest.Mode.MSA)
-        .format(JPredRequest.Format.FASTA)
+        .mode(mode)
+        .format(format)
         .input(input)
         .build();
     var jobId = client.submit(request);
@@ -63,6 +72,8 @@ public class JPred4WSClient implements SecStructPredWebServiceClientI
     var status = client.getStatus(job.getJobId());
     switch (status)
     {
+    case PENDING:
+      return JobStatus.QUEUED;
     case RUNNING:
       return JobStatus.RUNNING;
     case FINISHED:
@@ -96,16 +107,16 @@ public class JPred4WSClient implements SecStructPredWebServiceClientI
   public void cancel(WebServiceJobHandle job)
       throws IOException, UnsupportedOperationException
   {
-    
+
   }
-  
+
   @Override
   public AlignFile getAlignmentFile(WebServiceJobHandle job) throws IOException
   {
     var url = client.getResultURL(job.getJobId(), ResultTypes.PSIBLAST_ALIGN);
     return new FastaFile(url.toString(), DataSourceType.URL);
   }
-  
+
   @Override
   public JPredFile getPredictionFile(WebServiceJobHandle job) throws IOException
   {