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)