All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.itest.impl.ITestContextImpl Maven / Gradle / Ivy

Go to download

The newest version!
/**
 * 
 * The MIT License (MIT)
 * 
 * Copyright (c) 2014 Grzegorz Kochański
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 * 

 */
package org.itest.impl;

import org.itest.ITestConstants;
import org.itest.ITestContext;
import org.itest.exception.ITestException;
import org.itest.param.ITestParamState;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.StringTokenizer;

public class ITestContextImpl implements ITestContext {
    private final List path = new ArrayList();

    private final List owners = new ArrayList();

    private final List params = new ArrayList();

    private final Map, List> assignments = new HashMap, List>();

    private final Map, Map> staticITestAssignmentMap;

    private final ITestParamState rootParam;

    private final ITestValueHolder rootValueHolder;

    private final List valueHolders = new ArrayList();

    public ITestContextImpl(ITestParamState rootParam, Map, Map> staticITestAssignmentMap) {
        this.rootParam = rootParam;
        this.params.add(rootParam);
        this.rootValueHolder = new ITestValueHolder(rootParam);
        this.valueHolders.add(rootValueHolder);
        this.staticITestAssignmentMap = staticITestAssignmentMap;
    }

    @Override
    public void registerAssignment(String sourcePath) {
        // assignments.put(new ArrayList(path), normalizePath(path, Arrays.asList(StringUtils.split(sourcePath, '.'))));
    }

    @Override
    public void registerAssignment(Class clazz, String name) {
        registerAssignment(getStaticAssignment(clazz, name));
    }

    @Override
    public boolean isStaticAssignmentRegistered(Class clazz, String field) {
        return null != getStaticAssignment(clazz, field);
    }

    private String getStaticAssignment(Class clazz, String field) {
        Map fieldMap = staticITestAssignmentMap.get(clazz);
        return null == fieldMap ? null : fieldMap.get(field);
    }

    @Override
    public void enter(Object owner, String field) {
        getCurrentValueHolder().setValue(owner);
        params.add(null == getCurrentParam() ? null : getCurrentParam().getElement(field));
        ITestValueHolder vh = new ITestValueHolder(getCurrentParam());
        getCurrentValueHolder().addEelement(field, vh);
        valueHolders.add(vh);
        path.add(field);
        owners.add(owner);
    }

    public void setEmptyParam() {
        params.set(params.size() - 1, ITestRandomObjectGeneratorImpl.EMPTY_STATE);
    }

    @Override
    public void leave(Object value) {
        if ( getCurrentValueHolder().valueSet ) {
            assert getCurrentValueHolder().value == value;
        }
        getCurrentValueHolder().setValue(value);
        valueHolders.remove(valueHolders.size() - 1);
        params.remove(params.size() - 1);
        path.remove(path.size() - 1);
        owners.remove(owners.size() - 1);
    }

    @Override
    public int depth() {
        return path.size();
    }

    private ITestValueHolder getCurrentValueHolder() {
        return valueHolders.get(valueHolders.size() - 1);
    }

    static class ITestAssignment {
        List source;

        List target;
    }

    @Override
    public Map, List> getAssignments() {
        return assignments;
    }

    @Override
    public Object getCurrentOwner() {
        return owners.get(owners.size() - 1);
    }

    @Override
    public String getCurrentField() {
        return path.get(path.size() - 1);
    }

    public ITestParamState getCurrentParam() {
        return params.get(params.size() - 1);
    }

    private ITestValueHolder findGeneratedValueHolder(String targetPath) {
        ITestValueHolder res;
        int depth;
        if ( targetPath.startsWith(ITestConstants.SEPARATOR) ) {
            depth = 0;
        } else {
            depth = valueHolders.size() - 1;
        }
        res = valueHolders.get(depth);
        StringTokenizer st = new StringTokenizer(targetPath, ITestConstants.SEPARATOR);

        while (st.hasMoreTokens() && null != res) {
            String token = st.nextToken();
            if ( ITestConstants.PARENT.equals(token) ) {
                depth--;
                res = valueHolders.get(depth);
            } else if ( ITestConstants.NULL.equals(token) ) {
                return null;
            } else {
                res = res.getElement(token);
            }
        }
        if ( null == res ) {
            throw new ITestException("@ref:" + targetPath + " not found");
        }
        return res;
    }

    public ITestParamState findGeneratedState(String targetPath) {
        return findGeneratedValueHolder(targetPath).getParam();
    }

    public Object findGeneratedObject(String targetPath) {
        return findGeneratedValueHolder(targetPath).getValue();
    }
    @Override
    public void replaceCurrentState(ITestParamState iTestState) {
        params.set(params.size() - 1, iTestState);
    }

    static class ITestValueHolder {
        ITestParamState param;
        Object value;

        Map elements;

        boolean valueSet;

        public ITestValueHolder(ITestParamState currentParam) {
            this.param = currentParam;
        }

        public void addEelement(String field, ITestValueHolder vh) {
            if ( null == elements ) {
                elements = new HashMap();
            }
            elements.put(field, vh);
        }

        public ITestValueHolder getElement(String token) {
            return elements.get(token);
        }

        public void setValue(Object value) {
            this.value = value;
            this.valueSet = true;
        }

        public Object getValue() {
            if ( !valueSet ) {
                throw new ITestException("Value not set");
            }
            return value;
        }

        public ITestParamState getParam() {
            return param;
        }
    }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy