package org.codehaus.jsonpex;

import java.io.*;
import org.json.simple.JSONValue;

import com.sun.japex.*;

/**
 * @author Tatu Saloranta
 */
public class JsonSimpleDriver extends JapexDriverBase {
    
    int mHashCode;
    byte[] mInputData;
    
    public JsonSimpleDriver() { }

    @Override
    public void initializeDriver() {
        // All good as is
    }   
    
    @Override
    public void prepare(TestCase testCase) {
        String xmlFile = testCase.getParam("japex.inputFile");
        
        if (xmlFile == null) {
            throw new RuntimeException("japex.inputFile not specified");
        }
        try {
            // Load file to factor out I/O
            mInputData = Util.streamToByteArray(new FileInputStream(new File(xmlFile)));
        } catch (Exception e) { throw new RuntimeException(e); }
    }
    
    @Override
    public void warmup(TestCase testCase) {
        run(testCase);
    }
    
    @Override
    public void run(TestCase testCase) {
        try {
            // Bleh: they don't even have InputStream parse method...
            String input = new String(mInputData, "UTF-8");
            Object ob = JSONValue.parse(input);
            mHashCode = ob.hashCode(); // just to get some non-optimizable number
        }
        catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
    
    @Override
    public void finish(TestCase testCase) {
        // Set file size in KB on X axis
        testCase.setDoubleParam("japex.resultValueX", 
                mInputData.length / 1024.0);
        getTestSuite().setParam("japex.resultUnitX", "KB");
    }
    
}