KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > jruby > test > TestCommandlineParser


1 /***** BEGIN LICENSE BLOCK *****
2  * Version: CPL 1.0/GPL 2.0/LGPL 2.1
3  *
4  * The contents of this file are subject to the Common Public
5  * License Version 1.0 (the "License"); you may not use this file
6  * except in compliance with the License. You may obtain a copy of
7  * the License at http://www.eclipse.org/legal/cpl-v10.html
8  *
9  * Software distributed under the License is distributed on an "AS
10  * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
11  * implied. See the License for the specific language governing
12  * rights and limitations under the License.
13  *
14  * Copyright (C) 2002 Anders Bengtsson <ndrsbngtssn@yahoo.se>
15  * Copyright (C) 2005 Jason Voegele <jason@jvoegele.com>
16  * Copyright (C) 2005 Tim Azzopardi <tim@tigerfive.com>
17  * Copyright (C) 2006 Charles O Nutter <headius@headius.com>
18  *
19  * Alternatively, the contents of this file may be used under the terms of
20  * either of the GNU General Public License Version 2 or later (the "GPL"),
21  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
22  * in which case the provisions of the GPL or the LGPL are applicable instead
23  * of those above. If you wish to allow use of your version of this file only
24  * under the terms of either the GPL or the LGPL, and not to allow others to
25  * use your version of this file under the terms of the CPL, indicate your
26  * decision by deleting the provisions above and replace them with the notice
27  * and other provisions required by the GPL or the LGPL. If you do not delete
28  * the provisions above, a recipient may use your version of this file under
29  * the terms of any one of the CPL, the GPL or the LGPL.
30  ***** END LICENSE BLOCK *****/

31 package org.jruby.test;
32
33 import java.io.ByteArrayOutputStream JavaDoc;
34 import java.io.PrintStream JavaDoc;
35
36 import junit.framework.TestCase;
37
38 import org.jruby.Main;
39 import org.jruby.util.CommandlineParser;
40
41 public class TestCommandlineParser extends TestCase {
42     private PrintStream JavaDoc out;
43     private PrintStream JavaDoc err;
44
45     public void setUp() {
46         out = new PrintStream JavaDoc(new ByteArrayOutputStream JavaDoc());
47         err = new PrintStream JavaDoc(new ByteArrayOutputStream JavaDoc());
48     }
49     public void testParsing() {
50         CommandlineParser c = new CommandlineParser(new Main(System.in, out, err), new String JavaDoc[] { "-e", "hello", "-e", "world" });
51         assertEquals("hello\nworld\n", c.inlineScript());
52         assertNull(c.getScriptFileName());
53         assertEquals("-e", c.displayedFileName());
54
55         c = new CommandlineParser(new Main(System.in, out, err), new String JavaDoc[] { "--version" });
56         assertTrue(c.isShowVersion());
57
58         c = new CommandlineParser(new Main(System.in, out, err), new String JavaDoc[] { "-n", "myfile.rb" });
59         assertTrue(c.isAssumeLoop());
60         assertEquals("myfile.rb", c.getScriptFileName());
61         assertEquals("myfile.rb", c.displayedFileName());
62
63         c = new CommandlineParser(new Main(System.in, out, err), new String JavaDoc[0]);
64         assertEquals("-", c.displayedFileName());
65     }
66
67
68
69     public void testParsingWithDashDash() {
70       class TestableCommandlineParser extends CommandlineParser {
71
72         public TestableCommandlineParser(String JavaDoc[] arguments) {
73           super(new Main(System.in, out, err), arguments);
74         }
75       }
76       CommandlineParser c = new TestableCommandlineParser(new String JavaDoc[] { "-I", "someLoadPath", "--", "simple.rb", "-v", "--version" });
77       assertEquals("someLoadPath", c.loadPaths().get(0));
78       assertEquals("simple.rb",c.getScriptFileName());
79       assertEquals("simple.rb", c.displayedFileName());
80       assertTrue("Should not be verbose. The -v flag should be a parameter to the script, not the jruby interpreter", !c.isVerbose());
81       assertEquals("Script should have two parameters",2,c.getScriptArguments().length);
82       assertEquals("-v",c.getScriptArguments()[0]);
83       assertEquals("--version",c.getScriptArguments()[1]);
84     }
85
86     public void testPrintVersionDoesNotRunInterpreter() {
87         String JavaDoc[] args = new String JavaDoc[] { "-v" };
88         CommandlineParser parser = new CommandlineParser(new Main(System.in, out, err), args);
89         assertTrue(parser.isShowVersion());
90         assertFalse(parser.isShouldRunInterpreter());
91
92         args = new String JavaDoc[] { "--version" };
93         parser = new CommandlineParser(new Main(System.in, out, err), args);
94         assertTrue(parser.isShowVersion());
95         assertFalse(parser.isShouldRunInterpreter());
96     }
97
98     public void testHelpDoesNotRunIntepreter() {
99         String JavaDoc[] args = new String JavaDoc[] { "-h" };
100         CommandlineParser parser = new CommandlineParser(new Main(System.in, out, err), args);
101         assertFalse(parser.isShouldRunInterpreter());
102
103         args = new String JavaDoc[] { "--help" };
104         parser = new CommandlineParser(new Main(System.in, out, err), args);
105         assertFalse(parser.isShouldRunInterpreter());
106     }
107
108     public void testCommandTakesOneArgument() {
109         String JavaDoc[] args = new String JavaDoc[] { "--command", "gem" };
110         CommandlineParser parser = new CommandlineParser(new Main(System.in, out, err), args);
111         assertEquals(1, parser.requiredLibraries().size());
112         assertEquals("jruby/commands", parser.requiredLibraries().get(0));
113         assertEquals("JRuby::Commands.gem\n", parser.inlineScript());
114     }
115
116     public void testCommandAllowedOnlyOnceAndRemainderAreScriptArgs() {
117         String JavaDoc[] args = new String JavaDoc[] { "--command", "gem", "--command", "irb" };
118         CommandlineParser parser = new CommandlineParser(new Main(System.in, out, err), args);
119         assertEquals(2, parser.getScriptArguments().length);
120         assertEquals("--command", parser.getScriptArguments()[0]);
121         assertEquals("irb", parser.getScriptArguments()[1]);
122     }
123 }
124
Popular Tags