1 // Formatting library for C++ - test utilities
2 //
3 // Copyright (c) 2012 - present, Victor Zverovich
4 // All rights reserved.
5 //
6 // For the license information refer to format.h.
7
8 #include <cstdarg>
9 #include <cstdio>
10 #include <string>
11
12 #include "fmt/os.h"
13
14 enum { BUFFER_SIZE = 256 };
15
16 #ifdef _MSC_VER
17 # define FMT_VSNPRINTF vsprintf_s
18 #else
19 # define FMT_VSNPRINTF vsnprintf
20 #endif
21
22 template <size_t SIZE>
safe_sprintf(char (& buffer)[SIZE],const char * format,...)23 void safe_sprintf(char (&buffer)[SIZE], const char* format, ...) {
24 std::va_list args;
25 va_start(args, format);
26 FMT_VSNPRINTF(buffer, SIZE, format, args);
27 va_end(args);
28 }
29
30 // Increment a number in a string.
31 void increment(char* s);
32
33 std::string get_system_error(int error_code);
34
35 extern const char* const FILE_CONTENT;
36
37 // Opens a buffered file for reading.
38 fmt::buffered_file open_buffered_file(FILE** fp = nullptr);
39
safe_fopen(const char * filename,const char * mode)40 inline FILE* safe_fopen(const char* filename, const char* mode) {
41 #if defined(_WIN32) && !defined(__MINGW32__)
42 // Fix MSVC warning about "unsafe" fopen.
43 FILE* f = 0;
44 errno = fopen_s(&f, filename, mode);
45 return f;
46 #else
47 return std::fopen(filename, mode);
48 #endif
49 }
50
51 template <typename Char> class BasicTestString {
52 private:
53 std::basic_string<Char> value_;
54
55 static const Char EMPTY[];
56
57 public:
value_(value)58 explicit BasicTestString(const Char* value = EMPTY) : value_(value) {}
59
value()60 const std::basic_string<Char>& value() const { return value_; }
61 };
62
63 template <typename Char> const Char BasicTestString<Char>::EMPTY[] = {0};
64
65 typedef BasicTestString<char> TestString;
66 typedef BasicTestString<wchar_t> TestWString;
67
68 template <typename Char>
69 std::basic_ostream<Char>& operator<<(std::basic_ostream<Char>& os,
70 const BasicTestString<Char>& s) {
71 os << s.value();
72 return os;
73 }
74
75 class Date {
76 int year_, month_, day_;
77
78 public:
Date(int year,int month,int day)79 Date(int year, int month, int day) : year_(year), month_(month), day_(day) {}
80
year()81 int year() const { return year_; }
month()82 int month() const { return month_; }
day()83 int day() const { return day_; }
84 };
85