com.rest4j.generator.PythonGeneratorTest.java Source code

Introduction

Here is the source code for com.rest4j.generator.PythonGeneratorTest.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.rest4j.generator;

import org.apache.commons.exec.CommandLine;
import org.apache.commons.exec.DefaultExecutor;
import org.apache.commons.exec.ExecuteWatchdog;
import org.apache.commons.io.IOUtils;
import org.junit.Test;

import java.io.File;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

/**
 * @author Joseph Kapizza <joseph@rest4j.com>
 */
public class PythonGeneratorTest {
    Generator gen = new Generator();

    @Test
    public void testPythonClient() throws Exception {
        gen.setStylesheet("com/rest4j/client/python.xslt");
        new File("target/python").mkdir();
        gen.setApiXmlUrl(getClass().getResource("doc-generator-graph.xml"));
        gen.setOutputDir("target/python");
        gen.addParam(new TemplateParam("common-params", "access-token"));
        gen.addParam(new TemplateParam("additional-client-code", "\t# ADDITIONAL CODE"));
        gen.generate();

        // let's try compiling the damn thing
        CommandLine cmdLine = CommandLine.parse("python apiclient.py");
        DefaultExecutor executor = new DefaultExecutor();
        executor.setWorkingDirectory(new File("target/python"));
        ExecuteWatchdog watchdog = new ExecuteWatchdog(60000);
        executor.setWatchdog(watchdog);
        int exitValue = executor.execute(cmdLine);
        assertFalse(executor.isFailure(exitValue));

        String a = IOUtils.toString(new File("target/python/apiclient.py").toURI());

        // check doc comments
        assertTrue(a, a.contains("Some additional client info"));
        assertTrue(a, a.contains("Some additional python client info"));

        // check existence of Parameter Object class
        assertTrue(a, a.contains("class PatchBRequest"));

        // check additional-client-code
        assertTrue(a, a.contains("\t# ADDITIONAL CODE"));
    }
}