mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-10 10:50:37 -04:00
Mike Bayer
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
from sqlalchemy import *
|
|
from sqlalchemy.test import *
|
|
|
|
|
|
class FoundRowsTest(TestBase, AssertsExecutionResults):
|
|
"""tests rowcount functionality"""
|
|
@classmethod
|
|
def setup_class(cls):
|
|
metadata = MetaData(testing.db)
|
|
|
|
global employees_table
|
|
|
|
employees_table = Table('employees', metadata,
|
|
Column('employee_id', Integer, Sequence('employee_id_seq', optional=True), primary_key=True),
|
|
Column('name', String(50)),
|
|
Column('department', String(1)),
|
|
)
|
|
employees_table.create()
|
|
|
|
def setup(self):
|
|
global data
|
|
data = [ ('Angela', 'A'),
|
|
('Andrew', 'A'),
|
|
('Anand', 'A'),
|
|
('Bob', 'B'),
|
|
('Bobette', 'B'),
|
|
('Buffy', 'B'),
|
|
('Charlie', 'C'),
|
|
('Cynthia', 'C'),
|
|
('Chris', 'C') ]
|
|
|
|
i = employees_table.insert()
|
|
i.execute(*[{'name':n, 'department':d} for n, d in data])
|
|
def teardown(self):
|
|
employees_table.delete().execute()
|
|
|
|
@classmethod
|
|
def teardown_class(cls):
|
|
employees_table.drop()
|
|
|
|
def testbasic(self):
|
|
s = employees_table.select()
|
|
r = s.execute().fetchall()
|
|
|
|
assert len(r) == len(data)
|
|
|
|
def test_update_rowcount1(self):
|
|
# WHERE matches 3, 3 rows changed
|
|
department = employees_table.c.department
|
|
r = employees_table.update(department=='C').execute(department='Z')
|
|
print "expecting 3, dialect reports %s" % r.rowcount
|
|
if testing.db.dialect.supports_sane_rowcount:
|
|
assert r.rowcount == 3
|
|
|
|
def test_update_rowcount2(self):
|
|
# WHERE matches 3, 0 rows changed
|
|
department = employees_table.c.department
|
|
r = employees_table.update(department=='C').execute(department='C')
|
|
print "expecting 3, dialect reports %s" % r.rowcount
|
|
if testing.db.dialect.supports_sane_rowcount:
|
|
assert r.rowcount == 3
|
|
|
|
def test_delete_rowcount(self):
|
|
# WHERE matches 3, 3 rows deleted
|
|
department = employees_table.c.department
|
|
r = employees_table.delete(department=='C').execute()
|
|
print "expecting 3, dialect reports %s" % r.rowcount
|
|
if testing.db.dialect.supports_sane_rowcount:
|
|
assert r.rowcount == 3
|
|
|