def setUp(self): connection = MagicMock(spec=Connection) cursor = MagicMock(spec=Cursor) connection.cursor.return_value = cursor cursor.__aenter__ = CoroutineMock(return_value=cursor) cursor.execute = CoroutineMock() self.accessor = DataAccessor(connection) self.connection = connection self.cursor = cursor
def session(): sess = MagicMock() mock_resp = MagicMock() mock_resp.json = CoroutineMock() mock_resp_gen = CoroutineMock(return_value=mock_resp) request_ctx_mgr = MagicMock() request_ctx_mgr.__aenter__ = mock_resp_gen request_ctx_mgr.__aexit__ = CoroutineMock() sess.get = Mock(return_value=request_ctx_mgr) sess.post = Mock(return_value=request_ctx_mgr) return sess
async def test_context_manager_mocked(self): """Test with the context manager mocked out""" mock = MagicMock() context = CoroutineMock() context.say_hello = CoroutineMock(return_value="mocked!") mock.__aenter__ = CoroutineMock(return_value=context) mock.__aexit__ = CoroutineMock() self.uut.context_manager = mock self.assertEqual(await self.uut.test_me(), "mocked!")
def inner(return_value=None, side_effect=None): stub = stub_connection(return_value=return_value, side_effect=side_effect) context_manager = MagicMock() context_manager.__aenter__ = CoroutineMock(return_value=stub) context_manager.__aexit__ = CoroutineMock() manager = MagicMock() manager.new_connection = Mock(return_value=context_manager) manager.connection_stub = stub return manager
def setUp(self): connection = MagicMock(spec=Connection) cursor = MagicMock(spec=Cursor) connection.cursor.return_value = cursor cursor.__aenter__ = CoroutineMock(return_value=cursor) votes = [('user_a', 'item_a', 0), ('user_a', 'item_b', 1), ('user_b', 'item_a', 1), ('user_b', 'item_b', 1)] vote_scale = (0, 1) cursor.fetchall = CoroutineMock(return_value=votes) cursor.fetchone = CoroutineMock(return_value=vote_scale) cursor.execute = CoroutineMock() self.accessor = DataAccessor(connection) self.connection = connection self.cursor = cursor
def setUp(self): connection = MagicMock(spec=Connection) cursor = MagicMock(spec=Cursor) connection.cursor.return_value = cursor cursor.__aenter__ = CoroutineMock(return_value=cursor) cursor.fetchmany = CoroutineMock() self.accessor = DataAccessor(connection) self.connection = connection self.cursor = cursor self.posts = [{ 'id': 'test_post0' }, { 'id': 'test_post1' }, { 'id': 'test_post2' }, { 'id': 'test_post3' }]