Abseil Common Libraries (C++) (grcp 依赖)
https://abseil.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
129 lines
3.9 KiB
129 lines
3.9 KiB
// Copyright 2017 The Abseil Authors. |
|
// |
|
// 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 "absl/time/internal/test_util.h" |
|
|
|
#include <algorithm> |
|
#include <cstddef> |
|
#include <cstring> |
|
|
|
#include "absl/base/internal/raw_logging.h" |
|
#include "absl/time/internal/cctz/include/cctz/zone_info_source.h" |
|
|
|
namespace cctz = absl::time_internal::cctz; |
|
|
|
namespace absl { |
|
namespace time_internal { |
|
|
|
#if GTEST_USES_SIMPLE_RE |
|
extern const char kZoneAbbrRE[] = ".*"; // just punt |
|
#else |
|
extern const char kZoneAbbrRE[] = "[A-Za-z]{3,4}|[-+][0-9]{2}([0-9]{2})?"; |
|
#endif |
|
|
|
TimeZone LoadTimeZone(const std::string& name) { |
|
TimeZone tz; |
|
ABSL_RAW_CHECK(LoadTimeZone(name, &tz), name.c_str()); |
|
return tz; |
|
} |
|
|
|
} // namespace time_internal |
|
} // namespace absl |
|
|
|
namespace absl { |
|
namespace time_internal { |
|
namespace cctz_extension { |
|
namespace { |
|
|
|
// Embed the zoneinfo data for time zones used during tests and benchmarks. |
|
// The data was generated using "xxd -i zoneinfo-file". There is no need |
|
// to update the data as long as the tests do not depend on recent changes |
|
// (and the past rules remain the same). |
|
#include "absl/time/internal/zoneinfo.inc" |
|
|
|
const struct ZoneInfo { |
|
const char* name; |
|
const char* data; |
|
std::size_t length; |
|
} kZoneInfo[] = { |
|
// The three real time zones used by :time_test and :time_benchmark. |
|
{"America/Los_Angeles", // |
|
reinterpret_cast<char*>(America_Los_Angeles), America_Los_Angeles_len}, |
|
{"America/New_York", // |
|
reinterpret_cast<char*>(America_New_York), America_New_York_len}, |
|
{"Australia/Sydney", // |
|
reinterpret_cast<char*>(Australia_Sydney), Australia_Sydney_len}, |
|
|
|
// Other zones named in tests but which should fail to load. |
|
{"Invalid/TimeZone", nullptr, 0}, |
|
{"", nullptr, 0}, |
|
|
|
// Also allow for loading the local time zone under TZ=US/Pacific. |
|
{"US/Pacific", // |
|
reinterpret_cast<char*>(America_Los_Angeles), America_Los_Angeles_len}, |
|
|
|
// Allows use of the local time zone from a system-specific location. |
|
#ifdef _MSC_VER |
|
{"localtime", // |
|
reinterpret_cast<char*>(America_Los_Angeles), America_Los_Angeles_len}, |
|
#else |
|
{"/etc/localtime", // |
|
reinterpret_cast<char*>(America_Los_Angeles), America_Los_Angeles_len}, |
|
#endif |
|
}; |
|
|
|
class TestZoneInfoSource : public cctz::ZoneInfoSource { |
|
public: |
|
TestZoneInfoSource(const char* data, std::size_t size) |
|
: data_(data), end_(data + size) {} |
|
|
|
std::size_t Read(void* ptr, std::size_t size) override { |
|
const std::size_t len = std::min<std::size_t>(size, end_ - data_); |
|
memcpy(ptr, data_, len); |
|
data_ += len; |
|
return len; |
|
} |
|
|
|
int Skip(std::size_t offset) override { |
|
data_ += std::min<std::size_t>(offset, end_ - data_); |
|
return 0; |
|
} |
|
|
|
private: |
|
const char* data_; |
|
const char* const end_; |
|
}; |
|
|
|
std::unique_ptr<cctz::ZoneInfoSource> TestFactory( |
|
const std::string& name, |
|
const std::function<std::unique_ptr<cctz::ZoneInfoSource>( |
|
const std::string& name)>& /*fallback_factory*/) { |
|
for (const ZoneInfo& zoneinfo : kZoneInfo) { |
|
if (name == zoneinfo.name) { |
|
if (zoneinfo.data == nullptr) return nullptr; |
|
return std::unique_ptr<cctz::ZoneInfoSource>( |
|
new TestZoneInfoSource(zoneinfo.data, zoneinfo.length)); |
|
} |
|
} |
|
ABSL_RAW_LOG(FATAL, "Unexpected time zone \"%s\" in test", name.c_str()); |
|
return nullptr; |
|
} |
|
|
|
} // namespace |
|
|
|
ZoneInfoSourceFactory zone_info_source_factory = TestFactory; |
|
|
|
} // namespace cctz_extension |
|
} // namespace time_internal |
|
} // namespace absl
|
|
|