Merge "Add __next__(), next() to ResultProxy"

1 files changed, 17 insertions, 0 deletions

diff --git a/test/sql/test_resultset.py b/test/sql/test_resultset.py
index 41092efe9..5c7108ca0 100644
--- a/test/sql/test_resultset.py
+++ b/test/sql/test_resultset.py
@@ -61,6 +61,23 @@ class ResultProxyTest(fixtures.TablesTest):
             rows.append(row)
         eq_(len(rows), 3)
 
+    def test_row_next(self):
+        users = self.tables.users
+
+        users.insert().execute(
+            {'user_id': 7, 'user_name': 'jack'},
+            {'user_id': 8, 'user_name': 'ed'},
+            {'user_id': 9, 'user_name': 'fred'},
+        )
+        r = users.select().execute()
+        rows = []
+        while True:
+            row = next(r, 'foo')
+            if row == 'foo':
+                break
+            rows.append(row)
+        eq_(len(rows), 3)
+
     @testing.requires.subqueries
     def test_anonymous_rows(self):
         users = self.tables.users