diff options
Diffstat (limited to 'src/test/strtol.cc')
-rw-r--r-- | src/test/strtol.cc | 405 |
1 files changed, 405 insertions, 0 deletions
diff --git a/src/test/strtol.cc b/src/test/strtol.cc new file mode 100644 index 00000000..673bf297 --- /dev/null +++ b/src/test/strtol.cc @@ -0,0 +1,405 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2011 Dreamhost + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ + +#include "common/strtol.h" +#include <math.h> +#include <string> +#include <map> + +#include "gtest/gtest.h" + +static void test_strict_strtoll(const char *str, long long expected, int base) +{ + std::string err; + long long val = strict_strtoll(str, base, &err); + if (!err.empty()) { + ASSERT_EQ(err, ""); + } + else { + ASSERT_EQ(val, expected); + } +} + +static void test_strict_strtol(const char *str, long expected) +{ + std::string err; + long val = strict_strtol(str, 10, &err); + if (!err.empty()) { + ASSERT_EQ(err, ""); + } + else { + ASSERT_EQ(val, expected); + } +} + +static void test_strict_strtod(const char *str, double expected) +{ + std::string err; + double val = strict_strtod(str, &err); + if (!err.empty()) { + ASSERT_EQ(err, ""); + } + else { + // when comparing floats, use a margin of error + if ((expected - 0.001 > val) || (expected + 0.001 < val)) { + ASSERT_EQ(val, expected); + } + } +} + +static void test_strict_strtof(const char *str, float expected) +{ + std::string err; + float val = strict_strtof(str, &err); + if (!err.empty()) { + ASSERT_EQ(err, ""); + } + else { + // when comparing floats, use a margin of error + if ((expected - 0.001 > val) || (expected + 0.001 < val)) { + ASSERT_EQ(val, expected); + } + } +} + +TEST(StrToL, Simple1) { + test_strict_strtoll("123", 123, 10); + test_strict_strtoll("0", 0, 10); + test_strict_strtoll("-123", -123, 10); + test_strict_strtoll("8796093022208", 8796093022208LL, 10); + test_strict_strtoll("-8796093022208", -8796093022208LL, 10); + test_strict_strtoll("123", 123, 0); + test_strict_strtoll("0x7b", 123, 0); + test_strict_strtoll("4d2", 1234, 16); + + test_strict_strtol("208", 208); + test_strict_strtol("-4", -4); + test_strict_strtol("0", 0); + test_strict_strtol("2147483646", 2147483646); + + test_strict_strtof("0.05", 0.05); + test_strict_strtof("0", 0.0); + test_strict_strtof("-0", 0.0); + test_strict_strtof("10000000.5", 10000000.5); + + test_strict_strtod("-0.2", -0.2); + test_strict_strtod("0.1", 0.1); + test_strict_strtod("0", 0.0); +} + +static void test_strict_strtoll_err(const char *str) +{ + std::string err; + strict_strtoll(str, 10, &err); + ASSERT_NE(err, ""); +} + +static void test_strict_strtol_err(const char *str) +{ + std::string err; + strict_strtol(str, 10, &err); + ASSERT_NE(err, ""); +} + +static void test_strict_strtod_err(const char *str) +{ + std::string err; + strict_strtod(str, &err); + ASSERT_NE(err, ""); +} + +static void test_strict_strtof_err(const char *str) +{ + std::string err; + strict_strtof(str, &err); + ASSERT_NE(err, ""); +} + +TEST(StrToL, Error1) { + test_strict_strtoll_err("604462909807314587353088"); // overflow + test_strict_strtoll_err("aw shucks"); // invalid + test_strict_strtoll_err("343245 aw shucks"); // invalid chars at end + test_strict_strtoll_err("-"); // invalid + + test_strict_strtol_err("35 aw shucks"); // invalid chars at end + test_strict_strtol_err("--0"); + test_strict_strtol_err("-"); + + test_strict_strtod_err("345345.0-"); + test_strict_strtod_err("34.0 garbo"); + + test_strict_strtof_err("0.05.0"); +} + + +static void test_strict_iecstrtoll(const char *str) +{ + std::string err; + strict_iecstrtoll(str, &err); + ASSERT_EQ(err, ""); +} + +static void test_strict_iecstrtoll_units(const std::string& foo, + std::string u, const int m) +{ + std::string s(foo); + s.append(u); + const char *str = s.c_str(); + std::string err; + uint64_t r = strict_iecstrtoll(str, &err); + ASSERT_EQ(err, ""); + + str = foo.c_str(); + std::string err2; + long long tmp = strict_strtoll(str, 10, &err2); + ASSERT_EQ(err2, ""); + tmp = (tmp << m); + ASSERT_EQ(tmp, (long long)r); +} + +TEST(IECStrToLL, WithUnits) { + std::map<std::string,int> units; + units["B"] = 0; + units["K"] = 10; + units["M"] = 20; + units["G"] = 30; + units["T"] = 40; + units["P"] = 50; + units["E"] = 60; + units["Ki"] = 10; + units["Mi"] = 20; + units["Gi"] = 30; + units["Ti"] = 40; + units["Pi"] = 50; + units["Ei"] = 60; + + for (std::map<std::string,int>::iterator p = units.begin(); + p != units.end(); ++p) { + // the upper bound of uint64_t is 2^64 = 4E + test_strict_iecstrtoll_units("4", p->first, p->second); + test_strict_iecstrtoll_units("1", p->first, p->second); + test_strict_iecstrtoll_units("0", p->first, p->second); + } +} + +TEST(IECStrToLL, WithoutUnits) { + test_strict_iecstrtoll("1024"); + test_strict_iecstrtoll("1152921504606846976"); + test_strict_iecstrtoll("0"); +} + +static void test_strict_iecstrtoll_err(const char *str) +{ + std::string err; + strict_iecstrtoll(str, &err); + ASSERT_NE(err, ""); +} + +TEST(IECStrToLL, Error) { + test_strict_iecstrtoll_err("1024F"); + test_strict_iecstrtoll_err("QDDSA"); + test_strict_iecstrtoll_err("1b"); + test_strict_iecstrtoll_err("100k"); + test_strict_iecstrtoll_err("1000m"); + test_strict_iecstrtoll_err("1g"); + test_strict_iecstrtoll_err("20t"); + test_strict_iecstrtoll_err("100p"); + test_strict_iecstrtoll_err("1000e"); + test_strict_iecstrtoll_err("B"); + test_strict_iecstrtoll_err("M"); + test_strict_iecstrtoll_err("BM"); + test_strict_iecstrtoll_err("B0wef"); + test_strict_iecstrtoll_err("0m"); + test_strict_iecstrtoll_err("-1"); // it returns uint64_t + test_strict_iecstrtoll_err("-1K"); + test_strict_iecstrtoll_err("1Bi"); + test_strict_iecstrtoll_err("Bi"); + test_strict_iecstrtoll_err("bi"); + test_strict_iecstrtoll_err("gi"); + test_strict_iecstrtoll_err("100ki"); + test_strict_iecstrtoll_err("1000mi"); + test_strict_iecstrtoll_err("1gi"); + test_strict_iecstrtoll_err("20ti"); + test_strict_iecstrtoll_err("100pi"); + test_strict_iecstrtoll_err("1000ei"); + // the upper bound of uint64_t is 2^64 = 4E, so 1024E overflows + test_strict_iecstrtoll_err("1024E"); // overflows after adding the suffix +} + +// since strict_iecstrtoll is an alias of strict_iec_cast<uint64_t>(), quite a few +// of cases are covered by existing test cases of strict_iecstrtoll already. +TEST(StrictIECCast, Error) { + { + std::string err; + // the SI prefix is way too large for `int`. + (void)strict_iec_cast<int>("2E", &err); + ASSERT_NE(err, ""); + } + { + std::string err; + (void)strict_iec_cast<int>("-2E", &err); + ASSERT_NE(err, ""); + } + { + std::string err; + (void)strict_iec_cast<int>("1T", &err); + ASSERT_NE(err, ""); + } + { + std::string err; + (void)strict_iec_cast<int64_t>("2E", &err); + ASSERT_EQ(err, ""); + } + { + std::string err; + (void)strict_iec_cast<int64_t>("-2E", &err); + ASSERT_EQ(err, ""); + } + { + std::string err; + (void)strict_iec_cast<int64_t>("1T", &err); + ASSERT_EQ(err, ""); + } +} + + +static void test_strict_sistrtoll(const char *str) +{ + std::string err; + strict_sistrtoll(str, &err); + ASSERT_EQ(err, ""); +} + +static void test_strict_sistrtoll_units(const std::string& foo, + std::string u, const long long m) +{ + std::string s(foo); + s.append(u); + const char *str = s.c_str(); + std::string err; + uint64_t r = strict_sistrtoll(str, &err); + ASSERT_EQ(err, ""); + + str = foo.c_str(); + std::string err2; + long long tmp = strict_strtoll(str, 10, &err2); + ASSERT_EQ(err2, ""); + tmp = (tmp * m); + ASSERT_EQ(tmp, (long long)r); +} + +TEST(SIStrToLL, WithUnits) { + std::map<std::string,long long> units; + units["K"] = pow(10, 3); + units["M"] = pow(10, 6); + units["G"] = pow(10, 9); + units["T"] = pow(10, 12); + units["P"] = pow(10, 15); + units["E"] = pow(10, 18); + + for (std::map<std::string,long long>::iterator p = units.begin(); + p != units.end(); ++p) { + // the upper bound of uint64_t is 2^64 = 4E + test_strict_sistrtoll_units("4", p->first, p->second); + test_strict_sistrtoll_units("1", p->first, p->second); + test_strict_sistrtoll_units("0", p->first, p->second); + } +} + +TEST(SIStrToLL, WithoutUnits) { + test_strict_sistrtoll("1024"); + test_strict_sistrtoll("1152921504606846976"); + test_strict_sistrtoll("0"); +} + +static void test_strict_sistrtoll_err(const char *str) +{ + std::string err; + strict_sistrtoll(str, &err); + ASSERT_NE(err, ""); +} + +TEST(SIStrToLL, Error) { + test_strict_sistrtoll_err("1024F"); + test_strict_sistrtoll_err("QDDSA"); + test_strict_sistrtoll_err("1b"); + test_strict_sistrtoll_err("100k"); + test_strict_sistrtoll_err("1000m"); + test_strict_sistrtoll_err("1g"); + test_strict_sistrtoll_err("20t"); + test_strict_sistrtoll_err("100p"); + test_strict_sistrtoll_err("1000e"); + test_strict_sistrtoll_err("B"); + test_strict_sistrtoll_err("M"); + test_strict_sistrtoll_err("BM"); + test_strict_sistrtoll_err("B0wef"); + test_strict_sistrtoll_err("0m"); + test_strict_sistrtoll_err("-1"); // it returns uint64_t + test_strict_sistrtoll_err("-1K"); + test_strict_sistrtoll_err("1Bi"); + test_strict_sistrtoll_err("Bi"); + test_strict_sistrtoll_err("bi"); + test_strict_sistrtoll_err("gi"); + test_strict_sistrtoll_err("100ki"); + test_strict_sistrtoll_err("1000mi"); + test_strict_sistrtoll_err("1gi"); + test_strict_sistrtoll_err("20ti"); + test_strict_sistrtoll_err("100pi"); + test_strict_sistrtoll_err("1000ei"); + test_strict_sistrtoll_err("1B"); + // the upper bound of uint64_t is 2^64 = 4E, so 1024E overflows + test_strict_sistrtoll_err("1024E"); // overflows after adding the suffix +} + +// since strict_sistrtoll is an alias of strict_si_cast<uint64_t>(), quite a few +// of cases are covered by existing test cases of strict_sistrtoll already. +TEST(StrictSICast, Error) { + { + std::string err; + // the SI prefix is way too large for `int`. + (void)strict_si_cast<int>("2E", &err); + ASSERT_NE(err, ""); + } + { + std::string err; + (void)strict_si_cast<int>("-2E", &err); + ASSERT_NE(err, ""); + } + { + std::string err; + (void)strict_si_cast<int>("1T", &err); + ASSERT_NE(err, ""); + } + { + std::string err; + (void)strict_si_cast<int64_t>("2E", &err); + ASSERT_EQ(err, ""); + } + { + std::string err; + (void)strict_si_cast<int64_t>("-2E", &err); + ASSERT_EQ(err, ""); + } + { + std::string err; + (void)strict_si_cast<int64_t>("1T", &err); + ASSERT_EQ(err, ""); + } +} + +/* + * Local Variables: + * compile-command: "cd .. ; make unittest_strtol && ./unittest_strtol" + * End: + */ |