mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-26 01:51:41 +00:00
[wpiutil] Refactor SpanMatcher and TestPrinters from ntcore (#5658)
This commit is contained in:
75
wpiutil/src/test/native/include/wpi/SpanMatcher.h
Normal file
75
wpiutil/src/test/native/include/wpi/SpanMatcher.h
Normal file
@@ -0,0 +1,75 @@
|
||||
// Copyright (c) FIRST and other WPILib contributors.
|
||||
// Open Source Software; you can modify and/or share it under the terms of
|
||||
// the WPILib BSD license file in the root directory of this project.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <algorithm>
|
||||
#include <initializer_list>
|
||||
#include <memory>
|
||||
#include <ostream>
|
||||
#include <span>
|
||||
#include <type_traits>
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
||||
#include <gmock/gmock.h>
|
||||
|
||||
#include "wpi/TestPrinters.h"
|
||||
|
||||
namespace wpi {
|
||||
|
||||
template <typename T>
|
||||
class SpanMatcher : public ::testing::MatcherInterface<std::span<T>> {
|
||||
public:
|
||||
explicit SpanMatcher(std::span<T> good_) : good{good_.begin(), good_.end()} {}
|
||||
|
||||
bool MatchAndExplain(std::span<T> val,
|
||||
::testing::MatchResultListener* listener) const override;
|
||||
void DescribeTo(::std::ostream* os) const override;
|
||||
void DescribeNegationTo(::std::ostream* os) const override;
|
||||
|
||||
private:
|
||||
std::vector<std::remove_cv_t<T>> good;
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
inline ::testing::Matcher<std::span<const typename T::value_type>> SpanEq(
|
||||
const T& good) {
|
||||
return ::testing::MakeMatcher(
|
||||
new SpanMatcher(std::span<const typename T::value_type>(good)));
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
inline ::testing::Matcher<std::span<const T>> SpanEq(
|
||||
std::initializer_list<const T> good) {
|
||||
return ::testing::MakeMatcher(
|
||||
new SpanMatcher<const T>({good.begin(), good.end()}));
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
bool SpanMatcher<T>::MatchAndExplain(
|
||||
std::span<T> val, ::testing::MatchResultListener* listener) const {
|
||||
if (val.size() != good.size() ||
|
||||
!std::equal(val.begin(), val.end(), good.begin())) {
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
void SpanMatcher<T>::DescribeTo(::std::ostream* os) const {
|
||||
PrintTo(std::span<T>{good}, os);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
void SpanMatcher<T>::DescribeNegationTo(::std::ostream* os) const {
|
||||
*os << "is not equal to ";
|
||||
PrintTo(std::span<T>{good}, os);
|
||||
}
|
||||
|
||||
} // namespace wpi
|
||||
|
||||
inline std::span<const uint8_t> operator"" _us(const char* str, size_t len) {
|
||||
return {reinterpret_cast<const uint8_t*>(str), len};
|
||||
}
|
||||
41
wpiutil/src/test/native/include/wpi/TestPrinters.h
Normal file
41
wpiutil/src/test/native/include/wpi/TestPrinters.h
Normal file
@@ -0,0 +1,41 @@
|
||||
// Copyright (c) FIRST and other WPILib contributors.
|
||||
// Open Source Software; you can modify and/or share it under the terms of
|
||||
// the WPILib BSD license file in the root directory of this project.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <ostream>
|
||||
#include <span>
|
||||
#include <string>
|
||||
#include <string_view>
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include "wpi/json.h"
|
||||
|
||||
namespace wpi {
|
||||
|
||||
inline void PrintTo(std::string_view str, ::std::ostream* os) {
|
||||
::testing::internal::PrintStringTo(std::string{str}, os);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
void PrintTo(std::span<T> val, ::std::ostream* os) {
|
||||
*os << '{';
|
||||
bool first = true;
|
||||
for (auto v : val) {
|
||||
if (first) {
|
||||
first = false;
|
||||
} else {
|
||||
*os << ", ";
|
||||
}
|
||||
*os << ::testing::PrintToString(v);
|
||||
}
|
||||
*os << '}';
|
||||
}
|
||||
|
||||
inline void PrintTo(const json& val, ::std::ostream* os) {
|
||||
*os << val.dump();
|
||||
}
|
||||
|
||||
} // namespace wpi
|
||||
Reference in New Issue
Block a user