dhruv9vats commented on a change in pull request #12368:
URL: https://github.com/apache/arrow/pull/12368#discussion_r808206494
##########
File path: cpp/src/arrow/testing/matchers.h
##########
@@ -61,6 +61,67 @@ class PointeesEqualMatcher {
// Useful in conjunction with other googletest matchers.
inline PointeesEqualMatcher PointeesEqual() { return {}; }
+class AnyOfJSONMatcher : public
::testing::internal::MatcherBaseImpl<AnyOfJSONMatcher> {
+ public:
+ using AnyOfJSONMatcher::MatcherBaseImpl::MatcherBaseImpl;
+ AnyOfJSONMatcher(std::shared_ptr<DataType> type, std::string array_json)
+ : type_(std::move(type)), array_json_(std::move(array_json)) {}
+
+ template <typename arg_type>
+ operator testing::Matcher<arg_type>() const { // NOLINT runtime/explicit
+ class Impl : public ::testing::MatcherInterface<const arg_type&> {
+ public:
+ explicit Impl(std::shared_ptr<DataType> type, std::string array_json)
+ : type_(std::move(type)), array_json_(std::move(array_json)) {
+ array = ArrayFromJSON(type_, array_json_);
+ }
+ void DescribeTo(std::ostream* os) const override {
+ *os << "matches at least one scalar from ";
+ *os << array->ToString();
+ }
+ void DescribeNegationTo(::std::ostream* os) const override {
+ *os << "matches no scalar from ";
+ *os << array->ToString();
+ }
+ bool MatchAndExplain(
+ const arg_type& arg,
+ ::testing::MatchResultListener* result_listener) const override {
+ for (int64_t i = 0; i < array->length(); ++i) {
+ std::shared_ptr<Scalar> scalar;
+ auto maybe_scalar = array->GetScalar(i);
+ if (maybe_scalar.ok()) {
+ scalar = maybe_scalar.ValueOrDie();
+ } else {
+ *result_listener << "GetScalar() had status "
+ << maybe_scalar.status().ToString() << "at index
" << i
+ << " in the input JSON Array";
+ return false;
+ }
+
+ if (scalar->Equals(arg)) return true;
+ }
+ *result_listener << "Argument scalar: '" << arg->ToString()
+ << "' matches no scalar from " << array->ToString();
+ return false;
+ }
+ const std::shared_ptr<DataType> type_;
+ const std::string array_json_;
+ std::shared_ptr<Array> array;
+ };
+
+ return testing::Matcher<arg_type>(new Impl(type_, array_json_));
+ }
+
+ private:
+ const std::shared_ptr<DataType> type_;
+ const std::string array_json_;
+};
+
+inline AnyOfJSONMatcher AnyOfJSON(std::shared_ptr<DataType> type,
Review comment:
Had to learn the hard way that `inline` is important. 😅
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
