Пример #1
0
 def testCopyFromWithError(self):
     try:
         cursor = self.db.cursor()
         stream = BytesIO(b("f1Xf2\n\n1XY1Y\n"))
         cursor.execute(
             "COPY t1 (f1, f2) FROM STDIN WITH DELIMITER 'X' CSV HEADER "
             "QUOTE AS 'Y' FORCE NOT NULL f1",
             stream=stream)
         self.assertTrue(False, "Should have raised an exception")
     except:
         args_dict = {
             0: u('ERROR'),
             1: u('22P02'),
             2: u('invalid input syntax for integer: ""'),
             3: u('COPY t1, line 2, column f1: ""'),
             4: u('numutils.c'),
             6: u('pg_atoi'),
             7: u(''),
             8: u('')
         }
         args = exc_info()[1].args
         for k, v in iteritems(args_dict):
             self.assertEqual(args[k], v)
     finally:
         cursor.close()
Пример #2
0
 def testCopyFromWithError(self):
     try:
         cursor = self.db.cursor()
         stream = BytesIO(b("f1Xf2\n\n1XY1Y\n"))
         cursor.execute(
             "COPY t1 (f1, f2) FROM STDIN WITH DELIMITER 'X' CSV HEADER " "QUOTE AS 'Y' FORCE NOT NULL f1",
             stream=stream,
         )
         self.assertTrue(False, "Should have raised an exception")
     except:
         args_dict = {
             0: u("ERROR"),
             1: u("22P02"),
             2: u('invalid input syntax for integer: ""'),
             3: u('COPY t1, line 2, column f1: ""'),
             4: u("numutils.c"),
             6: u("pg_atoi"),
             7: u(""),
             8: u(""),
         }
         args = exc_info()[1].args
         for k, v in iteritems(args_dict):
             self.assertEqual(args[k], v)
     finally:
         cursor.close()
Пример #3
0
 def testUnicodeQuery(self):
     try:
         cursor = db.cursor()
         cursor.execute(
             u("CREATE TEMPORARY TABLE \u043c\u0435\u0441\u0442\u043e "
               "(\u0438\u043c\u044f VARCHAR(50), "
               "\u0430\u0434\u0440\u0435\u0441 VARCHAR(250))"))
     finally:
         cursor.close()
         db.commit()
Пример #4
0
 def testUnicodeQuery(self):
     try:
         cursor = self.db.cursor()
         cursor.execute(
             u(
                 "CREATE TEMPORARY TABLE \u043c\u0435\u0441\u0442\u043e "
                 "(\u0438\u043c\u044f VARCHAR(50), "
                 "\u0430\u0434\u0440\u0435\u0441 VARCHAR(250))"))
     finally:
         cursor.close()
         self.db.commit()