From 901e3d0628fd7ecbf5f0fd7e125f02f8125d65bf Mon Sep 17 00:00:00 2001
From: Tom Tan <Tom.Tan@microsoft.com>
Date: Fri, 30 Jul 2021 22:22:36 -0700
Subject: [PATCH] Fix race condition in circular buffer simulation test

---
 sdk/test/common/circular_buffer_test.cc | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/sdk/test/common/circular_buffer_test.cc b/sdk/test/common/circular_buffer_test.cc
index 11e64a65b5..a20c3e42aa 100644
--- a/sdk/test/common/circular_buffer_test.cc
+++ b/sdk/test/common/circular_buffer_test.cc
@@ -59,12 +59,11 @@ void RunNumberConsumer(CircularBuffer<uint32_t> &buffer,
 {
   while (true)
   {
-    auto allotment = buffer.Peek();
-    if (exit && allotment.empty())
+    if (exit && buffer.Peek().empty())
     {
       return;
     }
-    auto n = std::uniform_int_distribution<size_t>{0, allotment.size()}(RandomNumberGenerator);
+    auto n = std::uniform_int_distribution<size_t>{0, buffer.Peek().size()}(RandomNumberGenerator);
     buffer.Consume(n, [&](CircularBufferRange<AtomicUniquePtr<uint32_t>> range) noexcept {
       assert(range.size() == n);
       range.ForEach([&](AtomicUniquePtr<uint32_t> &ptr) noexcept {