Merge pull request #75 from michalc/test/higher-code-coverage-in-tests

test: avoid iterating once in for loops to avoid partial coverage in tests

Author: michalc

test.py

@@ -233,8 +233,9 @@ def test_select_multi(self):
                     SELECT my_col_a FROM my_table;
                     SELECT my_col_a FROM my_table LIMIT 10;
                 '''):
-                    for row in rows:
-                        raise Exception('Just after iterating first row')
+                    rows_it = iter(rows)
+                    next(rows_it)
+                    raise Exception('Just after iterating first row')
 
         with self.assertRaisesRegex(Exception, 'Multiple open statements'):
             with sqlite_s3_query_multi('http://localhost:9000/my-bucket/my.db', get_credentials=lambda now: (
@@ -248,12 +249,13 @@ def test_select_multi(self):
                     SELECT my_col_a FROM my_table LIMIT 10;
                 '''))
                 columns_1, rows_1 = next(it)
-                for row in rows_1:
-                    break
+                rows_1_it = iter(rows_1)
+                next(rows_1_it)
 
                 columns_2, rows_2 = next(it)
-                for row in rows_2:
-                    raise Exception('Multiple open statements')
+                rows_2_it = iter(rows_2)
+                next(rows_2_it)
+                raise Exception('Multiple open statements')
 
         with self.assertRaisesRegex(Exception, 'Attempting to use finalized statement'):
             with sqlite_s3_query_multi('http://localhost:9000/my-bucket/my.db', get_credentials=lambda now: (