| OLD | NEW |
| (Empty) |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import random | |
| 6 import time | |
| 7 | |
| 8 import mojo_unittest | |
| 9 | |
| 10 # pylint: disable=E0611 | |
| 11 from mojo import system | |
| 12 | |
| 13 DATA_SIZE = 1024 | |
| 14 | |
| 15 | |
| 16 def _GetRandomBuffer(size): | |
| 17 random.seed(size) | |
| 18 return bytearray(''.join(chr(random.randint(0, 255)) for i in xrange(size))) | |
| 19 | |
| 20 | |
| 21 class CoreTest(mojo_unittest.MojoTestCase): | |
| 22 | |
| 23 def testResults(self): | |
| 24 self.assertEquals(system.RESULT_OK, 0) | |
| 25 self.assertLess(system.RESULT_CANCELLED, 0) | |
| 26 self.assertLess(system.RESULT_UNKNOWN, 0) | |
| 27 self.assertLess(system.RESULT_INVALID_ARGUMENT, 0) | |
| 28 self.assertLess(system.RESULT_DEADLINE_EXCEEDED, 0) | |
| 29 self.assertLess(system.RESULT_NOT_FOUND, 0) | |
| 30 self.assertLess(system.RESULT_ALREADY_EXISTS, 0) | |
| 31 self.assertLess(system.RESULT_PERMISSION_DENIED, 0) | |
| 32 self.assertLess(system.RESULT_RESOURCE_EXHAUSTED, 0) | |
| 33 self.assertLess(system.RESULT_FAILED_PRECONDITION, 0) | |
| 34 self.assertLess(system.RESULT_ABORTED, 0) | |
| 35 self.assertLess(system.RESULT_OUT_OF_RANGE, 0) | |
| 36 self.assertLess(system.RESULT_UNIMPLEMENTED, 0) | |
| 37 self.assertLess(system.RESULT_INTERNAL, 0) | |
| 38 self.assertLess(system.RESULT_UNAVAILABLE, 0) | |
| 39 self.assertLess(system.RESULT_DATA_LOSS, 0) | |
| 40 self.assertLess(system.RESULT_BUSY, 0) | |
| 41 self.assertLess(system.RESULT_SHOULD_WAIT, 0) | |
| 42 | |
| 43 def testConstants(self): | |
| 44 self.assertGreaterEqual(system.DEADLINE_INDEFINITE, 0) | |
| 45 self.assertGreaterEqual(system.HANDLE_SIGNAL_NONE, 0) | |
| 46 self.assertGreaterEqual(system.HANDLE_SIGNAL_READABLE, 0) | |
| 47 self.assertGreaterEqual(system.HANDLE_SIGNAL_WRITABLE, 0) | |
| 48 self.assertGreaterEqual(system.WRITE_MESSAGE_FLAG_NONE, 0) | |
| 49 self.assertGreaterEqual(system.READ_MESSAGE_FLAG_NONE, 0) | |
| 50 self.assertGreaterEqual(system.READ_MESSAGE_FLAG_MAY_DISCARD, 0) | |
| 51 self.assertGreaterEqual(system.WRITE_DATA_FLAG_NONE, 0) | |
| 52 self.assertGreaterEqual(system.WRITE_DATA_FLAG_ALL_OR_NONE, 0) | |
| 53 self.assertGreaterEqual(system.READ_DATA_FLAG_NONE, 0) | |
| 54 self.assertGreaterEqual(system.READ_DATA_FLAG_ALL_OR_NONE, 0) | |
| 55 self.assertGreaterEqual(system.READ_DATA_FLAG_DISCARD, 0) | |
| 56 self.assertGreaterEqual(system.READ_DATA_FLAG_QUERY, 0) | |
| 57 self.assertGreaterEqual(system.MAP_BUFFER_FLAG_NONE, 0) | |
| 58 | |
| 59 def testGetTimeTicksNow(self): | |
| 60 v1 = system.GetTimeTicksNow() | |
| 61 time.sleep(1e-3) | |
| 62 v2 = system.GetTimeTicksNow() | |
| 63 self.assertGreater(v1, 0) | |
| 64 self.assertGreater(v2, v1 + 1e2) | |
| 65 self.assertLess(v2, v1 + 1e5) | |
| 66 | |
| 67 def _testHandlesCreation(self, *args): | |
| 68 for handle in args: | |
| 69 self.assertTrue(handle.IsValid()) | |
| 70 handle.Close() | |
| 71 self.assertFalse(handle.IsValid()) | |
| 72 | |
| 73 def _TestMessageHandleCreation(self, handles): | |
| 74 self._testHandlesCreation(handles.handle0, handles.handle1) | |
| 75 | |
| 76 def testCreateMessagePipe(self): | |
| 77 self._TestMessageHandleCreation(system.MessagePipe()) | |
| 78 | |
| 79 def testCreateMessagePipeWithNoneOptions(self): | |
| 80 self._TestMessageHandleCreation(system.MessagePipe(None)) | |
| 81 | |
| 82 def testCreateMessagePipeWithOptions(self): | |
| 83 self._TestMessageHandleCreation( | |
| 84 system.MessagePipe(system.CreateMessagePipeOptions())) | |
| 85 | |
| 86 def testWaitOverMessagePipe(self): | |
| 87 handles = system.MessagePipe() | |
| 88 handle = handles.handle0 | |
| 89 | |
| 90 self.assertEquals(system.RESULT_OK, handle.Wait( | |
| 91 system.HANDLE_SIGNAL_WRITABLE, system.DEADLINE_INDEFINITE)) | |
| 92 self.assertEquals(system.RESULT_DEADLINE_EXCEEDED, | |
| 93 handle.Wait(system.HANDLE_SIGNAL_READABLE, 0)) | |
| 94 | |
| 95 handles.handle1.WriteMessage() | |
| 96 | |
| 97 self.assertEquals( | |
| 98 system.RESULT_OK, | |
| 99 handle.Wait( | |
| 100 system.HANDLE_SIGNAL_READABLE, | |
| 101 system.DEADLINE_INDEFINITE)) | |
| 102 | |
| 103 def testWaitOverManyMessagePipe(self): | |
| 104 handles = system.MessagePipe() | |
| 105 handle0 = handles.handle0 | |
| 106 handle1 = handles.handle1 | |
| 107 | |
| 108 self.assertEquals( | |
| 109 0, | |
| 110 system.WaitMany( | |
| 111 [(handle0, system.HANDLE_SIGNAL_WRITABLE), | |
| 112 (handle1, system.HANDLE_SIGNAL_WRITABLE)], | |
| 113 system.DEADLINE_INDEFINITE)) | |
| 114 self.assertEquals( | |
| 115 system.RESULT_DEADLINE_EXCEEDED, | |
| 116 system.WaitMany( | |
| 117 [(handle0, system.HANDLE_SIGNAL_READABLE), | |
| 118 (handle1, system.HANDLE_SIGNAL_READABLE)], 0)) | |
| 119 | |
| 120 handle0.WriteMessage() | |
| 121 | |
| 122 self.assertEquals( | |
| 123 1, | |
| 124 system.WaitMany( | |
| 125 [(handle0, system.HANDLE_SIGNAL_READABLE), | |
| 126 (handle1, system.HANDLE_SIGNAL_READABLE)], | |
| 127 system.DEADLINE_INDEFINITE)) | |
| 128 | |
| 129 def testSendBytesOverMessagePipe(self): | |
| 130 handles = system.MessagePipe() | |
| 131 data = _GetRandomBuffer(DATA_SIZE) | |
| 132 handles.handle0.WriteMessage(data) | |
| 133 (res, buffers, next_message) = handles.handle1.ReadMessage() | |
| 134 self.assertEquals(system.RESULT_RESOURCE_EXHAUSTED, res) | |
| 135 self.assertEquals(None, buffers) | |
| 136 self.assertEquals((DATA_SIZE, 0), next_message) | |
| 137 result = bytearray(DATA_SIZE) | |
| 138 (res, buffers, next_message) = handles.handle1.ReadMessage(result) | |
| 139 self.assertEquals(system.RESULT_OK, res) | |
| 140 self.assertEquals(None, next_message) | |
| 141 self.assertEquals((data, []), buffers) | |
| 142 | |
| 143 def testSendEmptyDataOverMessagePipe(self): | |
| 144 handles = system.MessagePipe() | |
| 145 handles.handle0.WriteMessage(None) | |
| 146 (res, buffers, next_message) = handles.handle1.ReadMessage() | |
| 147 | |
| 148 self.assertEquals(system.RESULT_OK, res) | |
| 149 self.assertEquals(None, next_message) | |
| 150 self.assertEquals((None, []), buffers) | |
| 151 | |
| 152 def testSendHandleOverMessagePipe(self): | |
| 153 handles = system.MessagePipe() | |
| 154 handles_to_send = system.MessagePipe() | |
| 155 handles.handle0.WriteMessage(handles=[handles_to_send.handle0, | |
| 156 handles_to_send.handle1]) | |
| 157 (res, buffers, next_message) = handles.handle1.ReadMessage( | |
| 158 max_number_of_handles=2) | |
| 159 | |
| 160 self.assertFalse(handles_to_send.handle0.IsValid()) | |
| 161 self.assertFalse(handles_to_send.handle1.IsValid()) | |
| 162 self.assertEquals(system.RESULT_OK, res) | |
| 163 self.assertEquals(None, next_message) | |
| 164 self.assertEquals(None, buffers[0]) | |
| 165 self.assertEquals(2, len(buffers[1])) | |
| 166 | |
| 167 handles = buffers[1] | |
| 168 for handle in handles: | |
| 169 self.assertTrue(handle.IsValid()) | |
| 170 (res, buffers, next_message) = handle.ReadMessage() | |
| 171 self.assertEquals(system.RESULT_SHOULD_WAIT, res) | |
| 172 | |
| 173 for handle in handles: | |
| 174 handle.WriteMessage() | |
| 175 | |
| 176 for handle in handles: | |
| 177 (res, buffers, next_message) = handle.ReadMessage() | |
| 178 self.assertEquals(system.RESULT_OK, res) | |
| 179 | |
| 180 def _TestDataHandleCreation(self, handles): | |
| 181 self._testHandlesCreation( | |
| 182 handles.producer_handle, handles.consumer_handle) | |
| 183 | |
| 184 def testCreateDataPipe(self): | |
| 185 self._TestDataHandleCreation(system.DataPipe()) | |
| 186 | |
| 187 def testCreateDataPipeWithNoneOptions(self): | |
| 188 self._TestDataHandleCreation(system.DataPipe(None)) | |
| 189 | |
| 190 def testCreateDataPipeWithDefaultOptions(self): | |
| 191 self._TestDataHandleCreation( | |
| 192 system.DataPipe(system.CreateDataPipeOptions())) | |
| 193 | |
| 194 def testCreateDataPipeWithDiscardFlag(self): | |
| 195 options = system.CreateDataPipeOptions() | |
| 196 options.flags = system.CreateDataPipeOptions.FLAG_MAY_DISCARD | |
| 197 self._TestDataHandleCreation(system.DataPipe(options)) | |
| 198 | |
| 199 def testCreateDataPipeWithElementSize(self): | |
| 200 options = system.CreateDataPipeOptions() | |
| 201 options.element_num_bytes = 5 | |
| 202 self._TestDataHandleCreation(system.DataPipe(options)) | |
| 203 | |
| 204 def testCreateDataPipeWithCapacity(self): | |
| 205 options = system.CreateDataPipeOptions() | |
| 206 options.element_capacity_num_bytes = DATA_SIZE | |
| 207 self._TestDataHandleCreation(system.DataPipe(options)) | |
| 208 | |
| 209 def testCreateDataPipeWithIncorrectParameters(self): | |
| 210 options = system.CreateDataPipeOptions() | |
| 211 options.element_num_bytes = 5 | |
| 212 options.capacity_num_bytes = DATA_SIZE | |
| 213 with self.assertRaises(system.MojoException) as cm: | |
| 214 self._TestDataHandleCreation(system.DataPipe(options)) | |
| 215 self.assertEquals(system.RESULT_INVALID_ARGUMENT, cm.exception.mojo_result) | |
| 216 | |
| 217 def testSendEmptyDataOverDataPipe(self): | |
| 218 pipes = system.DataPipe() | |
| 219 self.assertEquals((system.RESULT_OK, 0), pipes.producer_handle.WriteData()) | |
| 220 self.assertEquals( | |
| 221 (system.RESULT_OK, None), pipes.consumer_handle.ReadData()) | |
| 222 | |
| 223 def testSendDataOverDataPipe(self): | |
| 224 pipes = system.DataPipe() | |
| 225 data = _GetRandomBuffer(DATA_SIZE) | |
| 226 self.assertEquals((system.RESULT_OK, DATA_SIZE), | |
| 227 pipes.producer_handle.WriteData(data)) | |
| 228 self.assertEquals((system.RESULT_OK, data), | |
| 229 pipes.consumer_handle.ReadData(bytearray(DATA_SIZE))) | |
| 230 | |
| 231 def testTwoPhaseWriteOnDataPipe(self): | |
| 232 pipes = system.DataPipe() | |
| 233 (res, buf) = pipes.producer_handle.BeginWriteData(DATA_SIZE) | |
| 234 self.assertEquals(system.RESULT_OK, res) | |
| 235 self.assertGreaterEqual(len(buf.buffer), DATA_SIZE) | |
| 236 data = _GetRandomBuffer(DATA_SIZE) | |
| 237 buf.buffer[0:DATA_SIZE] = data | |
| 238 self.assertEquals(system.RESULT_OK, buf.End(DATA_SIZE)) | |
| 239 self.assertEquals((system.RESULT_OK, data), | |
| 240 pipes.consumer_handle.ReadData(bytearray(DATA_SIZE))) | |
| 241 | |
| 242 def testTwoPhaseReadOnDataPipe(self): | |
| 243 pipes = system.DataPipe() | |
| 244 data = _GetRandomBuffer(DATA_SIZE) | |
| 245 self.assertEquals((system.RESULT_OK, DATA_SIZE), | |
| 246 pipes.producer_handle.WriteData(data)) | |
| 247 (res, buf) = pipes.consumer_handle.BeginReadData() | |
| 248 self.assertEquals(system.RESULT_OK, res) | |
| 249 self.assertEquals(DATA_SIZE, len(buf.buffer)) | |
| 250 self.assertEquals(data, buf.buffer) | |
| 251 self.assertEquals(system.RESULT_OK, buf.End(DATA_SIZE)) | |
| 252 | |
| 253 def testCreateSharedBuffer(self): | |
| 254 self._testHandlesCreation(system.CreateSharedBuffer(DATA_SIZE)) | |
| 255 | |
| 256 def testCreateSharedBufferWithNoneOptions(self): | |
| 257 self._testHandlesCreation(system.CreateSharedBuffer(DATA_SIZE, None)) | |
| 258 | |
| 259 def testCreateSharedBufferWithDefaultOptions(self): | |
| 260 self._testHandlesCreation( | |
| 261 system.CreateSharedBuffer( | |
| 262 DATA_SIZE, | |
| 263 system.CreateSharedBufferOptions())) | |
| 264 | |
| 265 def testDuplicateSharedBuffer(self): | |
| 266 handle = system.CreateSharedBuffer(DATA_SIZE) | |
| 267 self._testHandlesCreation(handle.Duplicate()) | |
| 268 | |
| 269 def testDuplicateSharedBufferWithNoneOptions(self): | |
| 270 handle = system.CreateSharedBuffer(DATA_SIZE) | |
| 271 self._testHandlesCreation(handle.Duplicate(None)) | |
| 272 | |
| 273 def testDuplicateSharedBufferWithDefaultOptions(self): | |
| 274 handle = system.CreateSharedBuffer(DATA_SIZE) | |
| 275 self._testHandlesCreation( | |
| 276 handle.Duplicate(system.DuplicateSharedBufferOptions())) | |
| 277 | |
| 278 def testSendBytesOverSharedBuffer(self): | |
| 279 handle = system.CreateSharedBuffer(DATA_SIZE) | |
| 280 duplicated = handle.Duplicate() | |
| 281 data = _GetRandomBuffer(DATA_SIZE) | |
| 282 (res1, buf1) = handle.Map(0, DATA_SIZE) | |
| 283 (res2, buf2) = duplicated.Map(0, DATA_SIZE) | |
| 284 self.assertEquals(system.RESULT_OK, res1) | |
| 285 self.assertEquals(system.RESULT_OK, res2) | |
| 286 self.assertEquals(DATA_SIZE, len(buf1.buffer)) | |
| 287 self.assertEquals(DATA_SIZE, len(buf2.buffer)) | |
| 288 self.assertEquals(buf1.buffer, buf2.buffer) | |
| 289 | |
| 290 buf1.buffer[:] = data | |
| 291 self.assertEquals(data, buf1.buffer) | |
| 292 self.assertEquals(data, buf2.buffer) | |
| 293 self.assertEquals(buf1.buffer, buf2.buffer) | |
| OLD | NEW |