framework/testtools/src/org/ofbiz/testtools/seleniumxml/DataLoader.java

/*
 * 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 org.ofbiz.testtools.seleniumxml;

import java.util.List;
import java.util.Map;

import javolution.util.FastMap;

import org.jdom.Element;
import org.python.core.PyArray;
import org.python.core.PyDictionary;
import org.python.core.PyList;
import org.python.core.PyObject;
import org.python.util.PythonInterpreter;
import org.ofbiz.testtools.seleniumxml.InitJython;
import org.ofbiz.testtools.seleniumxml.SeleniumXml;

public class DataLoader {

    private String file;
    private String iterations;
    private SeleniumXml parent;
    private SeleniumXml currentTest;
    private List<Element> children;

    private int currentRowIndx;


    //Objects initialized from csvreader script.
    private PyDictionary fieldNameMap;
    private PyList dataList;
    private PyList fieldNames;

    public DataLoader(String file, String iterations, SeleniumXml parent, List<Element> children) {
        super();
        this.file = file;
        this.iterations = iterations;
        this.parent = parent;
        this.children = children;
        initData();
    }

    private void initData() {
        // Run the python script
        // Read header and get record count
        PythonInterpreter interp = InitJython.getInterpreter();

        Map<String, Object> map = FastMap.newInstance();
        map.put("file", this.file);
        interp.set("params", map);

        interp.exec("from csvreader import CSVReader");
        String cmd = "reader = CSVReader('" + this.file + "')";
        interp.exec(cmd);
        this.dataList = (PyList) interp.eval("reader.dataList");
        this.fieldNames = (PyList) interp.eval("reader.fieldNames");
        this.fieldNameMap = (PyDictionary) interp.eval("reader.fieldNameMap");
        //interp.execfile("c:/dev/ag/seleniumxml/plugins/csvreader.py");
        //interp.execfile("c:/dev/ag/seleniumxml/plugins/TestCSVReader.py");

        //Now get output from script
        //this.dataList = (PyArray) map.get("dataList");
        //this.fieldNames = (PyDictionary) map.get("fieldNames");

    }

    private void next() {
        this.currentRowIndx = (this.currentRowIndx + 1) % this.dataList.__len__();
    }

    private void loadData() {

        int size = this.fieldNames.__len__();
        for(int i=0; i<size; i++ ) {
            PyObject name = this.fieldNames.__getitem__(i);
            PyObject valueList = this.dataList.__getitem__(this.currentRowIndx);
            PyObject columnIndx = this.fieldNameMap.__getitem__(name);
            Integer convIndx = (Integer) columnIndx.__tojava__(Integer.class);
            //int convIndx = Integer.parseInt((String) columnIndx.__tojava__(String.class));
            PyObject value = valueList.__getitem__(convIndx);
            this.currentTest.addParam((String) name.__tojava__(String.class), (String) value.__tojava__(String.class));
        }

    }

    public void runTest() throws TestCaseException {

        //Depending on the iteration instruction repeat the following until complete
        int iter = Integer.parseInt(this.iterations);

        //Iterate through entire list of data
        if(iter == -1) {
            iter = this.dataList.__len__();
        }

        this.currentTest = new SeleniumXml(this.parent);
        for( int i=0; i<iter; i++) {
            loadData();
            currentTest.runCommands(this.children);
            next();
        }

    }
}