265 lines
7.5 KiB
C++
265 lines
7.5 KiB
C++
|
// Formatting library for C++ - the core API
|
||
|
//
|
||
|
// Copyright (c) 2012 - present, Victor Zverovich
|
||
|
// All rights reserved.
|
||
|
//
|
||
|
// For the license information refer to format.h.
|
||
|
//
|
||
|
// Copyright (c) 2018 - present, Remotion (Igor Schulz)
|
||
|
// All Rights Reserved
|
||
|
// {fmt} support for ranges, containers and types tuple interface.
|
||
|
|
||
|
#include "fmt/ranges.h"
|
||
|
|
||
|
#include <map>
|
||
|
#include <string>
|
||
|
#include <vector>
|
||
|
|
||
|
#include "gtest/gtest.h"
|
||
|
|
||
|
#if !FMT_GCC_VERSION || FMT_GCC_VERSION >= 601
|
||
|
# define FMT_RANGES_TEST_ENABLE_C_STYLE_ARRAY
|
||
|
#endif
|
||
|
|
||
|
#if !FMT_MSC_VER || FMT_MSC_VER > 1910
|
||
|
# define FMT_RANGES_TEST_ENABLE_JOIN
|
||
|
# define FMT_RANGES_TEST_ENABLE_FORMAT_STRUCT
|
||
|
#endif
|
||
|
|
||
|
#ifdef FMT_RANGES_TEST_ENABLE_C_STYLE_ARRAY
|
||
|
TEST(ranges_test, format_array) {
|
||
|
int arr[] = {1, 2, 3, 5, 7, 11};
|
||
|
EXPECT_EQ(fmt::format("{}", arr), "[1, 2, 3, 5, 7, 11]");
|
||
|
}
|
||
|
|
||
|
TEST(ranges_test, format_2d_array) {
|
||
|
int arr[][2] = {{1, 2}, {3, 5}, {7, 11}};
|
||
|
EXPECT_EQ(fmt::format("{}", arr), "[[1, 2], [3, 5], [7, 11]]");
|
||
|
}
|
||
|
|
||
|
TEST(ranges_test, format_array_of_literals) {
|
||
|
const char* arr[] = {"1234", "abcd"};
|
||
|
EXPECT_EQ(fmt::format("{}", arr), "[\"1234\", \"abcd\"]");
|
||
|
}
|
||
|
#endif // FMT_RANGES_TEST_ENABLE_C_STYLE_ARRAY
|
||
|
|
||
|
TEST(ranges_test, format_vector) {
|
||
|
auto v = std::vector<int>{1, 2, 3, 5, 7, 11};
|
||
|
EXPECT_EQ(fmt::format("{}", v), "[1, 2, 3, 5, 7, 11]");
|
||
|
}
|
||
|
|
||
|
TEST(ranges_test, format_vector2) {
|
||
|
auto v = std::vector<std::vector<int>>{{1, 2}, {3, 5}, {7, 11}};
|
||
|
EXPECT_EQ(fmt::format("{}", v), "[[1, 2], [3, 5], [7, 11]]");
|
||
|
}
|
||
|
|
||
|
TEST(ranges_test, format_map) {
|
||
|
auto m = std::map<std::string, int>{{"one", 1}, {"two", 2}};
|
||
|
EXPECT_EQ(fmt::format("{}", m), "[(\"one\", 1), (\"two\", 2)]");
|
||
|
}
|
||
|
|
||
|
TEST(ranges_test, format_pair) {
|
||
|
auto p = std::pair<int, float>(42, 1.5f);
|
||
|
EXPECT_EQ(fmt::format("{}", p), "(42, 1.5)");
|
||
|
}
|
||
|
|
||
|
TEST(ranges_test, format_tuple) {
|
||
|
auto t =
|
||
|
std::tuple<int, float, std::string, char>(42, 1.5f, "this is tuple", 'i');
|
||
|
EXPECT_EQ(fmt::format("{}", t), "(42, 1.5, \"this is tuple\", 'i')");
|
||
|
EXPECT_EQ(fmt::format("{}", std::tuple<>()), "()");
|
||
|
}
|
||
|
|
||
|
#ifdef FMT_RANGES_TEST_ENABLE_FORMAT_STRUCT
|
||
|
struct tuple_like {
|
||
|
int i;
|
||
|
std::string str;
|
||
|
|
||
|
template <size_t N> fmt::enable_if_t<N == 0, int> get() const noexcept {
|
||
|
return i;
|
||
|
}
|
||
|
template <size_t N>
|
||
|
fmt::enable_if_t<N == 1, fmt::string_view> get() const noexcept {
|
||
|
return str;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
template <size_t N>
|
||
|
auto get(const tuple_like& t) noexcept -> decltype(t.get<N>()) {
|
||
|
return t.get<N>();
|
||
|
}
|
||
|
|
||
|
namespace std {
|
||
|
template <>
|
||
|
struct tuple_size<tuple_like> : std::integral_constant<size_t, 2> {};
|
||
|
|
||
|
template <size_t N> struct tuple_element<N, tuple_like> {
|
||
|
using type = decltype(std::declval<tuple_like>().get<N>());
|
||
|
};
|
||
|
} // namespace std
|
||
|
|
||
|
TEST(ranges_test, format_struct) {
|
||
|
auto t = tuple_like{42, "foo"};
|
||
|
EXPECT_EQ(fmt::format("{}", t), "(42, \"foo\")");
|
||
|
}
|
||
|
#endif // FMT_RANGES_TEST_ENABLE_FORMAT_STRUCT
|
||
|
|
||
|
TEST(ranges_test, format_to) {
|
||
|
char buf[10];
|
||
|
auto end = fmt::format_to(buf, "{}", std::vector<int>{1, 2, 3});
|
||
|
*end = '\0';
|
||
|
EXPECT_STREQ(buf, "[1, 2, 3]");
|
||
|
}
|
||
|
|
||
|
struct path_like {
|
||
|
const path_like* begin() const;
|
||
|
const path_like* end() const;
|
||
|
|
||
|
operator std::string() const;
|
||
|
};
|
||
|
|
||
|
TEST(ranges_test, path_like) {
|
||
|
EXPECT_FALSE((fmt::is_range<path_like, char>::value));
|
||
|
}
|
||
|
|
||
|
#ifdef FMT_USE_STRING_VIEW
|
||
|
struct string_like {
|
||
|
const char* begin();
|
||
|
const char* end();
|
||
|
explicit operator fmt::string_view() const { return "foo"; }
|
||
|
explicit operator std::string_view() const { return "foo"; }
|
||
|
};
|
||
|
|
||
|
TEST(ranges_test, format_string_like) {
|
||
|
EXPECT_EQ(fmt::format("{}", string_like()), "foo");
|
||
|
}
|
||
|
#endif // FMT_USE_STRING_VIEW
|
||
|
|
||
|
// A range that provides non-const only begin()/end() to test fmt::join handles
|
||
|
// that.
|
||
|
//
|
||
|
// Some ranges (e.g. those produced by range-v3's views::filter()) can cache
|
||
|
// information during iteration so they only provide non-const begin()/end().
|
||
|
template <typename T> class non_const_only_range {
|
||
|
private:
|
||
|
std::vector<T> vec;
|
||
|
|
||
|
public:
|
||
|
using const_iterator = typename ::std::vector<T>::const_iterator;
|
||
|
|
||
|
template <typename... Args>
|
||
|
explicit non_const_only_range(Args&&... args)
|
||
|
: vec(std::forward<Args>(args)...) {}
|
||
|
|
||
|
const_iterator begin() { return vec.begin(); }
|
||
|
const_iterator end() { return vec.end(); }
|
||
|
};
|
||
|
|
||
|
template <typename T> class noncopyable_range {
|
||
|
private:
|
||
|
std::vector<T> vec;
|
||
|
|
||
|
public:
|
||
|
using const_iterator = typename ::std::vector<T>::const_iterator;
|
||
|
|
||
|
template <typename... Args>
|
||
|
explicit noncopyable_range(Args&&... args)
|
||
|
: vec(std::forward<Args>(args)...) {}
|
||
|
|
||
|
noncopyable_range(noncopyable_range const&) = delete;
|
||
|
noncopyable_range(noncopyable_range&) = delete;
|
||
|
|
||
|
const_iterator begin() const { return vec.begin(); }
|
||
|
const_iterator end() const { return vec.end(); }
|
||
|
};
|
||
|
|
||
|
TEST(ranges_test, range) {
|
||
|
noncopyable_range<int> w(3u, 0);
|
||
|
EXPECT_EQ(fmt::format("{}", w), "[0, 0, 0]");
|
||
|
EXPECT_EQ(fmt::format("{}", noncopyable_range<int>(3u, 0)), "[0, 0, 0]");
|
||
|
|
||
|
non_const_only_range<int> x(3u, 0);
|
||
|
EXPECT_EQ(fmt::format("{}", x), "[0, 0, 0]");
|
||
|
EXPECT_EQ(fmt::format("{}", non_const_only_range<int>(3u, 0)), "[0, 0, 0]");
|
||
|
|
||
|
auto y = std::vector<int>(3u, 0);
|
||
|
EXPECT_EQ(fmt::format("{}", y), "[0, 0, 0]");
|
||
|
EXPECT_EQ(fmt::format("{}", std::vector<int>(3u, 0)), "[0, 0, 0]");
|
||
|
|
||
|
const auto z = std::vector<int>(3u, 0);
|
||
|
EXPECT_EQ(fmt::format("{}", z), "[0, 0, 0]");
|
||
|
}
|
||
|
|
||
|
#if !FMT_MSC_VER || FMT_MSC_VER >= 1927
|
||
|
struct unformattable {};
|
||
|
|
||
|
TEST(ranges_test, unformattable_range) {
|
||
|
EXPECT_FALSE((fmt::has_formatter<std::vector<unformattable>,
|
||
|
fmt::format_context>::value));
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
#ifdef FMT_RANGES_TEST_ENABLE_JOIN
|
||
|
TEST(ranges_test, join_tuple) {
|
||
|
// Value tuple args.
|
||
|
auto t1 = std::tuple<char, int, float>('a', 1, 2.0f);
|
||
|
EXPECT_EQ(fmt::format("({})", fmt::join(t1, ", ")), "(a, 1, 2)");
|
||
|
|
||
|
// Testing lvalue tuple args.
|
||
|
int x = 4;
|
||
|
auto t2 = std::tuple<char, int&>('b', x);
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(t2, " + ")), "b + 4");
|
||
|
|
||
|
// Empty tuple.
|
||
|
auto t3 = std::tuple<>();
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(t3, "|")), "");
|
||
|
|
||
|
// Single element tuple.
|
||
|
auto t4 = std::tuple<float>(4.0f);
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(t4, "/")), "4");
|
||
|
}
|
||
|
|
||
|
TEST(ranges_test, join_initializer_list) {
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join({1, 2, 3}, ", ")), "1, 2, 3");
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join({"fmt", "rocks", "!"}, " ")),
|
||
|
"fmt rocks !");
|
||
|
}
|
||
|
|
||
|
struct zstring_sentinel {};
|
||
|
|
||
|
bool operator==(const char* p, zstring_sentinel) { return *p == '\0'; }
|
||
|
bool operator!=(const char* p, zstring_sentinel) { return *p != '\0'; }
|
||
|
|
||
|
struct zstring {
|
||
|
const char* p;
|
||
|
const char* begin() const { return p; }
|
||
|
zstring_sentinel end() const { return {}; }
|
||
|
};
|
||
|
|
||
|
TEST(ranges_test, join_sentinel) {
|
||
|
auto hello = zstring{"hello"};
|
||
|
EXPECT_EQ(fmt::format("{}", hello), "['h', 'e', 'l', 'l', 'o']");
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(hello, "_")), "h_e_l_l_o");
|
||
|
}
|
||
|
|
||
|
TEST(ranges_test, join_range) {
|
||
|
noncopyable_range<int> w(3u, 0);
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(w, ",")), "0,0,0");
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(noncopyable_range<int>(3u, 0), ",")),
|
||
|
"0,0,0");
|
||
|
|
||
|
non_const_only_range<int> x(3u, 0);
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(x, ",")), "0,0,0");
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(non_const_only_range<int>(3u, 0), ",")),
|
||
|
"0,0,0");
|
||
|
|
||
|
auto y = std::vector<int>(3u, 0);
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(y, ",")), "0,0,0");
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(std::vector<int>(3u, 0), ",")),
|
||
|
"0,0,0");
|
||
|
|
||
|
const auto z = std::vector<int>(3u, 0);
|
||
|
EXPECT_EQ(fmt::format("{}", fmt::join(z, ",")), "0,0,0");
|
||
|
}
|
||
|
#endif // FMT_RANGES_TEST_ENABLE_JOIN
|