mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-14 22:56:12 -04:00
Alexey Polyudov
b724d792dd
Signed-off-by: Alexey Polyudov <apolyudov@google.com> Change-Id: I9a7d3c3d9790dc58c0d4c2bd2134499337b9a17b
109 lines
2.8 KiB
C++
109 lines
2.8 KiB
C++
// Copyright 2020 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 "platform/public/multi_thread_executor.h"
|
|
|
|
#include <atomic>
|
|
#include <functional>
|
|
|
|
#include "platform/base/exception.h"
|
|
#include "gtest/gtest.h"
|
|
#include "absl/synchronization/mutex.h"
|
|
#include "absl/time/clock.h"
|
|
#include "absl/time/time.h"
|
|
|
|
namespace location {
|
|
namespace nearby {
|
|
|
|
namespace {
|
|
const int kMaxThreads = 5;
|
|
}
|
|
|
|
TEST(MultiThreadExecutorTest, ConsructorDestructorWorks) {
|
|
MultiThreadExecutor executor(kMaxThreads);
|
|
}
|
|
|
|
TEST(MultiThreadExecutorTest, CanExecute) {
|
|
absl::CondVar cond;
|
|
std::atomic_bool done = false;
|
|
MultiThreadExecutor executor(kMaxThreads);
|
|
executor.Execute([&done, &cond]() {
|
|
done = true;
|
|
cond.SignalAll();
|
|
});
|
|
absl::Mutex mutex;
|
|
{
|
|
absl::MutexLock lock(&mutex);
|
|
if (!done) {
|
|
cond.WaitWithTimeout(&mutex, absl::Seconds(1));
|
|
}
|
|
}
|
|
EXPECT_TRUE(done);
|
|
}
|
|
|
|
TEST(MultiThreadExecutorTest, JobsExecuteInParallel) {
|
|
absl::Mutex mutex;
|
|
absl::CondVar thread_cond;
|
|
absl::CondVar test_cond;
|
|
MultiThreadExecutor executor(kMaxThreads);
|
|
int count = 0;
|
|
|
|
for (int i = 0; i < kMaxThreads; ++i) {
|
|
executor.Execute([&count, &mutex, &test_cond, &thread_cond]() {
|
|
absl::MutexLock lock(&mutex);
|
|
count++;
|
|
test_cond.Signal();
|
|
thread_cond.Wait(&mutex);
|
|
count--;
|
|
test_cond.Signal();
|
|
});
|
|
}
|
|
|
|
{
|
|
absl::Duration duration = absl::Milliseconds(kMaxThreads * 100);
|
|
absl::MutexLock lock(&mutex);
|
|
while (count < kMaxThreads) {
|
|
absl::Time start = absl::Now();
|
|
if (test_cond.WaitWithTimeout(&mutex, duration)) break;
|
|
duration -= absl::Now() - start;
|
|
}
|
|
}
|
|
|
|
EXPECT_EQ(count, kMaxThreads);
|
|
thread_cond.SignalAll();
|
|
|
|
{
|
|
absl::Duration duration = absl::Milliseconds(kMaxThreads * 100);
|
|
absl::MutexLock lock(&mutex);
|
|
while (count > 0) {
|
|
absl::Time start = absl::Now();
|
|
if (test_cond.WaitWithTimeout(&mutex, duration)) break;
|
|
duration -= absl::Now() - start;
|
|
}
|
|
}
|
|
EXPECT_EQ(count, 0);
|
|
}
|
|
|
|
TEST(MultiThreadExecutorTest, CanSubmit) {
|
|
MultiThreadExecutor executor(kMaxThreads);
|
|
Future<bool> future;
|
|
bool submitted =
|
|
executor.Submit<bool>([]() { return ExceptionOr<bool>{true}; }, &future);
|
|
EXPECT_TRUE(submitted);
|
|
EXPECT_TRUE(future.Get().result());
|
|
}
|
|
|
|
} // namespace nearby
|
|
} // namespace location
|