1  // Copyright (c) 2022 The Bitcoin Core developers
 2  // Distributed under the MIT software license, see the accompanying
 3  // file COPYING or http://www.opensource.org/licenses/mit-license.php.
 4  
 5  #include <util/result.h>
 6  
 7  #include <boost/test/unit_test.hpp>
 8  
 9  inline bool operator==(const bilingual_str& a, const bilingual_str& b)
10  {
11      return a.original == b.original && a.translated == b.translated;
12  }
13  
14  inline std::ostream& operator<<(std::ostream& os, const bilingual_str& s)
15  {
16      return os << "bilingual_str('" << s.original << "' , '" << s.translated << "')";
17  }
18  
19  BOOST_AUTO_TEST_SUITE(result_tests)
20  
21  struct NoCopy {
22      NoCopy(int n) : m_n{std::make_unique<int>(n)} {}
23      std::unique_ptr<int> m_n;
24  };
25  
26  bool operator==(const NoCopy& a, const NoCopy& b)
27  {
28      return *a.m_n == *b.m_n;
29  }
30  
31  std::ostream& operator<<(std::ostream& os, const NoCopy& o)
32  {
33      return os << "NoCopy(" << *o.m_n << ")";
34  }
35  
36  util::Result<int> IntFn(int i, bool success)
37  {
38      if (success) return i;
39      return util::Error{Untranslated(strprintf("int %i error.", i))};
40  }
41  
42  util::Result<bilingual_str> StrFn(bilingual_str s, bool success)
43  {
44      if (success) return s;
45      return util::Error{strprintf(Untranslated("str %s error."), s.original)};
46  }
47  
48  util::Result<NoCopy> NoCopyFn(int i, bool success)
49  {
50      if (success) return {i};
51      return util::Error{Untranslated(strprintf("nocopy %i error.", i))};
52  }
53  
54  template <typename T>
55  void ExpectResult(const util::Result<T>& result, bool success, const bilingual_str& str)
56  {
57      BOOST_CHECK_EQUAL(bool(result), success);
58      BOOST_CHECK_EQUAL(util::ErrorString(result), str);
59  }
60  
61  template <typename T, typename... Args>
62  void ExpectSuccess(const util::Result<T>& result, const bilingual_str& str, Args&&... args)
63  {
64      ExpectResult(result, true, str);
65      BOOST_CHECK_EQUAL(result.has_value(), true);
66      BOOST_CHECK_EQUAL(result.value(), T{std::forward<Args>(args)...});
67      BOOST_CHECK_EQUAL(&result.value(), &*result);
68  }
69  
70  template <typename T, typename... Args>
71  void ExpectFail(const util::Result<T>& result, const bilingual_str& str)
72  {
73      ExpectResult(result, false, str);
74  }
75  
76  BOOST_AUTO_TEST_CASE(check_returned)
77  {
78      ExpectSuccess(IntFn(5, true), {}, 5);
79      ExpectFail(IntFn(5, false), Untranslated("int 5 error."));
80      ExpectSuccess(NoCopyFn(5, true), {}, 5);
81      ExpectFail(NoCopyFn(5, false), Untranslated("nocopy 5 error."));
82      ExpectSuccess(StrFn(Untranslated("S"), true), {}, Untranslated("S"));
83      ExpectFail(StrFn(Untranslated("S"), false), Untranslated("str S error."));
84  }
85  
86  BOOST_AUTO_TEST_CASE(check_value_or)
87  {
88      BOOST_CHECK_EQUAL(IntFn(10, true).value_or(20), 10);
89      BOOST_CHECK_EQUAL(IntFn(10, false).value_or(20), 20);
90      BOOST_CHECK_EQUAL(NoCopyFn(10, true).value_or(20), 10);
91      BOOST_CHECK_EQUAL(NoCopyFn(10, false).value_or(20), 20);
92      BOOST_CHECK_EQUAL(StrFn(Untranslated("A"), true).value_or(Untranslated("B")), Untranslated("A"));
93      BOOST_CHECK_EQUAL(StrFn(Untranslated("A"), false).value_or(Untranslated("B")), Untranslated("B"));
94  }
95  
96  BOOST_AUTO_TEST_SUITE_END()