def test_fetchmany_wrapped_is_called_and_returned(self):
cursor = self.cursor
cursor.rowcount = 0
cursor.fetchmany.return_value = '__result__'
pin = Pin('pin_name', tracer=self.tracer)
traced_cursor = FetchTracedCursor(cursor, pin, {})
assert '__result__' == traced_cursor.fetchmany('arg_1',
kwarg1='kwarg1')
cursor.fetchmany.assert_called_once_with('arg_1', kwarg1='kwarg1')
def test_fetchmany_wrapped_is_called_and_returned(self):
cursor = self.cursor
cursor.rowcount = 0
cursor.fetchmany.return_value = "__result__"
pin = Pin("pin_name", tracer=self.tracer)
traced_cursor = FetchTracedCursor(cursor, pin, {})
assert "__result__" == traced_cursor.fetchmany("arg_1",
kwarg1="kwarg1")
cursor.fetchmany.assert_called_once_with("arg_1", kwarg1="kwarg1")
