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 unittest 6 7 # pylint: disable=F0401 8 import mojo.embedder 9 from mojo import system 10 11 12 def _Increment(array): 13 def _Closure(): 14 array.append(0) 15 return _Closure 16 17 18 class RunLoopTest(unittest.TestCase): 19 20 def setUp(self): 21 mojo.embedder.Init() 22 23 def testRunLoop(self): 24 loop = system.RunLoop() 25 array = [] 26 for _ in xrange(10): 27 loop.PostDelayedTask(_Increment(array)) 28 loop.RunUntilIdle() 29 self.assertEquals(len(array), 10) 30 31 def testRunLoopWithException(self): 32 loop = system.RunLoop() 33 def Throw(): 34 raise Exception("error") 35 array = [] 36 loop.PostDelayedTask(Throw) 37 loop.PostDelayedTask(_Increment(array)) 38 with self.assertRaisesRegexp(Exception, '^error$'): 39 loop.Run() 40 self.assertEquals(len(array), 0) 41 loop.RunUntilIdle() 42 self.assertEquals(len(array), 1) 43