ExecStreamUnitTestBase.h

Go to the documentation of this file.

/*
// $Id: //open/dev/fennel/test/ExecStreamUnitTestBase.h#21 $
// Fennel is a library of data storage and processing components.
// Copyright (C) 2005-2009 The Eigenbase Project
// Copyright (C) 2005-2009 SQLstream, Inc.
// Copyright (C) 2005-2009 LucidEra, Inc.
// Portions Copyright (C) 2004-2009 John V. Sichi
//
// This program is free software; you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by the Free
// Software Foundation; either version 2 of the License, or (at your option)
// any later version approved by The Eigenbase Project.
//
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#ifndef Fennel_ExecStreamUnitTestBase_Included
#define Fennel_ExecStreamUnitTestBase_Included

#include "fennel/test/ExecStreamTestBase.h"
#include "fennel/exec/MockProducerExecStream.h"
#include "fennel/test/ExecStreamGenerator.h"

FENNEL_BEGIN_NAMESPACE

class FENNEL_TEST_EXPORT ExecStreamUnitTestBase
    : public ExecStreamTestBase
{
protected:
    SharedExecStreamGraph pGraph;
    SharedExecStreamGraphEmbryo pGraphEmbryo;


    SharedExecStream prepareSourceGraph(
        ExecStreamEmbryo &sourceStreamEmbryo);

    SharedExecStream prepareTransformGraph(
        ExecStreamEmbryo &sourceStreamEmbryo,
        ExecStreamEmbryo &transformStreamEmbryo);

    SharedExecStream prepareTransformGraph(
        ExecStreamEmbryo &sourceStreamEmbryo,
        std::vector<ExecStreamEmbryo> &transforms);

    SharedExecStream prepareConfluenceGraph(
        ExecStreamEmbryo &sourceStreamEmbryo1,
        ExecStreamEmbryo &sourceStreamEmbryo2,
        ExecStreamEmbryo &confluenceStreamEmbryo);

    SharedExecStream prepareConfluenceTransformGraph(
        ExecStreamEmbryo &sourceStreamEmbryo1,
        ExecStreamEmbryo &sourceStreamEmbryo2,
        ExecStreamEmbryo &confluenceStreamEmbryo,
        ExecStreamEmbryo &transformStreamEmbryo);

    SharedExecStream prepareConfluenceGraph(
        std::vector<ExecStreamEmbryo> &sourceStreamEmbryos,
        ExecStreamEmbryo &confluenceStreamEmbryo);

    SharedExecStream prepareConfluenceGraph(
        std::vector<std::vector<ExecStreamEmbryo> > &sourceStreamEmbryosList,
        ExecStreamEmbryo &confluenceStreamEmbryo);

    SharedExecStream prepareDAG(
        ExecStreamEmbryo &srcStreamEmbryo,
        ExecStreamEmbryo &splitterStreamEmbryo,
        std::vector<ExecStreamEmbryo> &interStreamEmbryos,
        ExecStreamEmbryo &destStreamEmbryo,
        bool createSink = true,
        bool saveSrc = true);

    SharedExecStream prepareDAG(
        ExecStreamEmbryo &srcStreamEmbryo,
        ExecStreamEmbryo &splitterStreamEmbryo,
        std::vector<std::vector<ExecStreamEmbryo> > &interStreamEmbryosList,
        ExecStreamEmbryo &destStreamEmbryo,
        bool createSink = true,
        bool saveSrc = true);

    void verifyOutput(
        ExecStream &stream,
        uint nRowsExpected,
        MockProducerExecStreamGenerator &verifier,
        bool stopEarly = false);

    void verifyConstantOutput(
        ExecStream &stream,
        const TupleData  &expectedTuple,
        uint nRowsExpected);

    void verifyBufferedOutput(
        ExecStream &stream,
        TupleDescriptor outputTupleDesc,
        uint nRowsExpected,
        PBuffer expectedBuffer);

    void resetExecStreamTest();

    // refine ExecStreamTestBase
    virtual void tearDownExecStreamTest();

public:
    // refine ExecStreamTestBase
    virtual void testCaseSetUp();
};

FENNEL_END_NAMESPACE
#endif

// End ExecStreamUnitTestBase.h

---