001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * Contributor(s):
025: *
026: * The Original Software is NetBeans. The Initial Developer of the Original
027: * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun
028: * Microsystems, Inc. All Rights Reserved.
029: *
030: * If you wish your version of this file to be governed by only the CDDL
031: * or only the GPL Version 2, indicate your decision by adding
032: * "[Contributor] elects to include this software in this distribution
033: * under the [CDDL or GPL Version 2] license." If you do not indicate a
034: * single choice of license, a recipient has the option to distribute
035: * your version of this file under either the CDDL, the GPL Version 2 or
036: * to extend the choice of license to its licensees as provided above.
037: * However, if you add GPL Version 2 code and therefore, elected the GPL
038: * Version 2 license, then the option applies only if the new code is
039: * made subject to such option by the copyright holder.
040: */
041:
042: package org.netbeans;
043:
044: import java.io.*;
045: import junit.textui.TestRunner;
046: import org.netbeans.junit.*;
047: import java.util.*;
048: import java.util.jar.JarOutputStream;
049: import junit.framework.AssertionFailedError;
050: import org.openide.util.RequestProcessor;
051:
052: /**
053: * Test the command-line-interface handler.
054: * @author Jaroslav Tulach
055: */
056: public class MainCLITest extends NbTestCase {
057: public MainCLITest(String name) {
058: super (name);
059: }
060:
061: public void testHandlersCanBeInUserDir() throws Exception {
062: clearWorkDir();
063:
064: class H extends CLIHandler {
065: public H() {
066: super (WHEN_INIT);
067: }
068:
069: protected int cli(Args args) {
070: String[] arr = args.getArguments();
071: for (int i = 0; i < arr.length; i++) {
072: if (arr[i] == "--userdir") {
073: System.setProperty("netbeans.user", arr[i + 1]);
074: return 0;
075: }
076: }
077: fail("One of the arguments should be --userdir: "
078: + Arrays.asList(arr));
079: return 0;
080: }
081:
082: protected void usage(PrintWriter w) {
083: }
084: }
085:
086: File dir = super .getWorkDir();
087: File lib = new File(dir, "core");
088: lib.mkdirs();
089: File jar = new File(lib, "sample.jar");
090: JarOutputStream os = new JarOutputStream(new FileOutputStream(
091: jar));
092: os.putNextEntry(new java.util.zip.ZipEntry(
093: "META-INF/services/org.netbeans.CLIHandler"));
094: os.write(TestHandler.class.getName().getBytes());
095: String res = "/"
096: + TestHandler.class.getName().replace('.', '/')
097: + ".class";
098: os.putNextEntry(new java.util.zip.ZipEntry(res));
099: org.openide.filesystems.FileUtil.copy(getClass()
100: .getResourceAsStream(res), os);
101: os.close();
102:
103: TestHandler.called = false;
104:
105: String[] args = new String[] { "--userdir", dir.toString() };
106: assertFalse(
107: "User dir is not correct. Will be set by org.netbeans.core.CLIOptions",
108: dir.toString().equals(
109: System.getProperty("netbeans.user")));
110: int result = MainImpl.execute(args, null, null, null, null);
111: Main.finishInitialization();
112: assertEquals("User set", dir.toString(), System
113: .getProperty("netbeans.user"));
114: assertTrue("CLI Handler from user dir was called",
115: TestHandler.called);
116: }
117:
118: /** Sample handler
119: */
120: public static final class TestHandler extends CLIHandler {
121: public static boolean called;
122:
123: public TestHandler() {
124: super (CLIHandler.WHEN_INIT);
125: }
126:
127: protected int cli(org.netbeans.CLIHandler.Args args) {
128: called = true;
129: return 0;
130: }
131:
132: protected void usage(PrintWriter w) {
133: }
134:
135: }
136: }
|