turtle/include/turtle/constraints.hpp

// http://turtle.sourceforge.net
//
// Copyright Mathieu Champlon 2008
//
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)

#ifndef MOCK_CONSTRAINTS_HPP_INCLUDED
#define MOCK_CONSTRAINTS_HPP_INCLUDED

#include "config.hpp"
#include "constraint.hpp"
#include "detail/addressof.hpp"
#include <boost/ref.hpp>
#include <boost/utility/enable_if.hpp>
#include <boost/type_traits/is_convertible.hpp>
#include <boost/test/floating_point_comparison.hpp>

namespace mock
{
    MOCK_UNARY_CONSTRAINT( any, 0,, true && &actual )
    MOCK_UNARY_CONSTRAINT( affirm, 0,, !! actual )
    MOCK_UNARY_CONSTRAINT( negate, 0,, ! actual )
    MOCK_UNARY_CONSTRAINT( evaluate, 0,, actual() )

    MOCK_NARY_CONSTRAINT( equal, 1, ( expected ), actual == expected )
    MOCK_NARY_CONSTRAINT( less, 1, ( expected ), actual < expected )
    MOCK_NARY_CONSTRAINT( greater, 1, ( expected ), actual > expected )
    MOCK_NARY_CONSTRAINT( less_equal, 1, ( expected ), actual <= expected )
    MOCK_NARY_CONSTRAINT( greater_equal, 1, ( expected ), actual >= expected )

#ifdef BOOST_MSVC
#   pragma push_macro( "small" )
#   undef small
#endif
    MOCK_NARY_CONSTRAINT( small, 1, ( expected ),
        ( boost::test_tools::check_is_small( actual, expected ) ) )
#ifdef BOOST_MSVC
#   pragma pop_macro( "small" )
#endif

    MOCK_NARY_CONSTRAINT( close, 2, ( expected, tolerance ),
        ( boost::test_tools::check_is_close(
            actual, expected,
            boost::test_tools::percent_tolerance( tolerance ) ) ) )
    MOCK_NARY_CONSTRAINT( close_fraction, 2, ( expected, tolerance ),
        ( boost::test_tools::check_is_close(
            actual, expected,
            boost::test_tools::fraction_tolerance( tolerance ) ) ) )

#ifdef BOOST_MSVC
#   pragma push_macro( "near" )
#   undef near
#endif
    MOCK_NARY_CONSTRAINT( near, 2, ( expected, tolerance ),
        std::abs( actual - expected ) < tolerance )
#ifdef BOOST_MSVC
#   pragma pop_macro( "near" )
#endif

namespace detail
{
    template< typename Expected >
    struct same
    {
        explicit same( const Expected& expected )
            : expected_( detail::addressof( boost::unwrap_ref( expected ) ) )
        {}
        template< typename Actual >
        bool operator()( const Actual& actual ) const
        {
            return detail::addressof( actual ) == expected_;
        }
        friend std::ostream& operator<<( std::ostream& os, const same& s )
        {
            return os << "same( " << mock::format( *s.expected_ ) << " )";
        }
        const typename
            boost::unwrap_reference< Expected >::type* expected_;
    };

    template< typename Expected >
    struct retrieve
    {
        explicit retrieve( Expected& expected )
            : expected_( detail::addressof( boost::unwrap_ref( expected ) ) )
        {}
        template< typename Actual >
        bool operator()( const Actual& actual,
            typename boost::disable_if<
                boost::is_convertible<
                    const Actual*,
                    typename
                        boost::unwrap_reference< Expected >::type
                >
            >::type* = 0 ) const
        {
            *expected_ = actual;
            return true;
        }
        template< typename Actual >
        bool operator()( Actual& actual,
            typename boost::enable_if<
                boost::is_convertible< Actual*,
                    typename
                        boost::unwrap_reference< Expected >::type
                >
            >::type* = 0 ) const
        {
            *expected_ = detail::addressof( actual );
            return true;
        }
        friend std::ostream& operator<<( std::ostream& s, const retrieve& r )
        {
            return s << "retrieve( " << mock::format( *r.expected_ ) << " )";
        }
        typename
            boost::unwrap_reference< Expected >::type* expected_;
    };

    template< typename Expected >
    struct assign
    {
        explicit assign( const Expected& expected )
            : expected_( expected )
        {}
        template< typename Actual >
        bool operator()( Actual& actual ) const
        {
            actual = boost::unwrap_ref( expected_ );
            return true;
        }
        template< typename Actual >
        bool operator()( Actual* actual,
            typename boost::enable_if<
                boost::is_convertible<
                    typename
                        boost::unwrap_reference< Expected >::type,
                    Actual
                >
            >::type* = 0 ) const
        {
            *actual = boost::unwrap_ref( expected_ );
            return true;
        }
        friend std::ostream& operator<<( std::ostream& s, const assign& a )
        {
            return s << "assign( " << mock::format( a.expected_ ) << " )";
        }
        Expected expected_;
    };

    template< typename Expected >
    struct contain
    {
        explicit contain( const Expected& expected )
            : expected_( expected )
        {}
        bool operator()( const std::string& actual ) const
        {
            return actual.find( boost::unwrap_ref( expected_ ) )
                != std::string::npos;
        }
        friend std::ostream& operator<<( std::ostream& s, const contain& n )
        {
            return s << "contain( " << mock::format( n.expected_ ) << " )";
        }
        Expected expected_;
    };
}

    template< typename T >
    constraint< detail::same< T > > same( T& t )
    {
        return detail::same< T >( t );
    }
    template< typename T >
    constraint< detail::retrieve< T > > retrieve( T& t )
    {
        return detail::retrieve< T >( t );
    }
    template< typename T >
    constraint< detail::assign< T > > assign( T t )
    {
        return detail::assign< T >( t );
    }
    template< typename T >
    constraint< detail::contain< T > > contain( T t )
    {
        return detail::contain< T >( t );
    }

    template< typename T >
    constraint< T > call( T t )
    {
        return constraint< T >( t );
    }
} // mock

#endif // MOCK_CONSTRAINTS_HPP_INCLUDED