| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "mojo/system/dispatcher.h" | 5 #include "mojo/system/dispatcher.h" |
| 6 | 6 |
| 7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
| 8 #include "base/memory/ref_counted.h" | 8 #include "base/memory/ref_counted.h" |
| 9 #include "base/memory/scoped_vector.h" | 9 #include "base/memory/scoped_vector.h" |
| 10 #include "base/synchronization/waitable_event.h" | 10 #include "base/synchronization/waitable_event.h" |
| (...skipping 14 matching lines...) Expand all Loading... |
| 25 friend class base::RefCountedThreadSafe<TrivialDispatcher>; | 25 friend class base::RefCountedThreadSafe<TrivialDispatcher>; |
| 26 virtual ~TrivialDispatcher() {} | 26 virtual ~TrivialDispatcher() {} |
| 27 | 27 |
| 28 DISALLOW_COPY_AND_ASSIGN(TrivialDispatcher); | 28 DISALLOW_COPY_AND_ASSIGN(TrivialDispatcher); |
| 29 }; | 29 }; |
| 30 | 30 |
| 31 TEST(DispatcherTest, Basic) { | 31 TEST(DispatcherTest, Basic) { |
| 32 scoped_refptr<Dispatcher> d(new TrivialDispatcher()); | 32 scoped_refptr<Dispatcher> d(new TrivialDispatcher()); |
| 33 | 33 |
| 34 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | 34 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, |
| 35 d->WriteMessage(NULL, 0, NULL, 0, MOJO_WRITE_MESSAGE_FLAG_NONE)); | 35 d->WriteMessage(NULL, 0, NULL, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
| 36 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | 36 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, |
| 37 d->ReadMessage(NULL, NULL, NULL, NULL, | 37 d->ReadMessage(NULL, NULL, 0, NULL, |
| 38 MOJO_WRITE_MESSAGE_FLAG_NONE)); | 38 MOJO_WRITE_MESSAGE_FLAG_NONE)); |
| 39 Waiter w; | 39 Waiter w; |
| 40 w.Init(); | 40 w.Init(); |
| 41 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, | 41 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, |
| 42 d->AddWaiter(&w, MOJO_WAIT_FLAG_EVERYTHING, 0)); | 42 d->AddWaiter(&w, MOJO_WAIT_FLAG_EVERYTHING, 0)); |
| 43 // Okay to remove even if it wasn't added (or was already removed). | 43 // Okay to remove even if it wasn't added (or was already removed). |
| 44 d->RemoveWaiter(&w); | 44 d->RemoveWaiter(&w); |
| 45 d->RemoveWaiter(&w); | 45 d->RemoveWaiter(&w); |
| 46 | 46 |
| 47 EXPECT_EQ(MOJO_RESULT_OK, d->Close()); | 47 EXPECT_EQ(MOJO_RESULT_OK, d->Close()); |
| 48 | 48 |
| 49 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | 49 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, |
| 50 d->WriteMessage(NULL, 0, NULL, 0, MOJO_WRITE_MESSAGE_FLAG_NONE)); | 50 d->WriteMessage(NULL, 0, NULL, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
| 51 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | 51 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, |
| 52 d->ReadMessage(NULL, NULL, NULL, NULL, | 52 d->ReadMessage(NULL, NULL, 0, NULL, |
| 53 MOJO_WRITE_MESSAGE_FLAG_NONE)); | 53 MOJO_WRITE_MESSAGE_FLAG_NONE)); |
| 54 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | 54 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, |
| 55 d->AddWaiter(&w, MOJO_WAIT_FLAG_EVERYTHING, 0)); | 55 d->AddWaiter(&w, MOJO_WAIT_FLAG_EVERYTHING, 0)); |
| 56 d->RemoveWaiter(&w); | 56 d->RemoveWaiter(&w); |
| 57 } | 57 } |
| 58 | 58 |
| 59 class ThreadSafetyStressThread : public base::SimpleThread { | 59 class ThreadSafetyStressThread : public base::SimpleThread { |
| 60 public: | 60 public: |
| 61 enum DispatcherOp { | 61 enum DispatcherOp { |
| 62 CLOSE = 0, | 62 CLOSE = 0, |
| (...skipping 27 matching lines...) Expand all Loading... |
| 90 waiter_.Init(); | 90 waiter_.Init(); |
| 91 switch(op_) { | 91 switch(op_) { |
| 92 case CLOSE: { | 92 case CLOSE: { |
| 93 MojoResult r = dispatcher_->Close(); | 93 MojoResult r = dispatcher_->Close(); |
| 94 EXPECT_TRUE(r == MOJO_RESULT_OK || r == MOJO_RESULT_INVALID_ARGUMENT) | 94 EXPECT_TRUE(r == MOJO_RESULT_OK || r == MOJO_RESULT_INVALID_ARGUMENT) |
| 95 << "Result: " << r; | 95 << "Result: " << r; |
| 96 break; | 96 break; |
| 97 } | 97 } |
| 98 case WRITE_MESSAGE: | 98 case WRITE_MESSAGE: |
| 99 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | 99 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, |
| 100 dispatcher_->WriteMessage(NULL, 0, NULL, 0, | 100 dispatcher_->WriteMessage(NULL, 0, NULL, |
| 101 MOJO_WRITE_MESSAGE_FLAG_NONE)); | 101 MOJO_WRITE_MESSAGE_FLAG_NONE)); |
| 102 break; | 102 break; |
| 103 case READ_MESSAGE: | 103 case READ_MESSAGE: |
| 104 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | 104 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, |
| 105 dispatcher_->ReadMessage(NULL, NULL, NULL, NULL, | 105 dispatcher_->ReadMessage(NULL, NULL, 0, NULL, |
| 106 MOJO_WRITE_MESSAGE_FLAG_NONE)); | 106 MOJO_WRITE_MESSAGE_FLAG_NONE)); |
| 107 break; | 107 break; |
| 108 case ADD_WAITER: { | 108 case ADD_WAITER: { |
| 109 MojoResult r = dispatcher_->AddWaiter(&waiter_, | 109 MojoResult r = dispatcher_->AddWaiter(&waiter_, |
| 110 MOJO_WAIT_FLAG_EVERYTHING, 0); | 110 MOJO_WAIT_FLAG_EVERYTHING, 0); |
| 111 EXPECT_TRUE(r == MOJO_RESULT_FAILED_PRECONDITION || | 111 EXPECT_TRUE(r == MOJO_RESULT_FAILED_PRECONDITION || |
| 112 r == MOJO_RESULT_INVALID_ARGUMENT); | 112 r == MOJO_RESULT_INVALID_ARGUMENT); |
| 113 break; | 113 break; |
| 114 } | 114 } |
| 115 case REMOVE_WAITER: | 115 case REMOVE_WAITER: |
| (...skipping 64 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 180 event.Signal(); // Kicks off real work on the threads. | 180 event.Signal(); // Kicks off real work on the threads. |
| 181 } // Joins all the threads. | 181 } // Joins all the threads. |
| 182 | 182 |
| 183 EXPECT_EQ(MOJO_RESULT_OK, d->Close()); | 183 EXPECT_EQ(MOJO_RESULT_OK, d->Close()); |
| 184 } | 184 } |
| 185 } | 185 } |
| 186 | 186 |
| 187 } // namespace | 187 } // namespace |
| 188 } // namespace system | 188 } // namespace system |
| 189 } // namespace mojo | 189 } // namespace mojo |
| OLD | NEW |