mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-14 14:46:12 -04:00
120 lines
4.9 KiB
C++
120 lines
4.9 KiB
C++
// Copyright 2023 Google LLC
|
|
//
|
|
// 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
|
|
//
|
|
// https://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 "internal/flags/nearby_flags.h"
|
|
|
|
#include <cstdint>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <string_view>
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "protobuf-matchers/protocol-buffer-matchers.h"
|
|
#include "gtest/gtest.h"
|
|
#include "absl/strings/string_view.h"
|
|
#include "internal/flags/flag.h"
|
|
#include "internal/flags/flag_reader.h"
|
|
|
|
namespace nearby {
|
|
namespace {
|
|
|
|
constexpr auto kTestBoolFlag =
|
|
flags::Flag<bool>("test_package", "45401515", false);
|
|
constexpr auto kTestInt64Flag =
|
|
flags::Flag<int64_t>("test_package", "45401516", 666);
|
|
constexpr auto kTestDoubleFlag =
|
|
flags::Flag<double>("test_package", "45401517", 66.66);
|
|
constexpr auto kTestStringFlag =
|
|
flags::Flag<absl::string_view>("test_package", "45401518", "string");
|
|
|
|
constexpr bool kTestBoolFlagTestValue = true;
|
|
constexpr int64_t kTestInt64FlagTestValue = 666;
|
|
constexpr double kTestDoubleFlagTestValue = 88.8;
|
|
constexpr std::string_view kTestStringFlagTestValue = "hello";
|
|
|
|
class MockFlagReader : public flags::FlagReader {
|
|
public:
|
|
MOCK_METHOD(bool, GetBoolFlag, (const flags::Flag<bool>& flag), (override));
|
|
MOCK_METHOD(int64_t, GetInt64Flag, (const flags::Flag<int64_t>& flag),
|
|
(override));
|
|
MOCK_METHOD(double, GetDoubleFlag, (const flags::Flag<double>& flag),
|
|
(override));
|
|
MOCK_METHOD(std::string, GetStringFlag,
|
|
(const flags::Flag<absl::string_view>& flag), (override));
|
|
};
|
|
|
|
TEST(NearbyFlags, GetDefaultValues) {
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetBoolFlag(kTestBoolFlag),
|
|
kTestBoolFlag.default_value());
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetInt64Flag(kTestInt64Flag),
|
|
kTestInt64Flag.default_value());
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetDoubleFlag(kTestDoubleFlag),
|
|
kTestDoubleFlag.default_value());
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetStringFlag(kTestStringFlag),
|
|
kTestStringFlag.default_value());
|
|
}
|
|
|
|
TEST(NearbyFlags, OverrideDefaultValues) {
|
|
NearbyFlags::GetInstance().OverrideBoolFlagValue(kTestBoolFlag,
|
|
kTestBoolFlagTestValue);
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetBoolFlag(kTestBoolFlag),
|
|
kTestBoolFlagTestValue);
|
|
NearbyFlags::GetInstance().OverrideInt64FlagValue(kTestInt64Flag,
|
|
kTestInt64FlagTestValue);
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetInt64Flag(kTestInt64Flag),
|
|
kTestInt64FlagTestValue);
|
|
NearbyFlags::GetInstance().OverrideDoubleFlagValue(kTestDoubleFlag,
|
|
kTestDoubleFlagTestValue);
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetDoubleFlag(kTestDoubleFlag),
|
|
kTestDoubleFlagTestValue);
|
|
NearbyFlags::GetInstance().OverrideStringFlagValue(kTestStringFlag,
|
|
kTestStringFlagTestValue);
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetStringFlag(kTestStringFlag),
|
|
kTestStringFlagTestValue);
|
|
NearbyFlags::GetInstance().ResetOverridedValues();
|
|
}
|
|
|
|
TEST(NearbyFlags, SetFlagReader) {
|
|
auto flag_reader = std::make_unique<::testing::NiceMock<MockFlagReader>>();
|
|
NearbyFlags::GetInstance().SetFlagReader(*flag_reader.get());
|
|
EXPECT_CALL(*flag_reader, GetBoolFlag(::testing::_))
|
|
.WillOnce(::testing::Invoke([=](const flags::Flag<bool>& flag) {
|
|
return kTestBoolFlagTestValue;
|
|
}));
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetBoolFlag(kTestBoolFlag),
|
|
kTestBoolFlagTestValue);
|
|
EXPECT_CALL(*flag_reader, GetInt64Flag(::testing::_))
|
|
.WillOnce(::testing::Invoke([=](const flags::Flag<int64_t>& flag) {
|
|
return kTestInt64FlagTestValue;
|
|
}));
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetInt64Flag(kTestInt64Flag),
|
|
kTestInt64FlagTestValue);
|
|
EXPECT_CALL(*flag_reader, GetDoubleFlag(::testing::_))
|
|
.WillOnce(::testing::Invoke([=](const flags::Flag<double>& flag) {
|
|
return kTestDoubleFlagTestValue;
|
|
}));
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetDoubleFlag(kTestDoubleFlag),
|
|
kTestDoubleFlagTestValue);
|
|
EXPECT_CALL(*flag_reader, GetStringFlag(::testing::_))
|
|
.WillOnce(
|
|
::testing::Invoke([=](const flags::Flag<absl::string_view>& flag) {
|
|
return std::string(kTestStringFlagTestValue);
|
|
}));
|
|
EXPECT_EQ(NearbyFlags::GetInstance().GetStringFlag(kTestStringFlag),
|
|
kTestStringFlagTestValue);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace nearby
|