further work on registry & WSTester nearly there
[jabaws.git] / webservices / compbio / ws / client / WSTester.java
index 05bd0b8..a0e635b 100644 (file)
@@ -29,8 +29,6 @@ import java.io.PrintWriter;
 import java.util.Arrays;\r
 import java.util.List;\r
 \r
-import javax.xml.ws.WebServiceException;\r
-\r
 import org.apache.log4j.Logger;\r
 \r
 import compbio.data.msa.JABAService;\r
@@ -52,6 +50,8 @@ import compbio.metadata.ResultNotAvailableException;
 import compbio.metadata.RunnerConfig;\r
 import compbio.metadata.UnsupportedRuntimeException;\r
 import compbio.metadata.WrongParameterException;\r
+import compbio.util.FileUtil;\r
+import compbio.util.Util;\r
 \r
 /**\r
  * Class for testing web services\r
@@ -89,7 +89,7 @@ public class WSTester {
         * \r
         * @return List of FastaSequence records\r
         */\r
-       static List<FastaSequence> loadSeqs() {\r
+       private static List<FastaSequence> loadSeqs() {\r
                try {\r
                        return SequenceUtil.readFasta(new ByteArrayInputStream(fastaInput\r
                                        .getBytes()));\r
@@ -105,7 +105,7 @@ public class WSTester {
         * \r
         * @return List of FastaSequence records\r
         */\r
-       static List<FastaSequence> loadAlignment() {\r
+       private static List<FastaSequence> loadAlignment() {\r
                try {\r
                        return SequenceUtil.readFasta(new ByteArrayInputStream(\r
                                        fastaAlignment.getBytes()));\r
@@ -117,27 +117,34 @@ public class WSTester {
        }\r
 \r
        private final PrintWriter writer;\r
+       private final String hostname;\r
 \r
-       public WSTester(PrintWriter writer) {\r
+       public WSTester(String hostname, PrintWriter writer) {\r
+               if (Util.isEmpty(hostname)) {\r
+                       throw new NullPointerException("Hostname must be provided!");\r
+               }\r
+               this.hostname = hostname;\r
                this.writer = writer;\r
        }\r
 \r
        /**\r
         * Prints usage\r
         */\r
-       void printUsage() {\r
-               writer.println("Usage: <Class or Jar file name> " + hostkey\r
+       static void printUsage() {\r
+               System.out.println("Usage: <Class or Jar file name> " + hostkey\r
                                + pseparator + "host_and_context " + "<" + servicekey\r
                                + pseparator + "serviceName>");\r
-               writer.println();\r
-               writer.println(hostkey\r
-                               + pseparator\r
-                               + "<host_and_context> - a full URL to the JABAWS web server including context path e.g. http://10.31.1.159:8080/ws");\r
-               writer.println(servicekey\r
-                               + pseparator\r
-                               + "<ServiceName> - optional if unspecified all services are tested otherwise one of "\r
-                               + Arrays.toString(Services.values()));\r
-               writer.println();\r
+               System.out.println();\r
+               System.out\r
+                               .println(hostkey\r
+                                               + pseparator\r
+                                               + "<host_and_context> - a full URL to the JABAWS web server including context path e.g. http://10.31.1.159:8080/ws");\r
+               System.out\r
+                               .println(servicekey\r
+                                               + pseparator\r
+                                               + "<ServiceName> - optional if unspecified all services are tested otherwise one of "\r
+                                               + Arrays.toString(Services.values()));\r
+               System.out.println();\r
 \r
        }\r
 \r
@@ -150,7 +157,7 @@ public class WSTester {
         *            list of the Preset\r
         * @throws UnsupportedRuntimeException\r
         */\r
-       <T> boolean presetAlign(MsaWS<T> msaws, List<Preset<T>> presets)\r
+       private <T> boolean presetAlign(MsaWS<T> msaws, List<Preset<T>> presets)\r
                        throws UnsupportedRuntimeException {\r
                boolean succeed = false;\r
                for (Preset<T> preset : presets) {\r
@@ -196,9 +203,9 @@ public class WSTester {
 \r
        private <T> boolean testMsaWS(MsaWS<T> msaws)\r
                        throws UnsupportedRuntimeException {\r
+               assert msaws != null;\r
 \r
                boolean succeed = testDefaultAlignment(msaws);\r
-\r
                // If exception above is thrown than the tests below is not run\r
 \r
                PresetManager<T> pmanager = msaws.getPresets();\r
@@ -222,12 +229,13 @@ public class WSTester {
         *             executable does not exists for a server platform\r
         */\r
        @SuppressWarnings("unchecked")\r
-       public <T> boolean checkService(JABAService wservice) {\r
+       private <T> boolean checkService(JABAService wservice) {\r
                try {\r
                        if (wservice == null) {\r
                                throw new NullPointerException(\r
                                                "JABAService instance must be provided!");\r
                        }\r
+\r
                        if (wservice instanceof MsaWS) {\r
                                return testMsaWS((MsaWS<T>) wservice);\r
                        } else if (wservice instanceof SequenceAnnotation) {\r
@@ -245,13 +253,16 @@ public class WSTester {
 \r
        private <T> boolean testSequenceAnnotationWS(SequenceAnnotation<T> wservice)\r
                        throws UnsupportedRuntimeException {\r
+               writer.println("Calling analyse.........");\r
                boolean success = testDefaultAnalyse(loadAlignment(), wservice, null,\r
                                null);\r
+\r
                PresetManager<T> presetman = wservice.getPresets();\r
                if (presetman != null) {\r
                        List<Preset<T>> presets = presetman.getPresets();\r
                        if (presets != null && !presets.isEmpty()) {\r
                                Preset<T> preset = presets.get(0);\r
+                               writer.println("Calling analyse with Preset.........");\r
                                success = testDefaultAnalyse(loadAlignment(), wservice, preset,\r
                                                null);\r
                        }\r
@@ -260,7 +271,7 @@ public class WSTester {
                return success;\r
        }\r
 \r
-       <T> boolean testDefaultAnalyse(List<FastaSequence> fastalist,\r
+       private <T> boolean testDefaultAnalyse(List<FastaSequence> fastalist,\r
                        SequenceAnnotation<T> wsproxy, Preset<T> preset,\r
                        List<Option<T>> customOptions) throws UnsupportedRuntimeException {\r
 \r
@@ -274,7 +285,6 @@ public class WSTester {
                        } else {\r
                                jobId = wsproxy.analize(fastalist);\r
                        }\r
-                       writer.println("\n\ncalling analise.........");\r
                        Thread.sleep(1000);\r
                        scores = wsproxy.getAnnotation(jobId);\r
 \r
@@ -344,7 +354,7 @@ public class WSTester {
         * @param msaws\r
         * @throws UnsupportedRuntimeException\r
         */\r
-       public <T> boolean testDefaultAlignment(MsaWS<T> msaws)\r
+       private <T> boolean testDefaultAlignment(MsaWS<T> msaws)\r
                        throws UnsupportedRuntimeException {\r
                writer.print("Testing alignment with default parameters:");\r
                Alignment al = null;\r
@@ -361,20 +371,14 @@ public class WSTester {
                        writer.print("Retrieving results...");\r
                        al = msaws.getResult(taskId);\r
                        succeed = true;\r
-               } catch (ResultNotAvailableException e) {\r
-                       writer.println(FAILED);\r
-                       writer.println(e.getLocalizedMessage());\r
-                       e.printStackTrace(writer);\r
-               } catch (JobSubmissionException e) {\r
-                       writer.println(FAILED);\r
-                       writer.println();\r
-                       writer.println(e.getLocalizedMessage());\r
-                       e.printStackTrace(writer);\r
-               } catch (InterruptedException e) {\r
-                       writer.println(FAILED);\r
-                       writer.println();\r
-                       writer.println(e.getLocalizedMessage());\r
-                       e.printStackTrace(writer);\r
+               } catch (Exception e) {\r
+                       if (e instanceof UnsupportedRuntimeException) {\r
+                               throw (UnsupportedRuntimeException) e;\r
+                       } else {\r
+                               writer.println(FAILED);\r
+                               writer.println(e.getLocalizedMessage());\r
+                               e.printStackTrace(writer);\r
+                       }\r
                }\r
                if (al != null) {\r
                        writer.println(OK);\r
@@ -383,29 +387,6 @@ public class WSTester {
        }\r
 \r
        /**\r
-        * Connect to a WS using the host and the service name\r
-        * \r
-        * @param <T>\r
-        * @param host\r
-        * @param service\r
-        * @return\r
-        */\r
-       <T> JABAService connect(String host, Services service) {\r
-               JABAService jabaservice = null;\r
-               try {\r
-                       writer.print("Connecting to service " + service + " on " + host\r
-                                       + " ... ");\r
-                       jabaservice = Jws2Client.connect(host, service);\r
-                       writer.println(OK);\r
-               } catch (WebServiceException e) {\r
-                       writer.println(FAILED);\r
-                       writer.println(e.getLocalizedMessage());\r
-                       e.printStackTrace(writer);\r
-               }\r
-               return jabaservice;\r
-       }\r
-\r
-       /**\r
         * Test JWS2 web services\r
         * \r
         * @param <T>\r
@@ -419,20 +400,20 @@ public class WSTester {
         * @throws IOException\r
         */\r
        public static <T> void main(String[] args) throws IOException {\r
-               WSTester tester = new WSTester(new PrintWriter(System.out, true));\r
+\r
                if (args == null || args.length < 1) {\r
-                       tester.printUsage();\r
+                       WSTester.printUsage();\r
                        System.exit(0);\r
                }\r
                String host = CmdHelper.getHost(args);\r
                String serviceName = CmdHelper.getServiceName(args);\r
                if (!Jws2Client.validURL(host)) {\r
-                       tester.writer\r
+                       System.err\r
                                        .println("<host_and_context> parameter is not provided or is incorrect!");\r
                        System.exit(1);\r
                }\r
-               boolean succeed = false;\r
-               MsaWS<T> msaws = null;\r
+               WSTester tester = new WSTester(host, new PrintWriter(System.out, true));\r
+\r
                if (serviceName != null) {\r
                        Services service = Services.getService(serviceName);\r
                        if (service == null) {\r
@@ -440,21 +421,10 @@ public class WSTester {
                                                + "' is not supported. Valid values are: "\r
                                                + Arrays.toString(Services.values()));\r
                                tester.writer.println();\r
-                               tester.printUsage();\r
-                               System.exit(1);\r
-                       }\r
-\r
-                       msaws = (MsaWS<T>) tester.connect(host, service);\r
-                       if (msaws == null) {\r
+                               printUsage();\r
                                System.exit(1);\r
                        }\r
-                       try {\r
-                               succeed = tester.checkService(msaws);\r
-                       } finally {\r
-                               ((Closeable) msaws).close();\r
-                       }\r
-                       // TODO test results printing!\r
-                       tester.reportResults(service, succeed);\r
+                       tester.checkService(service);\r
                        System.exit(0);\r
                }\r
 \r
@@ -463,26 +433,36 @@ public class WSTester {
 \r
                for (Services serv : Services.values()) {\r
                        tester.writer.println();\r
-                       msaws = (MsaWS<T>) tester.connect(host, serv);\r
-                       if (msaws == null) {\r
-                               continue;\r
-                       }\r
-                       try {\r
-                               succeed = tester.checkService(msaws);\r
-                       } finally {\r
-                               ((Closeable) msaws).close();\r
-                       }\r
-                       tester.reportResults(serv, succeed);\r
+                       tester.checkService(serv);\r
                }\r
 \r
        }\r
+\r
+       public boolean checkService(Services service) {\r
+               JABAService ws = Jws2Client.connect(hostname, service);\r
+               if (ws == null) {\r
+                       writer.println("Cannot estabilish the connection to host "\r
+                                       + hostname + " with service " + service.toString());\r
+                       return false;\r
+               }\r
+               boolean succeed = false;\r
+               try {\r
+                       writer.println("Checking service " + service.toString());\r
+                       succeed = checkService(ws);\r
+               } finally {\r
+                       FileUtil.closeSilently(((Closeable) ws));\r
+               }\r
+               reportResults(service, succeed);\r
+               return succeed;\r
+       }\r
+\r
        private void reportResults(Services serv, boolean succeed) {\r
                if (succeed) {\r
                        writer.println("Check is completed. The Service " + serv\r
-                                       + " IS WORKING");\r
+                                       + " IS WORKING\n");\r
                } else {\r
                        writer.println("Check is completed. The Service " + serv\r
-                                       + " HAS SOME PROBLEMS");\r
+                                       + " HAS SOME PROBLEMS\n");\r
                }\r
        }\r
 }\r