dnk

eranpeer · Apr 21, 2024 · bcaef5e · bcaef5e
1 parent 2fbb239
commit bcaef5e
Show file tree

Hide file tree

Showing 2 changed files with 241 additions and 100 deletions.
diff --git a/include/fakeit/StubbingProgress.hpp b/include/fakeit/StubbingProgress.hpp
@@ -88,15 +88,23 @@ namespace fakeit {
 
             template<typename T>
             MethodStubbingProgress<R, arglist...>& ReturnCapture(T&& r) {
-                auto store = std::make_shared<typename std::remove_reference<R>::type>(std::forward<T>(r));
+                static_assert(std::is_constructible<
+                                typename std::remove_cv<typename std::remove_reference<R>::type>::type&,
+                                typename std::remove_cv<typename std::remove_reference<T>::type>::type&>::value,
+                        "The type captured by ReturnCapture() (named T) must be compatible with the return type of the function (named R), i.e. T t{...}; R& r = t; must compile without creating temporaries.");
+                auto store = std::make_shared<typename std::remove_cv<typename std::remove_reference<T>::type>::type>(std::forward<T>(r));
                 return Do([store](const typename fakeit::test_arg<arglist>::type...) mutable -> R {
                     return std::forward<R>(*store);
                 });
             }
 
             template<typename T>
             void AlwaysReturnCapture(T&& r) {
-                auto store = std::make_shared<typename std::remove_reference<R>::type>(std::forward<T>(r));
+                static_assert(std::is_constructible<
+                                typename std::remove_cv<typename std::remove_reference<R>::type>::type&,
+                                typename std::remove_cv<typename std::remove_reference<T>::type>::type&>::value,
+                        "The type captured by AlwaysReturnCapture() (named T) must be compatible with the return type of the function (named R), i.e. T t{...}; R& r = t; must compile without creating temporaries.");
+                auto store = std::make_shared<typename std::remove_cv<typename std::remove_reference<T>::type>::type>(std::forward<T>(r));
                 return AlwaysDo([store](const typename fakeit::test_arg<arglist>::type...) mutable -> R {
                     return std::forward<R>(*store);
                 });