Use the new do() API (no default language).

James William Pye · James William Pye · commit fc6e9c88867e · 2010-02-10T01:55:21.000-07:00
diff --git a/postgresql/test/test_driver.py b/postgresql/test/test_driver.py
@@ -198,13 +198,15 @@
 		],
 	),
 	('timetz', [
+			# timetz retains the offset
 			datetime.time(10,10,0, tzinfo=FixedOffset(4000)),
 			datetime.time(10,10,0, tzinfo=FixedOffset(6000)),
 			datetime.time(10,10,0, tzinfo=FixedOffset(7000)),
 			datetime.time(10,10,0, tzinfo=FixedOffset(2000)),
 		],
 	),
 	('interval', [
+			# no months :(
 			datetime.timedelta(40, 10, 1234),
 			datetime.timedelta(0, 0),
 			datetime.timedelta(-100, 0),
@@ -1415,14 +1417,9 @@ def testDo(self):
 			return
 		if 'plpgsql' not in self.db.sys.languages():
 			self.db.execute("CREATE LANGUAGE plpgsql")
-		self.db.do(
-			"BEGIN CREATE TEMP TABLE do_tmp_table(i int, t text); END",
-			language = 'plpgsql')
+		self.db.do('plpgsql', "BEGIN CREATE TEMP TABLE do_tmp_table(i int, t text); END",)
 		self.failUnlessEqual(len(self.db.prepare("SELECT * FROM do_tmp_table")()), 0)
-		# now, with the default language.
-		self.db.settings["default_do_language"] = 'plpgsql'
-		self.db.do(
-			"BEGIN INSERT INTO do_tmp_table VALUES (100, 'foo'); END")
+		self.db.do('plpgsql', "BEGIN INSERT INTO do_tmp_table VALUES (100, 'foo'); END")
 		self.failUnlessEqual(len(self.db.prepare("SELECT * FROM do_tmp_table")()), 1)
 
 if __name__ == '__main__':
