VisitableMockMethod2.h

Go to the documentation of this file.

/**************************************************************************

   begin                : Thu Oct 22 2005
   copyright            : (C) 2002-2010 by Ewald Arnold
   email                : mockpp at ewald-arnold dot de

   This program 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 2 of the License,
   or (at your option) any later version.

   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 Lesser General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

 **/

#ifndef MOCKPP_VisitableMockMethod2_H
#define MOCKPP_VisitableMockMethod2_H

#include <mockpp/mockpp.h>

#include <mockpp/visiting/VisitableMockObject.h>
#include <mockpp/visiting/VisitableMockMethod.h>
#include <mockpp/visiting/ResponseVector2.h>


MOCKPP_NS_START


template <typename R, typename P1, typename P2>
class VisitableMockMethod2Common : public VisitableMockReturningMethodBase<R>
{
  public:

    VisitableMockMethod2Common(const String &name, VisitableMockObject *parent)
      : VisitableMockReturningMethodBase<R>(name, parent)
      , responseThrowables(this->getMethodName() + MOCKPP_PCHAR("/responseThrowables") , this)
      , parameter1(this->getMethodName() + MOCKPP_PCHAR("/parameter1"), this)
      , parameter2(this->getMethodName() + MOCKPP_PCHAR("/parameter2"), this)
    {
    }


    void forward (const ConstraintHolder<P1> &p1, const ConstraintHolder<P2> &p2) const
    {
      MOCKPP_ASSERT_FALSE_MESSAGE(this->getVisitableMockObject()->getVerifiableName() + MOCKPP_PCHAR(".isActivated() != true"),
                                  this->getVisitableMockObject()->isActivated());
      this->getVisitableMockObject()->addExpectedMethod(this->getMethodIdentifier());
      parameter1.addExpected(p1);
      parameter2.addExpected(p2);
    }

  protected:

    void forward_param(const P1 &p1, const P2 &p2) const
    {
      if (!this->getVisitableMockObject()->isActivated() )
      {
        this->getVisitableMockObject()->addExpectedMethod(this->getMethodIdentifier());
        parameter1.addExpected(p1);
        parameter2.addExpected(p2);
      }

      else
      {
        MOCKPP_TRY
        {
          this->getVisitableMockObject()->addActualMethod(this->getMethodIdentifier());

          Throwable *t;
          if (this->responseThrowables.find(t, p1, p2))
             t->throw_me();

          this->throwAvailableException();
        }

        MOCKPP_CATCH_ALL
        {
          parameter1.balanceActual();
          parameter2.balanceActual();
          MOCKPP_RETHROW;
        }

        parameter1.addActual(p1);
        parameter2.addActual(p2);
      }
    }

  public:

    void addResponseThrowable(Throwable *t, const P1 &p1, const P2 &p2, unsigned count = MOCKPP_UNLIMITED)
    {
      MOCKPP_ASSERT_FALSE(this->getVisitableMockObject()->isActivated());
      this->responseThrowables.add(t, p1, p2, count);
    }

    void addResponseThrowable(Throwable *t, const ConstraintHolder<P1> &p1, const ConstraintHolder<P2> &p2, unsigned count = MOCKPP_UNLIMITED)
    {
      MOCKPP_ASSERT_FALSE(this->getVisitableMockObject()->isActivated());
      this->responseThrowables.add(t, p1, p2, count);
    }

  private:

    mutable ResponseThrowableVector2<P1, P2>   responseThrowables;

    mutable ConstraintList<P1>             parameter1;
    mutable ConstraintList<P2>             parameter2;
};


template <typename R, typename P1, typename P2>
class VisitableMockMethod2
  : public VisitableMockMethod2Common<R, P1, P2>
{
  public:

    VisitableMockMethod2(const String &name, VisitableMockObject *parent)
      : VisitableMockMethod2Common<R, P1, P2>(name, parent)
      , responseValues(this->getMethodName() + MOCKPP_PCHAR("/responseValues") , this)
    {
    }

    R forward(const P1 &p1, const P2 &p2) const
    {
      this->forward_param(p1, p2);

      if (this->getVisitableMockObject()->isActivated() )
      {
        R ret_val;
        if (this->responseValues.find(ret_val, p1, p2))
          return ret_val;

        return this->determineReturnValue();
      }
      else
        return R(); // only dummy value to make the compiler happy
    }

#if defined(__BORLANDC__) // ==> BCB5.5.1 ?? F1004 Internal compiler error at 0x12548c1 with base 0x1200000

    void forward (const ConstraintHolder<P1> &p1, const ConstraintHolder<P2> &p2) const
    {
       VisitableMockMethod2Common<R, P1, P2>::forward(p1, p2);
    }
#else
    using VisitableMockMethod2Common<R, P1, P2>::forward;
#endif

    void addResponseValue(const R &rv, const P1 &p1, const P2 &p2, unsigned count = MOCKPP_UNLIMITED)
    {
      MOCKPP_ASSERT_FALSE(this->getVisitableMockObject()->isActivated());
      this->responseValues.add(rv, p1, p2, count);
    }

    void addResponseValue(const R &rv, const ConstraintHolder<P1> &p1, const ConstraintHolder<P2> &p2, unsigned count = MOCKPP_UNLIMITED)
    {
      MOCKPP_ASSERT_FALSE(this->getVisitableMockObject()->isActivated());
      this->responseValues.add(rv, p1, p2, count);
    }

  private:

    mutable ResponseVector2<R, P1, P2>   responseValues;
};


template <typename P1, typename P2>
class VisitableMockMethod2<void, P1, P2>
   : public VisitableMockMethod2Common<void, P1, P2>
{
  public:

    VisitableMockMethod2(const String &name, VisitableMockObject *parent)
      : VisitableMockMethod2Common<void, P1, P2>(name, parent)
    {
    }

    void forward(const P1 &p1, const P2 &p2) const
    {
       this->forward_param(p1, p2);
    }

#if defined(__BORLANDC__) // ==> BCB5.5.1 ?? F1004 Internal compiler error at 0x12548c1 with base 0x1200000

    void forward (const ConstraintHolder<P1> &p1, const ConstraintHolder<P2> &p2) const
    {
       VisitableMockMethod2Common<void, P1, P2>::forward(p1, p2);
    }
#else
    using VisitableMockMethod2Common<void, P1, P2>::forward;
#endif
};


MOCKPP_NS_END


#endif // MOCKPP_VisitableMockMethod2_H

---