ReturnAndCallStub.h

Go to the documentation of this file.

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

   begin                : Mon Nov 10 2008
   copyright            : (C) 2008 by Mathieu Champlon
   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_ReturnAndCallStub_H
#define MOCKPP_ReturnAndCallStub_H

#include <mockpp/mockpp.h> // always first

#include <mockpp/compat/Formatter.h>

#include <mockpp/stub/TypelessStub.h>


MOCKPP_NS_START


template <  typename R     // Return type
          , typename F     // Functor
         >
class ReturnAndCallStub : public TypelessStub<R>
{
  public:

    ReturnAndCallStub(  const R &in_result
                      , F in_functor
                     )
      : result(in_result)
      , functor(in_functor)
    {}

    virtual R typelessInvoke()
    {
      functor();
      return result;
    }

    virtual String describeTo( String &buffer ) const
    {
      String fmt;
      fmt += MOCKPP_PCHAR("calls functor\n");
      fmt += MOCKPP_PCHAR("returns <%1>");
      fmt << result;
      buffer += fmt;

      return buffer;
    }

  private:

    R   result;
    F   functor;
};


MOCKPP_NS_END

#endif // MOCKPP_ReturnAndCallStub_H

---