KickJava   Java API By Example, From Geeks To Geeks.

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


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) 2001-2002 Benoit Cerrina <b.cerrina@wanadoo.fr>
15  * Copyright (C) 2001-2004 Jan Arne Petersen <jpetersen@uni-bonn.de>
16  * Copyright (C) 2002-2004 Anders Bengtsson <ndrsbngtssn@yahoo.se>
17  * Copyright (C) 2004 Thomas E Enebo <enebo@acm.org>
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 import java.io.StringReader JavaDoc;
36
37 import junit.framework.TestCase;
38
39 import org.jruby.Ruby;
40 import org.jruby.RubyIO;
41
42 /**
43  * @author Benoit
44  */
45 public class TestRubyBase extends TestCase {
46     protected Ruby runtime;
47     private PrintStream JavaDoc out;
48
49     public TestRubyBase() {
50     }
51     
52     public TestRubyBase(String JavaDoc name) {
53         super(name);
54     }
55
56     /**
57      * evaluate a string and returns the standard output.
58      * @param script the String to eval as a String
59      * @return the value printed out on stdout and stderr by
60      **/
61     protected String JavaDoc eval(String JavaDoc script) throws Exception JavaDoc {
62         ByteArrayOutputStream JavaDoc result = new ByteArrayOutputStream JavaDoc();
63         out = new PrintStream JavaDoc(result);
64         RubyIO lStream = new RubyIO(runtime, out);
65         runtime.getGlobalVariables().set("$stdout", lStream);
66         runtime.getGlobalVariables().set("$>", lStream);
67         runtime.getGlobalVariables().set("$stderr", lStream);
68         
69         runtime.loadScript("test", new StringReader JavaDoc(script), false);
70         StringBuffer JavaDoc sb = new StringBuffer JavaDoc(new String JavaDoc(result.toByteArray()));
71         for (int idx = sb.indexOf("\n"); idx != -1; idx = sb.indexOf("\n")) {
72             sb.deleteCharAt(idx);
73         }
74         
75         return sb.toString();
76     }
77
78     public void tearDown() {
79         if (out != null)
80             out.close();
81     }
82 }
83
Popular Tags