testpackagechildren.java

来自「cqME :java framework for TCK test.」· Java 代码 · 共 108 行

/*
 * $Id$
 *
 * Copyright 2007 Sun Microsystems, Inc. All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License version
 * 2 only, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License version 2 for more details (a copy is
 * included at /legal/license.txt).
 *
 * You should have received a copy of the GNU General Public License
 * version 2 along with this work; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
 * Clara, CA 95054 or visit www.sun.com if you need additional
 * information or have any questions.
 *
 */

package com.sun.testme.jtviewer;

import com.sun.javatest.TestDescription;
import com.sun.testme.JTUtils;
import java.io.File;
import java.util.Collections;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import org.openide.nodes.Children;
import org.openide.nodes.Node;

/**
 *
 * @author ag155795
 */
public class TestPackageChildren extends Children.Keys{
    
    private File root;
    private ChangeListener listener;
    private boolean isSuiteRoot;
    
    private static String [] testFilters = {"tests"};
    
    /** Creates a new instance of TestPackageChildren */
    public TestPackageChildren(File root) {
        this.root = root;
        isSuiteRoot = JTUtils.getTestFinder().getRootDir().equals(root);
}

     protected Node[] createNodes(Object key) {
        if (key instanceof TestDescription) {
            TestDescription td = (TestDescription) key;
            return new Node[] { new TestNode( td ) };
        }
        return new Node[] { new TestPackageNode( (File) key ) };
    }

    protected void addNotify() {
        super.addNotify();
        refreshTree();
        listener = new ChangeListener() {
            public void stateChanged(ChangeEvent ev) {
                refreshTree();
            }
        };
        TestNotifier.addChangeListener(listener);
    }
    
    protected void removeNotify() {
        if (listener != null) {
            TestNotifier.removeChangeListener(listener);
            listener = null;
        }
        setKeys(Collections.EMPTY_SET);
    }
    
    private void refreshTree() {
        TestDescription [] tds = JTUtils.getTests(root, true);
        File [] files = null;
        /*
        if (isSuiteRoot) {
            files = JTUtils.getFiles(root, false, testFilters);
        } else {
            files = JTUtils.getFiles(root, false);
        }*/
        files = JTUtils.getFiles(root, false);
        int filesLength = (files == null) ? 0 : files.length;
        int tdsLength = (tds == null) ? 0 : tds.length; 
        
        Object [] objs = new Object[tdsLength + filesLength];
        if (tds != null) {
            System.arraycopy(tds, 0, objs, 0, tds.length);
        }
        if (files != null) {
            System.arraycopy(files, 0, objs, tds.length, files.length);
        }
        setKeys(objs);
    }
    
}