|
| 1 | +#include <gtest/gtest.h> |
| 2 | + |
| 3 | +#include <launchdarkly/server_side/integrations/big_segments/big_segment_store_types.hpp> |
| 4 | +#include <launchdarkly/server_side/integrations/big_segments/ibig_segment_store.hpp> |
| 5 | + |
| 6 | +#include "config/builders/big_segments_builder.hpp" |
| 7 | + |
| 8 | +#include <chrono> |
| 9 | +#include <memory> |
| 10 | + |
| 11 | +using launchdarkly::server_side::config::builders::BigSegmentsBuilder; |
| 12 | +using launchdarkly::server_side::integrations::IBigSegmentStore; |
| 13 | +using launchdarkly::server_side::integrations::Membership; |
| 14 | +using launchdarkly::server_side::integrations::StoreMetadata; |
| 15 | + |
| 16 | +namespace { |
| 17 | + |
| 18 | +// Minimal stub used only to obtain a shared_ptr<IBigSegmentStore>. The builder |
| 19 | +// never invokes the store; it only stores the pointer for later use by the |
| 20 | +// wrapper, so the methods here are unreachable in these tests. |
| 21 | +class StubStore final : public IBigSegmentStore { |
| 22 | + public: |
| 23 | + GetMembershipResult GetMembership( |
| 24 | + std::string const& /*context_hash*/) const override { |
| 25 | + return Membership::FromSegmentRefs({}, {}); |
| 26 | + } |
| 27 | + GetMetadataResult GetMetadata() const override { |
| 28 | + return std::optional<StoreMetadata>{}; |
| 29 | + } |
| 30 | +}; |
| 31 | + |
| 32 | +std::shared_ptr<IBigSegmentStore> MakeStubStore() { |
| 33 | + return std::make_shared<StubStore>(); |
| 34 | +} |
| 35 | + |
| 36 | +} // namespace |
| 37 | + |
| 38 | +TEST(BigSegmentsBuilderTest, DefaultsMatchSpec) { |
| 39 | + auto store = MakeStubStore(); |
| 40 | + auto const cfg = BigSegmentsBuilder(store).Build(); |
| 41 | + |
| 42 | + EXPECT_EQ(cfg.context_cache_size, |
| 43 | + BigSegmentsBuilder::kDefaultContextCacheSize); |
| 44 | + EXPECT_EQ(cfg.context_cache_time, |
| 45 | + BigSegmentsBuilder::kDefaultContextCacheTime); |
| 46 | + EXPECT_EQ(cfg.status_poll_interval, |
| 47 | + BigSegmentsBuilder::kDefaultStatusPollInterval); |
| 48 | + EXPECT_EQ(cfg.stale_after, BigSegmentsBuilder::kDefaultStaleAfter); |
| 49 | +} |
| 50 | + |
| 51 | +TEST(BigSegmentsBuilderTest, BuildPreservesStoreIdentity) { |
| 52 | + auto store = MakeStubStore(); |
| 53 | + auto const cfg = BigSegmentsBuilder(store).Build(); |
| 54 | + EXPECT_EQ(cfg.store.get(), store.get()); |
| 55 | +} |
| 56 | + |
| 57 | +TEST(BigSegmentsBuilderTest, AcceptsNullStore) { |
| 58 | + // The builder doesn't validate the store; downstream components treat a |
| 59 | + // null store as "Big Segments not configured". |
| 60 | + auto const cfg = BigSegmentsBuilder(nullptr).Build(); |
| 61 | + EXPECT_EQ(cfg.store, nullptr); |
| 62 | +} |
| 63 | + |
| 64 | +TEST(BigSegmentsBuilderTest, SettersOverrideEachField) { |
| 65 | + using namespace std::chrono_literals; |
| 66 | + auto store = MakeStubStore(); |
| 67 | + auto const cfg = BigSegmentsBuilder(store) |
| 68 | + .ContextCacheSize(7) |
| 69 | + .ContextCacheTime(11s) |
| 70 | + .StatusPollInterval(13s) |
| 71 | + .StaleAfter(60s) |
| 72 | + .Build(); |
| 73 | + |
| 74 | + EXPECT_EQ(cfg.context_cache_size, 7u); |
| 75 | + EXPECT_EQ(cfg.context_cache_time, 11s); |
| 76 | + EXPECT_EQ(cfg.status_poll_interval, 13s); |
| 77 | + EXPECT_EQ(cfg.stale_after, 60s); |
| 78 | +} |
| 79 | + |
| 80 | +TEST(BigSegmentsBuilderTest, ZeroDurationsAreCoercedToDefaults) { |
| 81 | + auto store = MakeStubStore(); |
| 82 | + auto const cfg = BigSegmentsBuilder(store) |
| 83 | + .ContextCacheTime(std::chrono::milliseconds::zero()) |
| 84 | + .StatusPollInterval(std::chrono::milliseconds::zero()) |
| 85 | + .StaleAfter(std::chrono::milliseconds::zero()) |
| 86 | + .Build(); |
| 87 | + |
| 88 | + EXPECT_EQ(cfg.context_cache_time, |
| 89 | + BigSegmentsBuilder::kDefaultContextCacheTime); |
| 90 | + EXPECT_EQ(cfg.status_poll_interval, |
| 91 | + BigSegmentsBuilder::kDefaultStatusPollInterval); |
| 92 | + EXPECT_EQ(cfg.stale_after, BigSegmentsBuilder::kDefaultStaleAfter); |
| 93 | +} |
| 94 | + |
| 95 | +TEST(BigSegmentsBuilderTest, NegativeDurationsAreCoercedToDefaults) { |
| 96 | + auto store = MakeStubStore(); |
| 97 | + auto const cfg = BigSegmentsBuilder(store) |
| 98 | + .ContextCacheTime(std::chrono::milliseconds{-1}) |
| 99 | + .StatusPollInterval(std::chrono::milliseconds{-1}) |
| 100 | + .StaleAfter(std::chrono::milliseconds{-1}) |
| 101 | + .Build(); |
| 102 | + |
| 103 | + EXPECT_EQ(cfg.context_cache_time, |
| 104 | + BigSegmentsBuilder::kDefaultContextCacheTime); |
| 105 | + EXPECT_EQ(cfg.status_poll_interval, |
| 106 | + BigSegmentsBuilder::kDefaultStatusPollInterval); |
| 107 | + EXPECT_EQ(cfg.stale_after, BigSegmentsBuilder::kDefaultStaleAfter); |
| 108 | +} |
| 109 | + |
| 110 | +TEST(BigSegmentsBuilderTest, BuildClampsPollIntervalToStaleAfter) { |
| 111 | + // BIGSEG §1.4.6: when poll interval > stale-after, clamp poll to |
| 112 | + // stale-after so the SDK detects staleness within one poll cycle. |
| 113 | + using namespace std::chrono_literals; |
| 114 | + auto store = MakeStubStore(); |
| 115 | + auto const cfg = BigSegmentsBuilder(store) |
| 116 | + .StatusPollInterval(10s) |
| 117 | + .StaleAfter(3s) |
| 118 | + .Build(); |
| 119 | + |
| 120 | + EXPECT_EQ(cfg.status_poll_interval, 3s); |
| 121 | + EXPECT_EQ(cfg.stale_after, 3s); |
| 122 | +} |
| 123 | + |
| 124 | +TEST(BigSegmentsBuilderTest, BuildPreservesPollIntervalWhenWithinStaleAfter) { |
| 125 | + using namespace std::chrono_literals; |
| 126 | + auto store = MakeStubStore(); |
| 127 | + auto const cfg = BigSegmentsBuilder(store) |
| 128 | + .StatusPollInterval(3s) |
| 129 | + .StaleAfter(10s) |
| 130 | + .Build(); |
| 131 | + |
| 132 | + EXPECT_EQ(cfg.status_poll_interval, 3s); |
| 133 | + EXPECT_EQ(cfg.stale_after, 10s); |
| 134 | +} |
| 135 | + |
| 136 | +TEST(BigSegmentsBuilderTest, BuildIsRepeatable) { |
| 137 | + using namespace std::chrono_literals; |
| 138 | + auto store = MakeStubStore(); |
| 139 | + BigSegmentsBuilder builder(store); |
| 140 | + builder.ContextCacheSize(42).ContextCacheTime(2s); |
| 141 | + |
| 142 | + auto const cfg1 = builder.Build(); |
| 143 | + auto const cfg2 = builder.Build(); |
| 144 | + |
| 145 | + EXPECT_EQ(cfg1.context_cache_size, cfg2.context_cache_size); |
| 146 | + EXPECT_EQ(cfg1.context_cache_time, cfg2.context_cache_time); |
| 147 | + EXPECT_EQ(cfg1.store.get(), cfg2.store.get()); |
| 148 | +} |
0 commit comments