diff --git a/tests/test_app.py b/tests/test_app.py index 8f4f5912..35400931 100644 --- a/tests/test_app.py +++ b/tests/test_app.py @@ -607,9 +607,12 @@ class TestAppHandler(TestCase): with self.app.short_session(foo='bar') as s: short_session.assert_called_once() - self.assertEqual(len(short_session.call_args), 2) - self.assertEqual(short_session.call_args.kwargs['foo'], 'bar') - self.assertIsInstance(short_session.call_args.kwargs['factory'], partial) + # TODO: python 3.7 mock objects do not have attrs for + # args/kwargs, but once we drop that support we can + # use those instead of treating call_args as tuple + self.assertEqual(len(short_session.call_args[1]), 2) + self.assertEqual(short_session.call_args[1]['foo'], 'bar') + self.assertIsInstance(short_session.call_args[1]['factory'], partial) def test_cache_model(self): engine = sa.create_engine('sqlite://')