test/jalview/ws/jabaws/RNAStructExportImport.java

   1 /*
   2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
   3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
   4  *
   5  * This file is part of Jalview.
   6  *
   7  * Jalview is free software: you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation, either version 3
  10  * of the License, or (at your option) any later version.
  11  *
  12  * Jalview is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty
  14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  15  * PURPOSE.  See the GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
  19  * The Jalview Authors are detailed in the 'AUTHORS' file.
  20  */
  21 package jalview.ws.jabaws;
  22
  23 import java.util.Locale;
  24
  25 import static org.testng.AssertJUnit.assertNotNull;
  26 import static org.testng.AssertJUnit.assertTrue;
  27
  28 import jalview.bin.Cache;
  29 import jalview.bin.Console;
  30 import jalview.datamodel.AlignmentAnnotation;
  31 import jalview.datamodel.AlignmentI;
  32 import jalview.gui.JvOptionPane;
  33 import jalview.io.AnnotationFile;
  34 import jalview.io.DataSourceType;
  35 import jalview.io.FileFormat;
  36 import jalview.io.FormatAdapter;
  37 import jalview.io.StockholmFileTest;
  38 import jalview.project.Jalview2XML;
  39 import jalview.ws.jws2.Jws2Discoverer;
  40 import jalview.ws.jws2.RNAalifoldClient;
  41 import jalview.ws.jws2.SequenceAnnotationWSClient;
  42 import jalview.ws.jws2.jabaws2.Jws2Instance;
  43 import jalview.ws.params.AutoCalcSetting;
  44
  45 import java.awt.Component;
  46 import java.io.File;
  47 import java.util.ArrayList;
  48 import java.util.List;
  49
  50 import javax.swing.JMenu;
  51 import javax.swing.JMenuItem;
  52
  53 import org.testng.Assert;
  54 import org.testng.annotations.AfterClass;
  55 import org.testng.annotations.BeforeClass;
  56 import org.testng.annotations.Test;
  57
  58 import compbio.metadata.Argument;
  59 import compbio.metadata.WrongParameterException;
  60
  61 /*
  62  * All methods in this class are set to the Network group because setUpBeforeClass will fail
  63  * if there is no network.
  64  */
  65 @Test(singleThreaded = true)
  66 public class RNAStructExportImport
  67 {
  68
  69   @BeforeClass(alwaysRun = true)
  70   public void setUpJvOptionPane()
  71   {
  72     JvOptionPane.setInteractiveMode(false);
  73     JvOptionPane.setMockResponse(JvOptionPane.CANCEL_OPTION);
  74   }
  75
  76   private static final String JAR_FILE_NAME = "testRnalifold_param.jar";
  77
  78   public static String testseqs = "examples/RF00031_folded.stk";
  79
  80   public static Jws2Discoverer disc;
  81
  82   public static Jws2Instance rnaalifoldws;
  83
  84   jalview.ws.jws2.RNAalifoldClient alifoldClient;
  85
  86   public static jalview.gui.AlignFrame af = null;
  87
  88   @BeforeClass(alwaysRun = true)
  89   public static void setUpBeforeClass() throws Exception
  90   {
  91     Cache.loadProperties("test/jalview/io/testProps.jvprops");
  92     Console.initLogger();
  93     disc = JalviewJabawsTestUtils.getJabawsDiscoverer(false);
  94
  95     while (disc.isRunning())
  96     {
  97       // don't get services until discoverer has finished
  98       Thread.sleep(100);
  99     }
 100
 101     for (Jws2Instance svc : disc.getServices())
 102     {
 103
 104       if (svc.getServiceTypeURI().toLowerCase(Locale.ROOT).contains("rnaalifoldws"))
 105       {
 106         rnaalifoldws = svc;
 107       }
 108     }
 109
 110     System.out.println("State of rnaalifoldws: " + rnaalifoldws);
 111
 112     if (rnaalifoldws == null)
 113     {
 114       Assert.fail("no web service");
 115     }
 116
 117     jalview.io.FileLoader fl = new jalview.io.FileLoader(false);
 118
 119     af = fl.LoadFileWaitTillLoaded(testseqs, jalview.io.DataSourceType.FILE);
 120
 121     assertNotNull("Couldn't load test data ('" + testseqs + "')", af);
 122
 123     // remove any existing annotation
 124     List<AlignmentAnnotation> aal = new ArrayList<>();
 125     for (AlignmentAnnotation rna : af.getViewport().getAlignment()
 126             .getAlignmentAnnotation())
 127     {
 128       if (rna.isRNA())
 129       {
 130         aal.add(rna);
 131       }
 132     }
 133     for (AlignmentAnnotation rna : aal)
 134     {
 135       af.getViewport().getAlignment().deleteAnnotation(rna);
 136     }
 137     af.getViewport().alignmentChanged(af.alignPanel); // why is af.alignPanel
 138                                                       // public?
 139   }
 140
 141   @AfterClass(alwaysRun = true)
 142   public static void tearDownAfterClass() throws Exception
 143   {
 144     if (af != null)
 145     {
 146       af.setVisible(false);
 147       af.dispose();
 148       File f = new File(JAR_FILE_NAME);
 149       if (f.exists())
 150       {
 151         f.delete();
 152       }
 153     }
 154   }
 155
 156   @Test(groups = { "Network" })
 157   public void testRNAAliFoldValidStructure()
 158   {
 159
 160     alifoldClient = new RNAalifoldClient(rnaalifoldws, af, null, null);
 161
 162     af.getViewport().getCalcManager().startWorker(alifoldClient);
 163
 164     do
 165     {
 166       try
 167       {
 168         Thread.sleep(50);
 169       } catch (InterruptedException x)
 170       {
 171       }
 172     } while (af.getViewport().getCalcManager().isWorking());
 173
 174     AlignmentI orig_alig = af.getViewport().getAlignment();
 175     for (AlignmentAnnotation aa : orig_alig.getAlignmentAnnotation())
 176     {
 177       if (alifoldClient.involves(aa))
 178       {
 179         if (aa.isRNA())
 180         {
 181           assertTrue(
 182                   "Did not create valid structure from RNAALiFold prediction",
 183                   aa.isValidStruc());
 184         }
 185       }
 186     }
 187   }
 188
 189   @Test(groups = { "Network" })
 190   public void testRNAStructExport()
 191   {
 192
 193     alifoldClient = new RNAalifoldClient(rnaalifoldws, af, null, null);
 194
 195     af.getViewport().getCalcManager().startWorker(alifoldClient);
 196
 197     do
 198     {
 199       try
 200       {
 201         Thread.sleep(50);
 202       } catch (InterruptedException x)
 203       {
 204       }
 205     } while (af.getViewport().getCalcManager().isWorking());
 206
 207     AlignmentI orig_alig = af.getViewport().getAlignment();
 208     // JBPNote: this assert fails (2.10.2) because the 'Reference Positions'
 209     // annotation is mistakenly recognised as an RNA annotation row when read in
 210     // as an annotation file.
 211     verifyAnnotationFileIO("Testing RNAalifold Annotation IO", orig_alig);
 212
 213   }
 214
 215   static void verifyAnnotationFileIO(String testname, AlignmentI al)
 216   {
 217     try
 218     {
 219       // what format would be appropriate for RNAalifold annotations?
 220       String aligfileout = FileFormat.Pfam.getWriter(null).print(
 221               al.getSequencesArray(), true);
 222
 223       String anfileout = new AnnotationFile()
 224               .printAnnotationsForAlignment(al);
 225       assertNotNull(
 226               "Test "
 227                       + testname
 228                       + "\nAlignment annotation file was not regenerated. Null string",
 229               anfileout);
 230       assertTrue(
 231               "Test "
 232                       + testname
 233                       + "\nAlignment annotation file was not regenerated. Empty string",
 234               anfileout.length() > "JALVIEW_ANNOTATION".length());
 235
 236       System.out.println("Output annotation file:\n" + anfileout
 237               + "\n<<EOF\n");
 238
 239       // again what format would be appropriate?
 240       AlignmentI al_new = new FormatAdapter().readFile(aligfileout,
 241               DataSourceType.PASTE, FileFormat.Pfam);
 242       assertTrue(
 243               "Test "
 244                       + testname
 245                       + "\nregenerated annotation file did not annotate alignment.",
 246               new AnnotationFile().readAnnotationFile(al_new, anfileout,
 247                       DataSourceType.PASTE));
 248
 249       // test for consistency in io
 250       StockholmFileTest.testAlignmentEquivalence(al, al_new, false, false,
 251               false);
 252       return;
 253     } catch (Exception e)
 254     {
 255       e.printStackTrace();
 256     }
 257     Assert.fail("Test "
 258             + testname
 259             + "\nCouldn't complete Annotation file roundtrip input/output/input test.");
 260   }
 261
 262   @Test(groups = { "Network" })
 263   public void testRnaalifoldSettingsRecovery()
 264   {
 265     List<Argument> opts = new ArrayList<>();
 266     for (Argument rg : (List<Argument>) rnaalifoldws.getRunnerConfig()
 267             .getArguments())
 268     {
 269       if (rg.getDescription().contains("emperature"))
 270       {
 271         try
 272         {
 273           rg.setValue("292");
 274         } catch (WrongParameterException q)
 275         {
 276           Assert.fail("Couldn't set the temperature parameter "
 277                   + q.getStackTrace());
 278         }
 279         opts.add(rg);
 280       }
 281       if (rg.getDescription().contains("max"))
 282       {
 283         opts.add(rg);
 284       }
 285     }
 286     alifoldClient = new RNAalifoldClient(rnaalifoldws, af, null, opts);
 287
 288     af.getViewport().getCalcManager().startWorker(alifoldClient);
 289
 290     do
 291     {
 292       try
 293       {
 294         Thread.sleep(50);
 295       } catch (InterruptedException x)
 296       {
 297       }
 298       ;
 299     } while (af.getViewport().getCalcManager().isWorking());
 300     AutoCalcSetting oldacs = af.getViewport().getCalcIdSettingsFor(
 301             alifoldClient.getCalcId());
 302     String oldsettings = oldacs.getWsParamFile();
 303     // write out parameters
 304     jalview.gui.AlignFrame nalf = null;
 305     assertTrue("Couldn't write out the Jar file",
 306             new Jalview2XML(false).saveAlignment(af, JAR_FILE_NAME,
 307                     "trial parameter writeout"));
 308     assertTrue("Couldn't read back the Jar file", (nalf = new Jalview2XML(
 309             false).loadJalviewAlign(JAR_FILE_NAME)) != null);
 310     if (nalf != null)
 311     {
 312       AutoCalcSetting acs = af.getViewport().getCalcIdSettingsFor(
 313               alifoldClient.getCalcId());
 314       assertTrue("Calc ID settings not recovered from viewport stash",
 315               acs.equals(oldacs));
 316       assertTrue(
 317               "Serialised Calc ID settings not identical to those recovered from viewport stash",
 318               acs.getWsParamFile().equals(oldsettings));
 319       JMenu nmenu = new JMenu();
 320       new SequenceAnnotationWSClient().attachWSMenuEntry(nmenu,
 321               rnaalifoldws, af);
 322       assertTrue("Couldn't get menu entry for service",
 323               nmenu.getItemCount() > 0);
 324       for (Component itm : nmenu.getMenuComponents())
 325       {
 326         if (itm instanceof JMenuItem)
 327         {
 328           JMenuItem i = (JMenuItem) itm;
 329           if (i.getText().equals(
 330                   rnaalifoldws.getAlignAnalysisUI().getAAconToggle()))
 331           {
 332             i.doClick();
 333             break;
 334           }
 335         }
 336       }
 337       while (af.getViewport().isCalcInProgress())
 338       {
 339         try
 340         {
 341           Thread.sleep(200);
 342         } catch (Exception x)
 343         {
 344         }
 345         ;
 346       }
 347       AutoCalcSetting acs2 = af.getViewport().getCalcIdSettingsFor(
 348               alifoldClient.getCalcId());
 349       assertTrue(
 350               "Calc ID settings after recalculation has not been recovered.",
 351               acs2.getWsParamFile().equals(oldsettings));
 352     }
 353   }
 354 }