| Index: mojo/python/tests/system_unittest.py
|
| diff --git a/mojo/python/tests/system_unittest.py b/mojo/python/tests/system_unittest.py
|
| index 2f0e36a3c19520366f8d0e22624ad08487ab3819..991dcba2cf0d64d05f7df128eb53f6e5d47e10a6 100644
|
| --- a/mojo/python/tests/system_unittest.py
|
| +++ b/mojo/python/tests/system_unittest.py
|
| @@ -3,12 +3,11 @@
|
| # found in the LICENSE file.
|
|
|
| import random
|
| -import sys
|
| import time
|
| -import unittest
|
| +
|
| +import mojo_unittest
|
|
|
| # pylint: disable=F0401
|
| -import mojo.embedder
|
| from mojo import system
|
|
|
| DATA_SIZE = 1024
|
| @@ -19,13 +18,7 @@ def _GetRandomBuffer(size):
|
| return bytearray(''.join(chr(random.randint(0, 255)) for i in xrange(size)))
|
|
|
|
|
| -class BaseMojoTest(unittest.TestCase):
|
| -
|
| - def setUp(self):
|
| - mojo.embedder.Init()
|
| -
|
| -
|
| -class CoreTest(BaseMojoTest):
|
| +class CoreTest(mojo_unittest.MojoTestCase):
|
|
|
| def testResults(self):
|
| self.assertEquals(system.RESULT_OK, 0)
|
| @@ -300,11 +293,3 @@ class CoreTest(BaseMojoTest):
|
| self.assertEquals(data, buf1.buffer)
|
| self.assertEquals(data, buf2.buffer)
|
| self.assertEquals(buf1.buffer, buf2.buffer)
|
| -
|
| -
|
| -if __name__ == '__main__':
|
| - suite = unittest.TestLoader().loadTestsFromTestCase(CoreTest)
|
| - test_results = unittest.TextTestRunner(verbosity=0).run(suite)
|
| - if not test_results.wasSuccessful():
|
| - sys.exit(1)
|
| - sys.exit(0)
|
|
|