1401f6a7cc788cad5a4fb6a4ed79a200f862c63d
[jalview.git] / test / jalview / ws / PDBSequenceFetcherTest.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;
22
23 import static org.testng.AssertJUnit.assertTrue;
24
25 import jalview.bin.Cache;
26 import jalview.datamodel.AlignmentI;
27 import jalview.datamodel.SequenceI;
28 import jalview.structure.StructureImportSettings;
29 import jalview.structure.StructureImportSettings.StructureParser;
30 import jalview.ws.seqfetcher.DbSourceProxy;
31
32 import java.util.List;
33
34 import org.testng.annotations.BeforeMethod;
35 import org.testng.annotations.Test;
36
37 public class PDBSequenceFetcherTest
38 {
39
40   SequenceFetcher sf;
41
42   @BeforeMethod(alwaysRun = true)
43   public void setUp() throws Exception
44   {
45     Cache.loadProperties("test/jalview/io/testProps.jvprops");
46     // ensure 'add annotation from structure' is selected
47     Cache.applicationProperties.setProperty("STRUCT_FROM_PDB",
48             Boolean.TRUE.toString());
49     Cache.applicationProperties.setProperty("ADD_SS_ANN",
50             Boolean.TRUE.toString());
51
52     sf = new SequenceFetcher(false);
53   }
54
55   /**
56    * Test that RNA structure can be added by a call to the RNAML service.
57    * 
58    * Note this test depends on http://arn-ibmc.in2p3.fr/api/compute/2d which is
59    * not always reliable.
60    * 
61    * @throws Exception
62    */
63   @Test(groups = { "Network" }, enabled = true)
64   public void testRnaSeqRetrieve() throws Exception
65   {
66     Cache.applicationProperties.setProperty("PDB_DOWNLOAD_FORMAT",
67             "PDB");
68     List<DbSourceProxy> sps = sf.getSourceProxy("PDB");
69     AlignmentI response = sps.get(0).getSequenceRecords("2GIS");
70     assertTrue(response != null);
71     assertTrue(response.getHeight() == 1);
72     for (SequenceI sq : response.getSequences())
73     {
74       assertTrue("No annotation transfered to sequence.",
75               sq.getAnnotation().length > 0);
76       assertTrue("No PDBEntry on sequence.",
77               sq.getAllPDBEntries().size() > 0);
78       assertTrue(
79               "No RNA annotation on sequence, possibly http://arn-ibmc.in2p3.fr/api/compute/2d not available?",
80               sq.getRNA() != null);
81     }
82   }
83
84   @Test(groups = { "Network" }, enabled = true)
85   public void testPdbSeqRetrieve() throws Exception
86   {
87     StructureImportSettings.setDefaultStructureFileFormat("PDB");
88     StructureImportSettings
89             .setDefaultPDBFileParser(StructureParser.JALVIEW_PARSER);
90
91     testRetrieveProteinSeqFromPDB();
92   }
93
94   @Test(groups = { "Network" }, enabled = true)
95   public void testmmCifSeqRetrieve() throws Exception
96   {
97     StructureImportSettings.setDefaultStructureFileFormat("mmCIF");
98     testRetrieveProteinSeqFromPDB();
99   }
100
101   private void testRetrieveProteinSeqFromPDB() throws Exception
102   {
103     List<DbSourceProxy> sps = sf.getSourceProxy("PDB");
104     AlignmentI response = sps.get(0).getSequenceRecords("1QIP");
105     assertTrue(response != null);
106     assertTrue(response.getHeight() == 4);
107     for (SequenceI sq : response.getSequences())
108     {
109       assertTrue("No annotation transfered to sequence.",
110               sq.getAnnotation().length > 0);
111       assertTrue("No PDBEntry on sequence.",
112               sq.getAllPDBEntries().size() > 0);
113       org.testng.Assert
114               .assertEquals(sq.getEnd() - sq.getStart() + 1,
115                       sq.getLength(),
116                       "Sequence start/end doesn't match number of residues in sequence");
117     }
118   }
119
120 }