org.kuali.test.ui.components.repositorytree.RepositoryNode.java Source code

Introduction

Here is the source code for org.kuali.test.ui.components.repositorytree.RepositoryNode.java

Source

/*
 * Copyright 2014 The Kuali Foundation
 * 
 * Licensed under the Educational Community License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.opensource.org/licenses/ecl2.php
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.kuali.test.ui.components.repositorytree;

import java.io.File;
import java.io.IOException;
import javax.swing.tree.DefaultMutableTreeNode;
import org.apache.commons.beanutils.BeanUtils;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.log4j.Logger;
import org.kuali.test.KualiTestConfigurationDocument;
import org.kuali.test.Platform;
import org.kuali.test.SuiteTest;
import org.kuali.test.TestSuite;
import org.kuali.test.utils.Constants;
import org.kuali.test.utils.Utils;

/**
 *
 * @author rbtucker
 */
public class RepositoryNode extends DefaultMutableTreeNode {
    protected static final Logger LOG = Logger.getLogger(RepositoryNode.class);
    private KualiTestConfigurationDocument.KualiTestConfiguration configuration;

    /**
     *
     * @param configuration
     * @param userObject
     */
    public RepositoryNode(KualiTestConfigurationDocument.KualiTestConfiguration configuration, Object userObject) {
        super(userObject);
        this.configuration = configuration;
        try {
            if (isRoot()) {
                if (configuration.getPlatforms() != null) {
                    for (Platform platform : configuration.getPlatforms().getPlatformArray()) {
                        addNode(getPlatformFilePath(platform), platform);
                    }
                }
            } else if (userObject instanceof Platform) {
                Platform platform = (Platform) userObject;
                if (platform.getTestSuites() != null) {
                    for (TestSuite testSuite : platform.getTestSuites().getTestSuiteArray()) {
                        addNode(null, testSuite);
                    }
                }
            } else if (userObject instanceof TestSuite) {
                TestSuite testSuite = (TestSuite) userObject;
                Platform platform = findPlatform(testSuite.getPlatformName());
                if (platform != null) {
                    if (testSuite.getSuiteTests() != null) {
                        for (SuiteTest suiteTest : testSuite.getSuiteTests().getSuiteTestArray()) {
                            File f = new File(Utils.getTestFilePath(configuration, suiteTest.getTestHeader()));
                            if (f.exists() && f.isFile()) {
                                add(new RepositoryNode(configuration, suiteTest));
                            }
                        }
                    }
                }
            }
        }

        catch (IOException ex) {
            LOG.error(ex.toString(), ex);
        }
    }

    private Platform findPlatform(String platformName) {
        Platform retval = null;
        for (Platform platform : configuration.getPlatforms().getPlatformArray()) {
            if (platform.getName().equalsIgnoreCase(platformName)) {
                retval = platform;
                break;
            }
        }
        return retval;
    }

    private void addNode(String filepath, Object userObject) throws IOException {
        if (StringUtils.isNotBlank(filepath)) {
            File f = new File(filepath);
            if (!f.exists()) {
                FileUtils.forceMkdir(f);
            }
        }

        add(new RepositoryNode(configuration, userObject));
    }

    private String getPlatformFilePath(Platform platform) {
        StringBuilder retval = new StringBuilder(256);
        retval.append(configuration.getRepositoryLocation());
        retval.append(Constants.FORWARD_SLASH);
        retval.append(platform.getName());
        return retval.toString();
    }

    private String getTestSuiteFilePath(TestSuite testSuite) {
        StringBuilder retval = new StringBuilder(256);
        retval.append(configuration.getRepositoryLocation());
        retval.append(Constants.FORWARD_SLASH);
        retval.append(testSuite.getPlatformName());
        retval.append(Constants.FORWARD_SLASH);
        retval.append(testSuite.getName());
        return retval.toString();
    }

    @Override
    public boolean isLeaf() {
        return (getUserObject() instanceof SuiteTest);
    }

    @Override
    public boolean isRoot() {
        return (getUserObject() == null);
    }

    @Override
    public String toString() {
        String retval = "unknown";

        if (isRoot()) {
            File f = new File(configuration.getRepositoryLocation());
            retval = f.getName();
        } else {
            Object o = getUserObject();

            if (o != null) {
                try {
                    if (o instanceof SuiteTest) {
                        SuiteTest test = (SuiteTest) o;
                        retval = test.getTestHeader().getTestName();
                    } else {
                        retval = BeanUtils.getProperty(o, "name");
                    }
                }

                catch (Exception ex) {
                    LOG.warn(ex);
                }
            }
        }

        return retval;
    }
}