ExpectationSegment.h

Go to the documentation of this file.

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

   begin                : Thu Dec 18 2002
   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_EXPECTATIONSEGMENT_H
#define MOCKPP_EXPECTATIONSEGMENT_H


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

#include <mockpp/AbstractExpectation.h>
#include <mockpp/util/AssertMo.h>


MOCKPP_NS_START


template <class Str>
class ExpectationSegment : public AbstractExpectation<Str>
{
  public:

    ExpectationSegment(const String &name, VerifiableList *parent = 0)
      : AbstractExpectation<Str>(name, parent),
        expectNothing(false),
        haveActualValue(false)
    {
    }


    virtual void reset()
    {
      this->clearFailOnVerify();
      clearActual();
      clearExpectation();
    }


    void clearActual()
    {
      haveActualValue = false;
      myActualString = Str();
    }


    void setActual(const Str &aString)
    {
      haveActualValue = true;
      myActualString = aString;
      if (this->shouldCheckImmediately())
      {
        verify();
      }
    }


    void setExpected(const Str &segment)
    {
      myExpectedSegment = segment;
      expectNothing = false;
      this->setHasExpectations();
    }


    void setExpectNothing()
    {
      expectNothing = true;
      this->setHasExpectations();
    }


    virtual void verify()
    {
      if( expectNothing )
      {
        String fmt = mockpp_i18n(MOCKPP_PCHAR("%1 expected no value."));
        fmt << this->getVerifiableName();
        MOCKPP_ASSERT_FALSE_MESSAGE(fmt, haveActualValue );

      }
      else
      {
        if (!this->hasExpectations() )
          return;

        String fmt = mockpp_i18n(MOCKPP_PCHAR("%1 expected a value."));
        fmt << this->getVerifiableName();
        MOCKPP_ASSERT_TRUE_MESSAGE(fmt, haveActualValue );

        MOCKPP_ASSERT_INCLUDES_MESSAGE(
            mockpp_i18n(MOCKPP_PCHAR("Should include string segment.")),
            myExpectedSegment,
            myActualString);
      }
    }

  protected:

    virtual void clearExpectation()
    {
      this->clearHasExpectations();
    }


  private:

    Str   myExpectedSegment;
    Str   myActualString;
    bool  expectNothing;
    bool  haveActualValue;
};


MOCKPP_NS_END


#endif // MOCKPP_EXPECTATIONSEGMENT_H

---