| 81 | con.close() |
| 82 | |
| 83 | def test_fetchone(self): |
| 84 | con = self._connect() |
| 85 | try: |
| 86 | cur = con.cursor() |
| 87 | |
| 88 | # cursor.fetchone should raise an Error if called before |
| 89 | # executing a select-type query |
| 90 | self.assertRaises(self.driver.Error, cur.fetchone) |
| 91 | |
| 92 | # cursor.fetchone should raise an Error if called after |
| 93 | # executing a query that cannot return rows |
| 94 | self.executeDDL1(cur) |
| 95 | # self.assertRaises(self.driver.Error,cur.fetchone) |
| 96 | |
| 97 | cur.execute("select name from %sbooze" % self.table_prefix) |
| 98 | self.assertEqual( |
| 99 | cur.fetchone(), |
| 100 | None, |
| 101 | "cursor.fetchone should return None if a query retrieves " "no rows", |
| 102 | ) |
| 103 | self.assertTrue(cur.rowcount in (-1, 0)) |
| 104 | |
| 105 | # cursor.fetchone should raise an Error if called after |
| 106 | # executing a query that cannot return rows |
| 107 | cur.execute( |
| 108 | "insert into %sbooze values ('Victoria Bitter')" % (self.table_prefix) |
| 109 | ) |
| 110 | # self.assertRaises(self.driver.Error,cur.fetchone) |
| 111 | |
| 112 | cur.execute("select name from %sbooze" % self.table_prefix) |
| 113 | r = cur.fetchone() |
| 114 | self.assertEqual( |
| 115 | len(r), 1, "cursor.fetchone should have retrieved a single row" |
| 116 | ) |
| 117 | self.assertEqual( |
| 118 | r[0], "Victoria Bitter", "cursor.fetchone retrieved incorrect data" |
| 119 | ) |
| 120 | # self.assertEqual( |
| 121 | # cur.fetchone(), |
| 122 | # None, |
| 123 | # "cursor.fetchone should return None if no more rows available", |
| 124 | # ) |
| 125 | self.assertTrue(cur.rowcount in (-1, 1)) |
| 126 | finally: |
| 127 | con.close() |
| 128 | |
| 129 | # Same complaint as for fetchall and fetchone |
| 130 | def test_rowcount(self): |