fitnesse.runner.FormattingOption.java Source code

Introduction

Here is the source code for fitnesse.runner.FormattingOption.java

Source

// Copyright (C) 2003-2009 by Object Mentor, Inc. All rights reserved.
// Released under the terms of the CPL Common Public License version 1.0.
package fitnesse.runner;

import fitnesse.http.RequestBuilder;
import fitnesse.http.ResponseParser;
import org.apache.commons.io.IOUtils;
import util.CommandLine;
import util.CommandLineParseException;
import util.FileUtil;
import util.ImpossibleException;

import java.io.*;

public class FormattingOption {
    public String format;
    public boolean usingStdout = false;
    public OutputStream output;
    public String host;
    public int port;
    public String rootPath;
    private int status;
    public String filename;
    private String resultFilename;

    public static void main(String[] args) throws Exception {
        FormattingOption option = new FormattingOption();
        option.args(args);
        File inputFile = new File(option.resultFilename);
        FileInputStream input = new FileInputStream(inputFile);
        int byteCount = (int) inputFile.length();
        option.process(input, byteCount);
    }

    private FormattingOption() {
    }

    private void args(String[] args) throws FileNotFoundException {
        try {
            CommandLine commandLine = new CommandLine("resultFilename format outputFilename host port rootPath",
                    args);
            resultFilename = commandLine.getArgument("resultFilename");
            format = commandLine.getArgument("format");
            filename = commandLine.getArgument("outputFilename");
            host = commandLine.getArgument("host");
            port = Integer.parseInt(commandLine.getArgument("port"));
            rootPath = commandLine.getArgument("rootPath");
            setOutput(System.out);
        } catch (CommandLineParseException e) {
            usage();
        }
    }

    private void usage() {
        System.out.println(
                "java fitnesse.runner.FormattingOption resultFilename format outputFilename host port rootPath");
        System.out.println("\tresultFilename:\tthe name of the file containing test results");
        System.out.println("\tformat:        \traw|html|xml|...");
        System.out.println("\toutputfilename:\tstdout|a filename where the formatted results are to be stored");
        System.out.println("\thost:          \tthe domain name of the hosting FitNesse server");
        System.out.println("\tport:          \tthe port on which the hosting FitNesse server is running");
        System.out.println("\trootPath:      \tname of the test page or suite page");
        System.exit(-1);
    }

    public FormattingOption(String format, String filename, OutputStream stdout, String host, int port,
            String rootPath) throws Exception {
        this.format = format;
        this.filename = filename;
        setOutput(stdout);
        this.host = host;
        this.port = port;
        this.rootPath = rootPath;
    }

    private void setOutput(OutputStream stdout) throws FileNotFoundException {
        if ("stdout".equals(filename)) {
            this.output = stdout;
            this.usingStdout = true;
        } else
            this.output = new FileOutputStream(filename);
    }

    public void process(InputStream inputStream, int size) throws IOException {
        try {
            if ("raw".equals(format))
                FileUtil.copyBytes(inputStream, output);
            else {
                RequestBuilder request = buildRequest(inputStream, size);
                ResponseParser response = ResponseParser.performHttpRequest(host, port, request);
                status = response.getStatus();
                try {
                    output.write(response.getBody().getBytes("UTF-8"));
                } catch (IOException e) {
                    throw new ImpossibleException("UTF-8 is a supported encoding", e);
                }
            }
        } finally {
            if (!usingStdout)
                IOUtils.closeQuietly(output);
        }
    }

    public boolean wasSuccessful() {
        return status == 200;
    }

    public RequestBuilder buildRequest(InputStream inputStream, int size) {
        RequestBuilder request = new RequestBuilder("/" + rootPath);
        request.setMethod("POST");
        request.setHostAndPort(host, port);
        request.addInput("responder", "format");
        request.addInput("format", format);
        request.addInputAsPart("results", inputStream, size, "text/plain");
        return request;
    }
}