123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115 |
- /*
- * Copyright 2015-present Facebook, Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #include <folly/Format.h>
- #include <glog/logging.h>
- #include <folly/FBVector.h>
- #include <folly/FileUtil.h>
- #include <folly/Portability.h>
- #include <folly/dynamic.h>
- #include <folly/json.h>
- #include <folly/portability/GTest.h>
- #include <folly/small_vector.h>
- using namespace folly;
- TEST(FormatOther, file) {
- // Test writing to FILE. I'd use open_memstream but that's not available
- // outside of Linux (even though it's in POSIX.1-2008).
- {
- int fds[2];
- CHECK_ERR(pipe(fds));
- SCOPE_EXIT {
- // fclose on Windows automatically closes the underlying
- // file descriptor.
- if (!kIsWindows) {
- closeNoInt(fds[1]);
- }
- };
- {
- FILE* fp = fdopen(fds[1], "wb");
- PCHECK(fp);
- SCOPE_EXIT {
- fclose(fp);
- };
- writeTo(fp, format("{} {}", 42, 23)); // <= 512 bytes (PIPE_BUF)
- }
- char buf[512];
- ssize_t n = readFull(fds[0], buf, sizeof(buf));
- CHECK_GE(n, 0);
- EXPECT_EQ("42 23", std::string(buf, n));
- }
- }
- TEST(FormatOther, dynamic) {
- auto dyn = parseJson(
- "{\n"
- " \"hello\": \"world\",\n"
- " \"x\": [20, 30],\n"
- " \"y\": {\"a\" : 42}\n"
- "}");
- EXPECT_EQ("world", sformat("{0[hello]}", dyn));
- EXPECT_THROW(sformat("{0[none]}", dyn), std::out_of_range);
- EXPECT_EQ("world", sformat("{0[hello]}", defaulted(dyn, "meow")));
- EXPECT_EQ("meow", sformat("{0[none]}", defaulted(dyn, "meow")));
- EXPECT_EQ("20", sformat("{0[x.0]}", dyn));
- EXPECT_THROW(sformat("{0[x.2]}", dyn), std::out_of_range);
- // No support for "deep" defaulting (dyn["x"] is not defaulted)
- auto v = dyn.at("x");
- EXPECT_EQ("20", sformat("{0[0]}", v));
- EXPECT_THROW(sformat("{0[2]}", v), std::out_of_range);
- EXPECT_EQ("20", sformat("{0[0]}", defaulted(v, 42)));
- EXPECT_EQ("42", sformat("{0[2]}", defaulted(v, 42)));
- EXPECT_EQ("42", sformat("{0[y.a]}", dyn));
- EXPECT_EQ("(null)", sformat("{}", dynamic(nullptr)));
- }
- namespace {
- template <class T>
- void testFormatSeq() {
- T v{10, 20, 30};
- EXPECT_EQ("30 10", sformat("{0[2]} {0[0]}", v));
- EXPECT_EQ("0020", sformat("{0[1]:04}", v));
- EXPECT_EQ("0020", svformat("{1:04}", v));
- EXPECT_EQ("10 20", svformat("{} {}", v));
- EXPECT_EQ("10 20 0030", svformat("{} {} {:04}", v));
- }
- } // namespace
- TEST(FormatOther, fbvector) {
- testFormatSeq<fbvector<int>>();
- }
- TEST(FormatOther, small_vector) {
- testFormatSeq<small_vector<int, 2>>();
- }
- int main(int argc, char* argv[]) {
- testing::InitGoogleTest(&argc, argv);
- gflags::ParseCommandLineFlags(&argc, &argv, true);
- return RUN_ALL_TESTS();
- }
|