From f84456261ff7fa230cfc4cfc406adb277f130ccb Mon Sep 17 00:00:00 2001 From: Inada Naoki Date: Wed, 24 May 2023 00:23:24 +0900 Subject: [PATCH] Add test for iterator interface --- pymysql/tests/test_cursor.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/pymysql/tests/test_cursor.py b/pymysql/tests/test_cursor.py index 66d968df..16d297f6 100644 --- a/pymysql/tests/test_cursor.py +++ b/pymysql/tests/test_cursor.py @@ -25,6 +25,14 @@ def setUp(self): self.test_connection = pymysql.connect(**self.databases[0]) self.addCleanup(self.test_connection.close) + def test_cursor_is_iterator(self): + """Test that the cursor is an iterator""" + conn = self.test_connection + cursor = conn.cursor() + cursor.execute("select * from test") + self.assertEqual(cursor.__iter__(), cursor) + self.assertEqual(cursor.__next__(), ("row1",)) + def test_cleanup_rows_unbuffered(self): conn = self.test_connection cursor = conn.cursor(pymysql.cursors.SSCursor)