mockinputstreamfactory.java

来自「Java的面向对象数据库系统的源代码」· Java 代码 · 共 58 行

// You can redistribute this software and/or modify it under the terms of
// the Ozone Library License version 1 published by ozone-db.org.
//
// The original code and portions created by Joakim Ohlrogge are
// Copyright (C) 1997-@year@ by Joakim Ohlrogge. All rights reserved.
//
// $Id$
package test.ozoneDB.tools.OPP.srcgen.streamfactory;

import com.mockobjects.Verifiable;
import com.mockobjects.ExpectationMap;
import com.mockobjects.util.Verifier;
import org.ozoneDB.tools.OPP.srcgen.streamfactory.InputStreamFactory;

import java.io.InputStream;
import java.io.IOException;
import java.io.ByteArrayInputStream;

/**
 * User: Jocke
 * Date: 2004-jan-03
 * Time: 14:28:16
 */
public class MockInputStreamFactory implements InputStreamFactory, Verifiable {
    ExpectationMap instanceReturnValues = new ExpectationMap("expected data");

    /**
     * Use this method to setup each expected return value for the given class name.
     * The values are returned by key in the order they are added for each key. If you want a value to
     * be returned twice, set it up twice.
     * @param className The name of the class that the data is for
     * @param data The data to return
     */
    public void setupNewInstanceReturnValue(String className, byte data[]) {
        instanceReturnValues.addExpected(className, data);
    }

    /**
     * Use this method to setup each expected return value for the given class name.
     * The values are returned by key in the order they are added for each key. If you want a value to
     * be returned twice, set it up twice.
     * @param className The name of the class that the data is for
     * @param data The data to return
     */
    public void setupNewInstanceReturnValue(String className, String data) {
        setupNewInstanceReturnValue(className, data.getBytes());
    }

    public InputStream newInstance(String className) throws IOException {
        InputStream is = new ByteArrayInputStream((byte[]) instanceReturnValues.get(className));
        return is;
    }

    public void verify() {
        Verifier.verifyObject(this);
    }
}