def create_stream_cursor(mock_session, mock_statement_result_execute,
mock_statement_result_fetch):
"""
Create a stream cursor with execute and fetch page statement results.
"""
mock_session.return_value = None
mock_session._fetch_page.return_value = mock_statement_result_fetch
stream_cursor = StreamCursor(mock_statement_result_execute, mock_session,
"1")
stream_cursor._value_holder_to_ion_value = MagicMock(
name='_value_holder_to_ion_value')
return stream_cursor
def test_value_holder_to_ion_value(self):
ion_value = 'IonValue'
value_holder = {'IonBinary': dumps(ion_value)}
result = StreamCursor._value_holder_to_ion_value(value_holder)
self.assertEqual(result, ion_value)