Add ThreadIterator for iterating over all Threads

runtime/vm/thread_test.cc

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "platform/assert.h"
 #include "vm/isolate.h"
 #include "vm/lockers.h"
 #include "vm/unit_test.h"
 #include "vm/profiler.h"
 #include "vm/thread_pool.h"
@@ skipping 389 matching lines @@
   while (true) {
     isolate->thread_registry()->CheckSafepoint();
     MutexLocker ml(&mutex);
     if (exited == SafepointTestTask::kTaskCount) {
       break;
     }
   }
 }
 
 
+TEST_CASE(ThreadIterator_Count) {
+  intptr_t thread_count_0 = 0;
+  intptr_t thread_count_1 = 0;
+
+  {
+    ThreadIterator ti;
+    while (ti.HasNext()) {
+      Thread* thread = ti.Next();
+      EXPECT(thread != NULL);
+      thread_count_0++;
+    }
+  }
+
+  {
+    ThreadIterator ti;
+    while (ti.HasNext()) {
+      Thread* thread = ti.Next();
+      EXPECT(thread != NULL);
+      thread_count_1++;
+    }
+  }
+
+  EXPECT(thread_count_0 > 0);
+  EXPECT(thread_count_1 > 0);
+  EXPECT(thread_count_0 >= thread_count_1);
+}
+
+
+TEST_CASE(ThreadIterator_FindSelf) {
+  Thread* current = Thread::Current();
+
+  bool found_self = false;
+
+  {
+    ThreadIterator ti;
+    while (ti.HasNext()) {
+      Thread* thread = ti.Next();
+      EXPECT(thread != NULL);
+      if (thread == current) {
+        found_self = true;
+        break;
+      }
+    }
+  }
+
+  EXPECT(found_self);
+}
+
+
+class ThreadIteratorTestHelper : public AllStatic {
+ public:
+  static void AddThreadToList(Thread* thread) {
+    Thread::AddThreadToList(thread);
+  }
+
+  static void RemoveThreadFromList(Thread* thread) {
+    Thread::RemoveThreadFromList(thread);
+  }
+};
+
+
+TEST_CASE(ThreadIterator_RemoveSelf) {
+  Thread* current = Thread::Current();
+
+  // Remove self.
+  ThreadIteratorTestHelper::RemoveThreadFromList(current);

[siva, 2015/10/12 23:03:14]

This test does not seem right, we should not allow a thread that is currently
part of an isolate to be removed from the thread list (reason why it should not
be possible to call this from any random place).

Similarly we should not allow a thread to randomly add itself to the list.

For this test you could create a thread, have it call Dart_EnterIsolate and see
if it has been added to the iterator list and then ensure the thread calls
Dart_ExitIsolate(), the thread dies  and then ensure it is not in the iterator
list.

[Cutch, 2015/10/13 16:41:24]

I've rewritten the test and added assertions that a thread doesn't have an
isolate when it is added / removed from the list.

+
+  // Not in thread list.
+  {
+    bool found_self = false;
+    ThreadIterator ti;
+    while (ti.HasNext()) {
+      Thread* thread = ti.Next();
+      EXPECT(thread != NULL);
+      if (thread == current) {
+        found_self = true;
+        break;
+      }
+    }
+    EXPECT(!found_self);
+  }
+
+  ThreadIteratorTestHelper::AddThreadToList(current);
+
+  // Now in thread list.
+  {
+    bool found_self = false;
+    ThreadIterator ti;
+    while (ti.HasNext()) {
+      Thread* thread = ti.Next();
+      EXPECT(thread != NULL);
+      if (thread == current) {
+        found_self = true;
+        break;
+      }
+    }
+    EXPECT(found_self);
+  }
+
+  // Remove self.
+  ThreadIteratorTestHelper::RemoveThreadFromList(current);
+
+  // Not in thread list.
+  {
+    bool found_self = false;
+    ThreadIterator ti;
+    while (ti.HasNext()) {
+      Thread* thread = ti.Next();
+      EXPECT(thread != NULL);
+      if (thread == current) {
+        found_self = true;
+        break;
+      }
+    }
+    EXPECT(!found_self);
+  }
+
+  ThreadIteratorTestHelper::AddThreadToList(current);
+}
+
 // Test rendezvous of:
 // - helpers in VM code, and
 // - main thread in VM code,
 // organized by
 // - main thread, and
 // - helpers.
 TEST_CASE(SafepointTestVM2) {
   Isolate* isolate = thread->isolate();
   Mutex mutex;
   intptr_t expected_count = 0;
@@ skipping 80 matching lines @@
       isolate->thread_registry()->CheckSafepoint();
       MonitorLocker ml(&done_monitor);
       if (done) {
         break;
       }
     }
   }
 }
 
 }  // namespace dart