org.evosuite.testcase.fm.SpecifiedValuesAnswer Maven / Gradle / Ivy

Go to download
/**
 * Copyright (C) 2010-2018 Gordon Fraser, Andrea Arcuri and EvoSuite
 * contributors
 *
 * This file is part of EvoSuite.
 *
 * EvoSuite is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3.0 of the License, or
 * (at your option) any later version.
 *
 * EvoSuite is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with EvoSuite. If not, see .
 */
package org.evosuite.testcase.fm;

import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

/**
 * Specified a list of values, which will be returned in the order they are
 * specified each time an answer is queried.
 * If there are more queries than values, the last value will be returned
 *
 * Created by Andrea Arcuri on 27/07/15.
 */
@Deprecated // there is easier way to do it in Mockito, ie "Stubbing consecutive calls (iterator-style stubbing)"
public class SpecifiedValuesAnswer implements Answer{

    private final T[] values;
    private int counter;

    public SpecifiedValuesAnswer(T... values){
        this.values = values;
        counter = 0;
    }

    @Override
    public T answer(InvocationOnMock invocationOnMock) throws Throwable {

        if(values==null || values.length==0){
            /*
                return default value.
                we use 0 instead of null to avoid possible problems with
                unboxing of primitive values (eg int)
             */
            if(values instanceof Integer[]){
                return (T) new Integer(0);
            } else if(values instanceof Double[]){
                return (T) new Double(0);
            } else if(values instanceof Float[]){
                return (T) new Float(0);
            } else if(values instanceof Long[]){
                return (T) new Long(0);
            } else if(values instanceof Short[]){
                return (T) new Short((short)0);
            } else {
                return null;
            }
        }

        if(counter >= values.length){
            //return the last element
            return values[values.length-1];
        }

        return values[counter++];
    }
}