//  Boost string_algo library find_format_test.cpp file  ------------------//

//  Copyright (c) 2009 Steven Watanabe
//  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)

//  See http://www.boost.org for updates, documentation, and revision history.

#include <boost/algorithm/string/find_format.hpp>
#include <boost/algorithm/string/finder.hpp>
#include <boost/algorithm/string/formatter.hpp>

// Include unit test framework
#include <boost/test/included/test_exec_monitor.hpp>

#include <boost/test/test_tools.hpp>

void find_format_test()
{
    const std::string source = "$replace $replace";
    std::string expected = "ok $replace";
    std::string output(80, '\0');

    std::string::iterator pos =
        boost::find_format_copy(output.begin(),
                                source,
                                boost::first_finder("$replace"),
                                boost::const_formatter("ok"));
    BOOST_CHECK(pos == output.begin() + expected.size());
    output.erase(std::remove(output.begin(), output.end(), '\0'), output.end());
    BOOST_CHECK_EQUAL(output, expected);

    output = boost::find_format_copy(source, boost::first_finder("$replace"), boost::const_formatter("ok"));
    BOOST_CHECK_EQUAL(output, expected);

    // now try finding a string that doesn't exist
    output.resize(80);
    pos = boost::find_format_copy(output.begin(),
                                source,
                                boost::first_finder("$noreplace"),
                                boost::const_formatter("bad"));
    BOOST_CHECK(pos == output.begin() + source.size());
    output.erase(std::remove(output.begin(), output.end(), '\0'), output.end());
    BOOST_CHECK_EQUAL(output, source);

    output = boost::find_format_copy(source, boost::first_finder("$noreplace"), boost::const_formatter("bad"));
    BOOST_CHECK_EQUAL(output, source);

    // in place version
    output = source;
    boost::find_format(output, boost::first_finder("$replace"), boost::const_formatter("ok"));
    BOOST_CHECK_EQUAL(output, expected);
    output = source;
    boost::find_format(output, boost::first_finder("$noreplace"), boost::const_formatter("bad"));
    BOOST_CHECK_EQUAL(output, source);
}

void find_format_all_test()
{
    const std::string source = "$replace $replace";
    std::string expected = "ok ok";
    std::string output(80, '\0');

    std::string::iterator pos =
        boost::find_format_all_copy(output.begin(),
                                source,
                                boost::first_finder("$replace"),
                                boost::const_formatter("ok"));
    BOOST_CHECK(pos == output.begin() + expected.size());
    output.erase(std::remove(output.begin(), output.end(), '\0'), output.end());
    BOOST_CHECK_EQUAL(output, expected);

    output = boost::find_format_all_copy(source, boost::first_finder("$replace"), boost::const_formatter("ok"));
    BOOST_CHECK_EQUAL(output, expected);

    // now try finding a string that doesn't exist
    output.resize(80);
    pos = boost::find_format_all_copy(output.begin(),
                                source,
                                boost::first_finder("$noreplace"),
                                boost::const_formatter("bad"));
    BOOST_CHECK(pos == output.begin() + source.size());
    output.erase(std::remove(output.begin(), output.end(), '\0'), output.end());
    BOOST_CHECK_EQUAL(output, source);

    output = boost::find_format_all_copy(source, boost::first_finder("$noreplace"), boost::const_formatter("bad"));
    BOOST_CHECK_EQUAL(output, source);

    // in place version
    output = source;
    boost::find_format_all(output, boost::first_finder("$replace"), boost::const_formatter("ok"));
    BOOST_CHECK_EQUAL(output, expected);
    output = source;
    boost::find_format_all(output, boost::first_finder("$noreplace"), boost::const_formatter("bad"));
    BOOST_CHECK_EQUAL(output, source);
}

int test_main( int, char*[] )
{
    find_format_test();
    find_format_all_test();

    return 0;
}