mirror/fmt
1
0
Fork 0
mirror of https://github.com/fmtlib/fmt.git synced 2025-01-26 12:35:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use a mock to test custom argument formatter

Browse Source
This commit is contained in:
vitaut 2016-03-19 07:36:28 -07:00
parent 52f89065e1
commit a99891e7a5
1 changed files with 12 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
test/format-test.cc
View File

@ -1659,35 +1659,27 @@ TEST(FormatTest, EmptyCustomOutput) {
EXPECT_EQ("", fmt::format("{}", EmptyTest())); EXPECT_EQ("", fmt::format("{}", EmptyTest()));
} }
class CustomArgFormatter : class MockArgFormatter :
public fmt::internal::ArgFormatterBase<CustomArgFormatter, char> { public fmt::internal::ArgFormatterBase<MockArgFormatter, char> {
public: public:
typedef fmt::internal::ArgFormatterBase<CustomArgFormatter, char> Base; typedef fmt::internal::ArgFormatterBase<MockArgFormatter, char> Base;
CustomArgFormatter(fmt::BasicFormatter<char, CustomArgFormatter> &f, MockArgFormatter(fmt::BasicFormatter<char, MockArgFormatter> &f,
fmt::FormatSpec &s, const char *) fmt::FormatSpec &s, const char *)
: fmt::internal::ArgFormatterBase<CustomArgFormatter, char>(f.writer(), s) { : fmt::internal::ArgFormatterBase<MockArgFormatter, char>(f.writer(), s) {
EXPECT_CALL(*this, visit_int(42));
} }
void visit_int(int value) { MOCK_METHOD1(visit_int, void (int value));
fmt::FormatSpec &spec = this->spec();
if (spec.type() == 'x')
visit_uint(value); // convert to unsigned and format
else
Base::visit_int(value);
}
}; };
std::string custom_format(const char *format_str, fmt::ArgList args) { void custom_format(const char *format_str, fmt::ArgList args) {
fmt::MemoryWriter writer; fmt::MemoryWriter writer;
fmt::BasicFormatter<char, CustomArgFormatter> formatter(args, writer); fmt::BasicFormatter<char, MockArgFormatter> formatter(args, writer);
formatter.format(format_str); formatter.format(format_str);
return writer.str();
} }
FMT_VARIADIC(std::string, custom_format, const char *) FMT_VARIADIC(void, custom_format, const char *)
TEST(FormatTest, CustomArgFormatter) { TEST(FormatTest, CustomArgFormatter) {
int x = -0xbeef; custom_format("{}", 42);
EXPECT_EQ(fmt::format("{:x}", static_cast<unsigned>(x)),
custom_format("{:x}", x));
} }