Initial work

external/android-emugl/shared/emugl/common/thread_unittest.cpp

@@ -0,0 +1,164 @@
+// Copyright (C) 2014 The Android Open Source Project
+//
+// 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
+//
+// http://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 "emugl/common/thread.h"
+
+#include "emugl/common/mutex.h"
+
+#include <gtest/gtest.h>
+
+namespace emugl {
+
+namespace {
+
+// A simple thread instance that does nothing at all and exits immediately.
+class EmptyThread : public ::emugl::Thread {
+public:
+    intptr_t main() { return 42; }
+};
+
+class CountingThread : public ::emugl::Thread {
+public:
+    class State {
+    public:
+        State() : mLock(), mCount(0) {}
+        ~State() {}
+
+        void increment() {
+            mLock.lock();
+            mCount++;
+            mLock.unlock();
+        }
+
+        int count() const {
+            int ret;
+            mLock.lock();
+            ret = mCount;
+            mLock.unlock();
+            return ret;
+        }
+
+    private:
+        mutable Mutex mLock;
+        int mCount;
+    };
+
+    CountingThread(State* state) : mState(state) {}
+
+    intptr_t main() {
+        mState->increment();
+        return 0;
+    }
+
+private:
+    State* mState;
+};
+
+class WaitingThread : public ::emugl::Thread {
+public:
+    WaitingThread(Mutex* lock) : mLock(lock) {}
+
+    intptr_t main() {
+        // Try to acquire lock.
+        mLock->lock();
+
+        // Then try to release it.
+        mLock->unlock();
+        return 0;
+    }
+private:
+    Mutex* mLock;
+};
+
+}  // namespace
+
+TEST(ThreadTest, WaitForSimpleThread) {
+    Thread* thread = new EmptyThread();
+    EXPECT_TRUE(thread);
+    EXPECT_TRUE(thread->start());
+    intptr_t status;
+    EXPECT_TRUE(thread->wait(&status));
+    EXPECT_EQ(42, status);
+}
+
+TEST(ThreadTest, WaitForMultipleThreads) {
+    CountingThread::State state;
+    const size_t kMaxThreads = 100;
+    Thread* threads[kMaxThreads];
+
+    // Create all threads.
+    for (size_t n = 0; n < kMaxThreads; ++n) {
+        threads[n] = new CountingThread(&state);
+        EXPECT_TRUE(threads[n]) << "thread " << n;
+    }
+
+    // Start them all.
+    for (size_t n = 0; n < kMaxThreads; ++n) {
+        EXPECT_TRUE(threads[n]->start()) << "thread " << n;
+    }
+
+    // Wait for them all.
+    for (size_t n = 0; n < kMaxThreads; ++n) {
+        EXPECT_TRUE(threads[n]->wait(NULL)) << "thread " << n;
+    }
+
+    // Check state.
+    EXPECT_EQ((int)kMaxThreads, state.count());
+
+    // Delete them all.
+    for (size_t n = 0; n < kMaxThreads; ++n) {
+        delete threads[n];
+    }
+}
+
+TEST(ThreadTest, TryWaitForMultipleThreads) {
+    Mutex lock;
+    const size_t kMaxThreads = 100;
+    Thread* threads[kMaxThreads];
+
+    // Create all threads.
+    for (size_t n = 0; n < kMaxThreads; ++n) {
+        threads[n] = new WaitingThread(&lock);
+        EXPECT_TRUE(threads[n]) << "thread " << n;
+    }
+
+    // Acquire the lock, this will block all threads.
+    lock.lock();
+
+    // Start them all.
+    for (size_t n = 0; n < kMaxThreads; ++n) {
+        EXPECT_TRUE(threads[n]->start()) << "thread " << n;
+    }
+
+    // Check that tryWait() fails for all threads.
+    for (size_t n = 0; n < kMaxThreads; ++n) {
+        EXPECT_FALSE(threads[n]->tryWait(NULL)) << "thread" << n;
+    }
+
+    // Release the lock, this will unblock all threads.
+    lock.unlock();
+
+    // Wait for them all.
+    for (size_t n = 0; n < kMaxThreads; ++n) {
+        EXPECT_TRUE(threads[n]->wait(NULL)) << "thread " << n;
+        EXPECT_TRUE(threads[n]->tryWait(NULL)) << "thread " << n;
+    }
+
+    // Delete them all.
+    for (size_t n = 0; n < kMaxThreads; ++n) {
+        delete threads[n];
+    }
+}
+
+}  // namespace emugl