1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
# This file is part of python-sql. The COPYRIGHT file at the top level of
# this repository contains the full copyright notices and license terms.
import unittest
from sql import Delete, Table, With
class TestDelete(unittest.TestCase):
table = Table('t')
def test_delete1(self):
query = self.table.delete()
self.assertEqual(str(query), 'DELETE FROM "t"')
self.assertEqual(query.params, ())
def test_delete2(self):
query = self.table.delete(where=(self.table.c == 'foo'))
self.assertEqual(str(query), 'DELETE FROM "t" WHERE ("c" = %s)')
self.assertEqual(query.params, ('foo',))
def test_delete3(self):
t1 = Table('t1')
t2 = Table('t2')
query = t1.delete(where=(t1.c.in_(t2.select(t2.c))))
self.assertEqual(str(query),
'DELETE FROM "t1" WHERE ("c" IN ('
'SELECT "a"."c" FROM "t2" AS "a"))')
self.assertEqual(query.params, ())
def test_delete_invalid_table(self):
with self.assertRaises(ValueError):
Delete('foo')
def test_delete_invalid_where(self):
with self.assertRaises(ValueError):
self.table.delete(where='foo')
def test_delete_returning(self):
query = self.table.delete(returning=[self.table.c])
self.assertEqual(str(query), 'DELETE FROM "t" RETURNING "c"')
self.assertEqual(query.params, ())
def test_delet_returning_select(self):
query = self.table.delete(returning=[self.table.select()])
self.assertEqual(
str(query),
'DELETE FROM "t" RETURNING (SELECT * FROM "t")')
self.assertEqual(query.params, ())
def test_delete_invalid_returning(self):
with self.assertRaises(ValueError):
self.table.delete(returning='foo')
def test_with(self):
t1 = Table('t1')
w = With(query=t1.select(t1.c1))
query = self.table.delete(with_=[w],
where=self.table.c2.in_(w.select(w.c3)))
self.assertEqual(str(query),
'WITH "a" AS (SELECT "b"."c1" FROM "t1" AS "b") '
'DELETE FROM "t" WHERE '
'("c2" IN (SELECT "a"."c3" FROM "a" AS "a"))')
self.assertEqual(query.params, ())
|
