test_test.py

# Copyright (C) 2016-present the asyncpg authors and contributors
# <see AUTHORS file>
#
# This module is part of asyncpg and is released under
# the Apache 2.0 License: http://www.apache.org/licenses/LICENSE-2.0


importasyncio
importtypes
importunittest


fromasyncpgimport_testbaseastb


classBaseSimpleTestCase:

asyncdeftest_tests_zero_error(self):
awaitasyncio.sleep(0.01)
1/0


classTestTests(unittest.TestCase):

deftest_tests_fail_1(self):
SimpleTestCase=types.new_class('SimpleTestCase',
 (BaseSimpleTestCase, tb.TestCase))

suite=unittest.TestSuite()
suite.addTest(SimpleTestCase('test_tests_zero_error'))

result=unittest.TestResult()
suite.run(result)

self.assertIn('ZeroDivisionError', result.errors[0][1])


classTestHelpers(tb.TestCase):

asyncdeftest_tests_assertLoopErrorHandlerCalled_01(self):
withself.assertRaisesRegex(AssertionError, r'no message.*was logged'):
withself.assertLoopErrorHandlerCalled('aa'):
self.loop.call_exception_handler({'message': 'bb a bb'})

withself.assertLoopErrorHandlerCalled('aa'):
self.loop.call_exception_handler({'message': 'bbaabb'})