com.wavemaker.tools.ws.WSDLTest.java Source code

Introduction

Here is the source code for com.wavemaker.tools.ws.WSDLTest.java

Source

/*
 *  Copyright (C) 2007-2009 WaveMaker Software, Inc.
 *
 *  This file is part of the WaveMaker Server Runtime.
 *
 *  Licensed under the Apache 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.apache.org/licenses/LICENSE-2.0
 *
 *  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 com.wavemaker.tools.ws;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.util.ResourceUtils;

import com.wavemaker.common.util.ClassLoaderUtils;
import com.wavemaker.infra.WMTestCase;
import com.wavemaker.tools.io.local.LocalFolder;
import com.wavemaker.tools.ws.wsdl.WSDL;
import com.wavemaker.tools.ws.wsdl.WSDLManager;

/**
 * @author Frankie Fu
 */
public class WSDLTest extends WMTestCase {

    private static final Log log = LogFactory.getLog(WSDLTest.class);

    public static WSDL buildWSDL(String wsdlResource) throws Exception {
        String resource = ClassLoaderUtils.getResource(wsdlResource);
        WSDL wsdl = WSDLManager.processWSDL(resource, null, null, null);
        return wsdl;
    }

    public static JAXBTypeMapper buildJAXBTypeMapper(WSDL wsdl, List<String> bindingResources) throws Exception {
        List<File> bindingFiles = new ArrayList<File>();
        if (bindingResources != null) {
            for (String bindingResource : bindingResources) {
                if (log.isDebugEnabled()) {
                    log.debug("Loading resource " + bindingResource + " for JAXB type mapper.");
                }
                File bindingFile = ResourceUtils.getFile("classpath:" + bindingResource);
                bindingFiles.add(bindingFile);
            }
        }

        JAXBTypeMapper mapper = new JAXBTypeMapper(wsdl, convertToFileSystemFileList(bindingFiles));
        return mapper;
    }

    private static List<com.wavemaker.tools.io.File> convertToFileSystemFileList(List<File> files) {
        List<com.wavemaker.tools.io.File> rtn = new ArrayList<com.wavemaker.tools.io.File>();
        for (File file : files) {
            rtn.add(new LocalFolder(file.getParent()).getFile(file.getName()));
        }
        return rtn;
    }

    public static WSDL getWSDL(String wsdlResource, List<String> bindingResources) throws Exception {
        WSDL wsdl = buildWSDL(wsdlResource);
        JAXBTypeMapper mapper = buildJAXBTypeMapper(wsdl, bindingResources);
        wsdl.setTypeMapper(mapper);
        return wsdl;
    }

    public static WSDL getWSDL(String wsdlResource) throws Exception {
        return getWSDL(wsdlResource, null);
    }

    public void testWSDLBuilder() throws Exception {
        WSDL wsdl = buildWSDL("com/wavemaker/tools/ws/stockquote.wsdl");
        assertNotNull(wsdl);
    }

    public void testJAXBTypeMapperBuilder() throws Exception {
        WSDL wsdl = buildWSDL("com/wavemaker/tools/ws/stockquote.wsdl");
        JAXBTypeMapper mapper = buildJAXBTypeMapper(wsdl, null);
        assertNotNull(mapper);
    }

    public void testRPCEncodedWSDL() throws Exception {
        try {
            buildWSDL("com/wavemaker/tools/ws/GoogleSearch.wsdl");
            fail("should fail an exception for RPC/encoded style.");
        } catch (Exception e) {
            // expected exception
        }
    }

}