Merge branch 'develop' into update_212_Dec_merge_with_21125_chamges
[jalview.git] / test / jalview / ws / jabaws / DisorderAnnotExportImport.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 import static org.testng.AssertJUnit.assertNotNull;
25 import static org.testng.AssertJUnit.assertTrue;
26
27 import jalview.bin.Cache;
28 import jalview.bin.Console;
29 import jalview.datamodel.AlignmentAnnotation;
30 import jalview.datamodel.AlignmentI;
31 import jalview.gui.JvOptionPane;
32 import jalview.io.AnnotationFile;
33 import jalview.io.DataSourceType;
34 import jalview.io.FileFormat;
35 import jalview.io.FormatAdapter;
36 import jalview.io.StockholmFileTest;
37 import jalview.ws.api.ServiceWithParameters;
38 import jalview.ws.jws2.Jws2Discoverer;
39 import jalview.ws.jws2.SeqAnnotationServiceCalcWorker;
40 import jalview.ws.slivkaws.SlivkaWSDiscoverer;
41
42 import java.util.ArrayList;
43 import java.util.List;
44
45 import org.testng.Assert;
46 import org.testng.annotations.AfterClass;
47 import org.testng.annotations.BeforeClass;
48 import org.testng.annotations.DataProvider;
49 import org.testng.annotations.Test;
50
51 /*
52  * All methods in this class are set to the Network group because setUpBeforeClass will fail
53  * if there is no network.
54  */
55 @Test(singleThreaded = true)
56 public class DisorderAnnotExportImport
57 {
58
59   @BeforeClass(alwaysRun = true)
60   public void setUpJvOptionPane()
61   {
62     JvOptionPane.setInteractiveMode(false);
63     JvOptionPane.setMockResponse(JvOptionPane.CANCEL_OPTION);
64   }
65
66   public static String testseqs = "examples/uniref50.fa";
67
68   public static Jws2Discoverer disc;
69
70   public static List<ServiceWithParameters> iupreds;
71
72   jalview.ws.jws2.SeqAnnotationServiceCalcWorker disorderClient;
73
74   public static jalview.gui.AlignFrame af = null;
75
76   @BeforeClass(alwaysRun = true)
77   public static void setUpBeforeClass() throws Exception
78   {
79     Cache.loadProperties("test/jalview/io/testProps.jvprops");
80     Console.initLogger();
81     disc = JalviewJabawsTestUtils.getJabawsDiscoverer();
82
83     while (disc.isRunning())
84     {
85       // don't get services until discoverer has finished
86       Thread.sleep(100);
87     }
88
89     SlivkaWSDiscoverer disc2 = SlivkaWSDiscoverer.getInstance();
90     disc2.startDiscoverer();
91     while (disc2.isRunning())
92     {
93       if (svc.getServiceTypeURI().toLowerCase(Locale.ROOT)
94               .contains("iupredws"))
95       {
96         iupreds.add(svc);
97       }
98     }
99     assertTrue("Couldn't discover any IUPred services to use to test.",
100             iupreds.size() > 0);
101     jalview.io.FileLoader fl = new jalview.io.FileLoader(false);
102     af = fl.LoadFileWaitTillLoaded(testseqs,
103             jalview.io.DataSourceType.FILE);
104     assertNotNull("Couldn't load test data ('" + testseqs + "')", af);
105   }
106
107   @AfterClass(alwaysRun = true)
108   public static void tearDownAfterClass() throws Exception
109   {
110     if (af != null)
111     {
112       af.setVisible(false);
113       af.dispose();
114       af = null;
115     }
116   }
117   
118   @DataProvider(name="getIuPreds",parallel = false)
119   public static ServiceWithParameters[][] getIuPreds()
120   {
121     ServiceWithParameters[][] services = new ServiceWithParameters[iupreds
122             .size()][1];
123
124     int i = 0;
125     for (ServiceWithParameters iupred : iupreds)
126     {
127       services[i++][0] = iupred;
128     }
129     return services;
130   }
131
132   /**
133    * test for patches to JAL-1294
134    */
135   @Test(groups = { "External", "Network" },dataProvider = "getIuPreds")
136   public void testDisorderAnnotExport(ServiceWithParameters iuPred)
137   {
138     disorderClient = new SeqAnnotationServiceCalcWorker(iuPred, af, null,
139             null);
140     af.getViewport().getCalcManager().startWorker(disorderClient);
141     do
142     {
143       try
144       {
145         Thread.sleep(50);
146       } catch (InterruptedException x)
147       {
148       }
149       ;
150     } while (af.getViewport().getCalcManager().isWorking());
151     AlignmentI orig_alig = af.getViewport().getAlignment();
152     // NOTE: Consensus annotation row cannot be exported and reimported
153     // faithfully - so we remove them
154     List<AlignmentAnnotation> toremove = new ArrayList<>();
155     for (AlignmentAnnotation aa : orig_alig.getAlignmentAnnotation())
156     {
157       if (aa.autoCalculated)
158       {
159         toremove.add(aa);
160       }
161     }
162     for (AlignmentAnnotation aa : toremove)
163     {
164       orig_alig.deleteAnnotation(aa);
165     }
166     checkAnnotationFileIO("Testing IUPred Annotation IO", orig_alig);
167
168   }
169
170   static void checkAnnotationFileIO(String testname, AlignmentI al)
171   {
172     try
173     {
174       String aligfileout = FileFormat.Pfam.getWriter(al)
175               .print(al.getSequencesArray(), true);
176       String anfileout = new AnnotationFile()
177               .printAnnotationsForAlignment(al);
178       assertTrue("Test " + testname
179               + "\nAlignment annotation file was not regenerated. Null string",
180               anfileout != null);
181       assertTrue("Test " + testname
182               + "\nAlignment annotation file was not regenerated. Empty string",
183               anfileout.length() > "JALVIEW_ANNOTATION".length());
184
185       System.out.println(
186               "Output annotation file:\n" + anfileout + "\n<<EOF\n");
187
188       AlignmentI al_new = new FormatAdapter().readFile(aligfileout,
189               DataSourceType.PASTE, FileFormat.Pfam);
190       assertTrue("Test " + testname
191               + "\nregenerated annotation file did not annotate alignment.",
192               new AnnotationFile().readAnnotationFile(al_new, anfileout,
193                       DataSourceType.PASTE));
194
195       // test for consistency in io
196       StockholmFileTest.testAlignmentEquivalence(al, al_new, true, false,
197               false);
198       return;
199     } catch (Exception e)
200     {
201       e.printStackTrace();
202     }
203     Assert.fail("Test " + testname
204             + "\nCouldn't complete Annotation file roundtrip input/output/input test.");
205   }
206
207 }