Implemented submittable_executor and some other missed files

A file was named wrong. It was changed.
2026-09-16 07:36:10 -04:00 · 2023-08-07 15:16:43 -05:00
parent e6f52fba98
commit fa197d2036
7 changed files with 745 additions and 0 deletions
@@ -0,0 +1,250 @@
+// Copyright 2021 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/platform/implementation/linux/submittable_executor.h"
+
+#include <utility>
+#include <thread>
+
+#include "gtest/gtest.h"
+#include "absl/synchronization/blocking_counter.h"
+#include "absl/synchronization/notification.h"
+#include "absl/time/time.h"
+#include "internal/platform/implementation/linux/test_data.h"
+
+namespace nearby {
+namespace linux {
+namespace {
+
+constexpr absl::Duration kWaitTimeout = absl::Milliseconds(200);
+
+TEST(SubmittableExecutorTests, SingleThreadedExecuteSucceeds) {
+  absl::Notification notification;
+  // Arrange
+  std::string expected(RUNNABLE_0_TEXT.c_str());
+
+  auto submittableExecutor = std::make_unique<SubmittableExecutor>();
+  std::string output = std::string();
+  // Container to note threads that ran
+  auto threadIds = std::make_unique<std::vector<std::thread::id>>();
+
+  threadIds->push_back(std::this_thread::get_id());
+
+  // Act
+  submittableExecutor->Execute([&]() {
+    threadIds->push_back(std::this_thread::get_id());
+    output.append(RUNNABLE_0_TEXT.c_str());
+    notification.Notify();
+  });
+
+  ASSERT_TRUE(notification.WaitForNotificationWithTimeout(kWaitTimeout));
+  submittableExecutor->Shutdown();
+
+  //  Assert
+  //  We should've run 1 time on the main thread, and 1 times on the
+  //  workerThread
+  ASSERT_EQ(threadIds->size(), 2);
+  //  We should still be on the main thread
+  ASSERT_EQ(std::this_thread::get_id(), threadIds->at(0));
+  //  We should've run all runnables on the worker thread
+  ASSERT_EQ(output, expected);
+}
+
+TEST(SubmittableExecutorTests, SingleThreadedExecuteAfterShutdownFails) {
+  // Arrange
+  std::string expected("");
+
+  auto submittableExecutor = std::make_unique<SubmittableExecutor>();
+  std::string output = std::string();
+  // Container to note threads that ran
+  auto threadIds = std::make_unique<std::vector<std::thread::id>>();
+
+  threadIds->push_back(std::this_thread::get_id());
+
+  submittableExecutor->Shutdown();
+
+  // Act
+  submittableExecutor->Execute([&output, &threadIds]() {
+    threadIds->push_back(std::this_thread::get_id());
+    output.append(RUNNABLE_0_TEXT.c_str());
+  });
+
+  //  Assert
+  //  We should've run 1 time on the main thread, and 0 times on the
+  //  workerThread
+  ASSERT_EQ(threadIds->size(), 1);
+  //  We should still be on the main thread
+  ASSERT_EQ(std::this_thread::get_id(), threadIds->at(0));
+  //  We should've run all runnables on the worker thread
+  ASSERT_EQ(output, expected);
+}
+
+TEST(SubmittableExecutorTests, SingleThreadedDoSubmitSucceeds) {
+  absl::Notification notification;
+  // Arrange
+  std::string expected(RUNNABLE_0_TEXT.c_str());
+
+  auto submittableExecutor = std::make_unique<SubmittableExecutor>();
+  std::string output = std::string();
+  // Container to note threads that ran
+  auto threadIds = std::make_unique<std::vector<std::thread::id>>();
+
+  threadIds->push_back(std::this_thread::get_id());
+
+  // Act
+  auto result = submittableExecutor->DoSubmit([&]() {
+    threadIds->push_back(std::this_thread::get_id());
+    output.append(RUNNABLE_0_TEXT.c_str());
+    notification.Notify();
+  });
+
+  ASSERT_TRUE(notification.WaitForNotificationWithTimeout(kWaitTimeout));
+  submittableExecutor->Shutdown();
+
+  //  Assert
+  //  We should've said we were going to run this one
+  ASSERT_TRUE(result);
+  //  We should've run 1 time on the main thread, and 1 times on the
+  //  workerThread
+  ASSERT_EQ(threadIds->size(), 2);
+  //  We should still be on the main thread
+  ASSERT_EQ(std::this_thread::get_id(), threadIds->at(0));
+  //  We should've run all runnables on the worker thread
+  ASSERT_EQ(output, expected);
+}
+
+TEST(SubmittableExecutorTests,
+     SingleThreadedDoSubmitAfterShutdownReturnsFalse) {
+  // Arrange
+  std::string expected("");
+
+  auto submittableExecutor = std::make_unique<SubmittableExecutor>();
+  std::unique_ptr<std::string> output = std::make_unique<std::string>();
+  // Container to note threads that ran
+  auto threadIds = std::make_unique<std::vector<std::thread::id>>();
+
+  threadIds->push_back(std::this_thread::get_id());
+
+  submittableExecutor->Shutdown();
+
+  // Act
+  auto result = submittableExecutor->DoSubmit([&output, &threadIds]() {
+    threadIds->push_back(std::this_thread::get_id());
+    output->append(RUNNABLE_0_TEXT.c_str());
+  });
+
+  //  Assert
+  //  We should've said we were going to run this one
+  ASSERT_FALSE(result);
+  //  We should've run 1 time on the main thread, and 1 times on the
+  //  workerThread
+  ASSERT_EQ(threadIds->size(), 1);
+  //  We should still be on the main thread
+  ASSERT_EQ(std::this_thread::get_id(), threadIds->at(0));
+  //  We should've run all runnables on the worker thread
+  ASSERT_EQ(*output.get(), expected);
+}
+
+TEST(SubmittableExecutorTests, SingleThreadedExecuteMultipleTasksSucceeds) {
+  absl::BlockingCounter blocking_counter(5);
+
+  // Arrange
+  std::string expected(RUNNABLE_ALL_TEXT.c_str());
+
+  auto submittableExecutor = std::make_unique<SubmittableExecutor>();
+  std::unique_ptr<std::string> output = std::make_unique<std::string>();
+  // Container to note threads that ran
+  auto threadIds = std::make_unique<std::vector<std::thread::id>>();
+
+  threadIds->push_back(std::this_thread::get_id());
+
+  // Act
+  for (int index = 0; index < 5; index++) {
+    submittableExecutor->Execute([&, index]() {
+      threadIds->push_back(std::this_thread::get_id());
+      char buffer[128];
+      snprintf(buffer, sizeof(buffer), "%s%d, ", RUNNABLE_TEXT.c_str(), index);
+      output->append(std::string(buffer));
+      blocking_counter.DecrementCount();
+    });
+  }
+
+  blocking_counter.Wait();
+  submittableExecutor->Shutdown();
+
+  //  Assert
+  //  We should've run 1 time on the main thread, and 5 times on the
+  //  workerThread
+  ASSERT_EQ(threadIds->size(), 6);
+  //  We should still be on the main thread
+  ASSERT_EQ(std::this_thread::get_id(), threadIds->at(0));
+  //  We should've run all runnables on the worker thread
+  auto workerThreadId = threadIds->at(1);
+  for (int index = 1; index < threadIds->size(); index++) {
+    ASSERT_EQ(threadIds->at(index), workerThreadId);
+  }
+
+  //  We should of run them in the order submitted
+  ASSERT_EQ(*output.get(), expected);
+}
+
+TEST(SubmittableExecutorTests, SingleThreadedDoSubmitMultipleTasksSucceeds) {
+  absl::BlockingCounter blocking_counter(5);
+
+  // Arrange
+  std::string expected(RUNNABLE_ALL_TEXT.c_str());
+
+  auto submittableExecutor = std::make_unique<SubmittableExecutor>();
+  std::unique_ptr<std::string> output = std::make_unique<std::string>();
+  // Container to note threads that ran
+  auto threadIds = std::make_unique<std::vector<std::thread::id>>();
+
+  threadIds->push_back(std::this_thread::get_id());
+
+  // Act
+  bool result = true;
+  for (int index = 0; index < 5; index++) {
+    result &= submittableExecutor->DoSubmit([&, index]() {
+      threadIds->push_back(std::this_thread::get_id());
+      char buffer[128];
+      snprintf(buffer, sizeof(buffer), "%s%d, ", RUNNABLE_TEXT.c_str(), index);
+      output->append(std::string(buffer));
+      blocking_counter.DecrementCount();
+    });
+  }
+
+  blocking_counter.Wait();
+  submittableExecutor->Shutdown();
+
+  //  Assert
+  //  All of these should have submitted
+  ASSERT_TRUE(result);
+  //  We should've run 1 time on the main thread, and 5 times on the
+  //  workerThread
+  ASSERT_EQ(threadIds->size(), 6);
+  //  We should still be on the main thread
+  ASSERT_EQ(std::this_thread::get_id(), threadIds->at(0));
+  //  We should've run all runnables on the worker thread
+  auto workerThreadId = threadIds->at(1);
+  for (int index = 1; index < threadIds->size(); index++) {
+    ASSERT_EQ(threadIds->at(index), workerThreadId);
+  }
+
+  //  We should of run them in the order submitted
+  ASSERT_EQ(*output.get(), expected);
+}
+
+}  // namespace
+}  // namespace linux
+}  // namespace nearby