ExpectationBuilder.hpp

Go to the documentation of this file.

//          Copyright Dominic (DNKpp) Koepke 2024 - 2025.
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE_1_0.txt or copy at
//          https://www.boost.org/LICENSE_1_0.txt)
 
#ifndef MIMICPP_EXPECTATION_BUILDER_HPP
#define MIMICPP_EXPECTATION_BUILDER_HPP
 
#pragma once
 
#include "mimic++/Expectation.hpp"
#include "mimic++/Sequence.hpp"
#include "mimic++/matchers/Common.hpp"
#include "mimic++/matchers/GeneralMatchers.hpp"
#include "mimic++/matchers/StringMatchers.hpp"
#include "mimic++/policies/ArgRequirementPolicies.hpp"
#include "mimic++/policies/ControlPolicies.hpp"
#include "mimic++/policies/GeneralPolicies.hpp"
 
namespace mimicpp
{
    template <
        bool timesConfigured,
        typename SequenceConfig,
        typename Signature,
        typename FinalizePolicy,
        expectation_policy_for<Signature>... Policies>
    class BasicExpectationBuilder
    {
    public:
        using StorageT = ExpectationCollection<Signature>;
        using PolicyListT = std::tuple<Policies...>;
        using ReturnT = typename Expectation<Signature>::ReturnT;
 
        ~BasicExpectationBuilder() = default;
 
        template <typename FinalizePolicyArg, typename PolicyListArg>
            requires std::constructible_from<FinalizePolicy, FinalizePolicyArg>
                      && std::constructible_from<PolicyListT, PolicyListArg>
        [[nodiscard]]
        explicit constexpr BasicExpectationBuilder(
            std::shared_ptr<StorageT> storage,
            detail::TimesConfig timesConfig,
            SequenceConfig sequenceConfig,
            FinalizePolicyArg&& finalizePolicyArg,
            PolicyListArg&& policyListArg) noexcept
            : m_Storage{std::move(storage)},
              m_TimesConfig{std::move(timesConfig)},
              m_SequenceConfig{std::move(sequenceConfig)},
              m_FinalizePolicy{std::forward<FinalizePolicyArg>(finalizePolicyArg)},
              m_ExpectationPolicies{std::forward<PolicyListArg>(policyListArg)}
        {
            assert(m_Storage && "Storage is nullptr.");
        }
 
        BasicExpectationBuilder(const BasicExpectationBuilder&) = delete;
        BasicExpectationBuilder& operator=(const BasicExpectationBuilder&) = delete;
 
        [[nodiscard]]
        BasicExpectationBuilder(BasicExpectationBuilder&&) = default;
        BasicExpectationBuilder& operator=(BasicExpectationBuilder&&) = default;
 
        template <typename Policy>
            requires std::same_as<expectation_policies::InitFinalize, FinalizePolicy>
                  && (!std::same_as<expectation_policies::InitFinalize, std::remove_cvref_t<Policy>>)
                  && finalize_policy_for<std::remove_cvref_t<Policy>, Signature>
        [[nodiscard]]
        friend constexpr auto operator&&(BasicExpectationBuilder&& builder, Policy&& policy)
        {
            using ExtendedExpectationBuilderT = BasicExpectationBuilder<
                timesConfigured,
                SequenceConfig,
                Signature,
                std::remove_cvref_t<Policy>,
                Policies...>;
 
            return ExtendedExpectationBuilderT{
                std::move(builder.m_Storage),
                std::move(builder.m_TimesConfig),
                std::move(builder.m_SequenceConfig),
                std::forward<Policy>(policy),
                std::move(builder.m_ExpectationPolicies)};
        }
 
        template <typename Policy>
            requires expectation_policy_for<std::remove_cvref_t<Policy>, Signature>
        [[nodiscard]]
        friend constexpr auto operator&&(BasicExpectationBuilder&& builder, Policy&& policy)
        {
            using ExtendedExpectationBuilderT = BasicExpectationBuilder<
                timesConfigured,
                SequenceConfig,
                Signature,
                FinalizePolicy,
                Policies...,
                std::remove_cvref_t<Policy>>;
 
            return ExtendedExpectationBuilderT{
                std::move(builder.m_Storage),
                std::move(builder.m_TimesConfig),
                std::move(builder.m_SequenceConfig),
                std::move(builder.m_FinalizePolicy),
                std::tuple_cat(
                    std::move(builder.m_ExpectationPolicies),
                    std::forward_as_tuple(std::forward<Policy>(policy)))};
        }
 
        [[nodiscard]]
        friend constexpr auto operator&&(BasicExpectationBuilder&& builder, detail::TimesConfig&& config)
            requires(!timesConfigured)
        {
            using NewBuilderT = BasicExpectationBuilder<
                true,
                SequenceConfig,
                Signature,
                FinalizePolicy,
                Policies...>;
 
            return NewBuilderT{
                std::move(builder.m_Storage),
                std::move(config),
                std::move(builder.m_SequenceConfig),
                std::move(builder.m_FinalizePolicy),
                std::move(builder.m_ExpectationPolicies)};
        }
 
        template <typename... Sequences>
        [[nodiscard]]
        friend constexpr auto operator&&(BasicExpectationBuilder&& builder, sequence::detail::Config<Sequences...>&& config)
        {
            sequence::detail::Config newConfig = builder.m_SequenceConfig.concat(std::move(config));
 
            using ExtendedExpectationBuilderT = BasicExpectationBuilder<
                timesConfigured,
                decltype(newConfig),
                Signature,
                FinalizePolicy,
                Policies...>;
 
            return ExtendedExpectationBuilderT{
                std::move(builder.m_Storage),
                std::move(builder.m_TimesConfig),
                std::move(newConfig),
                std::move(builder.m_FinalizePolicy),
                std::move(builder.m_ExpectationPolicies)};
        }
 
        [[nodiscard]]
        ScopedExpectation finalize(const std::source_location& sourceLocation) &&
        {
            static_assert(
                finalize_policy_for<FinalizePolicy, Signature>,
                "For non-void return types, a finalize policy must be set.");
 
            return ScopedExpectation{
                std::move(m_Storage),
                std::apply(
                    [&](auto&... policies) {
                        ControlPolicy controlPolicy{
                            std::move(m_TimesConfig),
                            std::move(m_SequenceConfig)};
 
                        using ExpectationT = BasicExpectation<
                            Signature,
                            decltype(controlPolicy),
                            FinalizePolicy,
                            Policies...>;
 
                        return std::make_unique<ExpectationT>(
                            sourceLocation,
                            std::move(controlPolicy),
                            std::move(m_FinalizePolicy),
                            std::move(policies)...);
                    },
                    m_ExpectationPolicies)};
        }
 
    private:
        std::shared_ptr<StorageT> m_Storage;
        detail::TimesConfig m_TimesConfig{};
        SequenceConfig m_SequenceConfig{};
        FinalizePolicy m_FinalizePolicy{};
        PolicyListT m_ExpectationPolicies{};
    };
}
 
namespace mimicpp::detail
{
    template <typename Param, typename Arg>
        requires matcher_for<
            std::remove_cvref_t<Arg>,
            Param>
    [[nodiscard]]
    constexpr auto make_arg_matcher(Arg&& arg, [[maybe_unused]] const priority_tag<2>)
    {
        return std::forward<Arg>(arg);
    }
 
    template <string Param, string Arg>
    [[nodiscard]]
    constexpr auto make_arg_matcher(Arg&& arg, [[maybe_unused]] const priority_tag<1>)
    {
        return matches::str::eq(std::forward<Arg>(arg));
    }
 
    template <typename Param, std::equality_comparable_with<Param> Arg>
    [[nodiscard]]
    constexpr auto make_arg_matcher(Arg&& arg, [[maybe_unused]] const priority_tag<0>)
    {
        return matches::eq(std::forward<Arg>(arg));
    }
 
    constexpr priority_tag<2> max_make_arg_matcher_tag{};
 
    template <typename Arg, typename Target>
    concept requirement_for = requires {
        {
            detail::make_arg_matcher<Target, Arg>(
                std::declval<Arg>(),
                max_make_arg_matcher_tag)
        } -> matcher_for<Target>;
    };
 
    template <
        typename Signature,
        std::size_t index,
        typename Arg,
        typename... Canary,
        typename Param = signature_param_type_t<index, Signature>>
        requires requirement_for<Arg, Param>
    constexpr auto make_arg_policy(Arg&& arg)
    {
        return expect::arg<index>(
            detail::make_arg_matcher<Param, Arg>(
                std::forward<Arg>(arg),
                max_make_arg_matcher_tag));
    }
 
    template <typename Signature, typename Builder, std::size_t... indices, typename... Args>
    [[nodiscard]]
    constexpr auto extend_builder_with_arg_policies(
        Builder&& builder,
        const std::index_sequence<indices...>,
        Args&&... args)
    {
        return (
            std::forward<Builder>(builder)
            && ...
            && detail::make_arg_policy<Signature, indices>(
                std::forward<Args>(args)));
    }
 
    template <typename Signature, typename... Args>
    constexpr auto make_expectation_builder(
        std::shared_ptr<ExpectationCollection<Signature>> expectations,
        Args&&... args)
    {
        using BaseBuilderT = BasicExpectationBuilder<
            false,
            sequence::detail::Config<>,
            Signature,
            expectation_policies::InitFinalize>;
 
        return detail::extend_builder_with_arg_policies<Signature>(
            BaseBuilderT{
                std::move(expectations),
                TimesConfig{},
                sequence::detail::Config<>{},
                expectation_policies::InitFinalize{},
                std::tuple{}},
            std::index_sequence_for<Args...>{},
            std::forward<Args>(args)...);
    }
}
 
#define MIMICPP_DETAIL_UNIQUE_NAME(prefix, counter) prefix##counter
#define MIMICPP_DETAIL_SCOPED_EXPECTATION_IMPL(counter) \
    [[maybe_unused]]                                    \
    const ::mimicpp::ScopedExpectation MIMICPP_DETAIL_UNIQUE_NAME(_mimicpp_expectation_, counter) =

#define MIMICPP_SCOPED_EXPECTATION MIMICPP_DETAIL_SCOPED_EXPECTATION_IMPL(__COUNTER__)
 
#ifndef MIMICPP_CONFIG_ONLY_PREFIXED_MACROS

    #define SCOPED_EXP MIMICPP_SCOPED_EXPECTATION
 
#endif
 
#endif