[Sqlalchemy-commits] [5768] sqlalchemy/trunk/test/dialect/mssql.py: Added a few IDENTITY tests for
Brought to you by:
zzzeek
From: <co...@sq...> - 2009-01-31 21:20:10
|
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd"> <html xmlns="http://www.w3.org/1999/xhtml"> <head><meta http-equiv="content-type" content="text/html; charset=utf-8" /><style type="text/css"><!-- #msg dl { border: 1px #006 solid; background: #369; padding: 6px; color: #fff; } #msg dt { float: left; width: 6em; font-weight: bold; } #msg dt:after { content:':';} #msg dl, #msg dt, #msg ul, #msg li, #header, #footer { font-family: verdana,arial,helvetica,sans-serif; font-size: 10pt; } #msg dl a { font-weight: bold} #msg dl a:link { color:#fc3; } #msg dl a:active { color:#ff0; } #msg dl a:visited { color:#cc6; } h3 { font-family: verdana,arial,helvetica,sans-serif; font-size: 10pt; font-weight: bold; } #msg pre { overflow: auto; background: #ffc; border: 1px #fc0 solid; padding: 6px; } #msg ul, pre { overflow: auto; } #header, #footer { color: #fff; background: #636; border: 1px #300 solid; padding: 6px; } #patch { width: 100%; } #patch h4 {font-family: verdana,arial,helvetica,sans-serif;font-size:10pt;padding:8px;background:#369;color:#fff;margin:0;} #patch .propset h4, #patch .binary h4 {margin:0;} #patch pre {padding:0;line-height:1.2em;margin:0;} #patch .diff {width:100%;background:#eee;padding: 0 0 10px 0;overflow:auto;} #patch .propset .diff, #patch .binary .diff {padding:10px 0;} #patch span {display:block;padding:0 10px;} #patch .modfile, #patch .addfile, #patch .delfile, #patch .propset, #patch .binary, #patch .copfile {border:1px solid #ccc;margin:10px 0;} #patch ins {background:#dfd;text-decoration:none;display:block;padding:0 10px;} #patch del {background:#fdd;text-decoration:none;display:block;padding:0 10px;} #patch .lines, .info {color:#888;background:#fff;} --></style> <title>[5768] sqlalchemy/trunk/test/dialect/mssql.py: Added a few IDENTITY tests for mssql.</title> </head> <body> <div id="msg"> <dl> <dt>Revision</dt> <dd>5768</dd> <dt>Author</dt> <dd>empty</dd> <dt>Date</dt> <dd>2009-01-31 16:20:04 -0500 (Sat, 31 Jan 2009)</dd> </dl> <h3>Log Message</h3> <pre>Added a few IDENTITY tests for mssql.</pre> <h3>Modified Paths</h3> <ul> <li><a href="#sqlalchemytrunktestdialectmssqlpy">sqlalchemy/trunk/test/dialect/mssql.py</a></li> </ul> </div> <div id="patch"> <h3>Diff</h3> <a id="sqlalchemytrunktestdialectmssqlpy"></a> <div class="modfile"><h4>Modified: sqlalchemy/trunk/test/dialect/mssql.py (5767 => 5768)</h4> <pre class="diff"><span> <span class="info">--- sqlalchemy/trunk/test/dialect/mssql.py 2009-01-31 21:10:26 UTC (rev 5767) +++ sqlalchemy/trunk/test/dialect/mssql.py 2009-01-31 21:20:04 UTC (rev 5768) </span><span class="lines">@@ -125,6 +125,61 @@ </span><span class="cx"> self.assert_compile(func.current_date(), "GETDATE()") </span><span class="cx"> self.assert_compile(func.length(3), "LEN(:length_1)") </span><span class="cx"> </span><ins>+ +class IdentityInsertTest(TestBase, AssertsCompiledSQL): + __only_on__ = 'mssql' + __dialect__ = mssql.MSSQLDialect() + + def setUpAll(self): + global metadata, cattable + metadata = MetaData(testing.db) + + cattable = Table('cattable', metadata, + Column('id', Integer), + Column('description', String(50)), + PrimaryKeyConstraint('id', name='PK_cattable'), + ) + + def setUp(self): + metadata.create_all() + + def tearDown(self): + metadata.drop_all() + + def test_compiled(self): + self.assert_compile(cattable.insert().values(id=9, description='Python'), "INSERT INTO cattable (id, description) VALUES (:id, :description)") + + def test_execute(self): + cattable.insert().values(id=9, description='Python').execute() + + cats = cattable.select().order_by(cattable.c.id).execute() + self.assertEqual([(9, 'Python')], list(cats)) + + result = cattable.insert().values(description='PHP').execute() + self.assertEqual([10], result.last_inserted_ids()) + lastcat = cattable.select().order_by(desc(cattable.c.id)).execute() + self.assertEqual((10, 'PHP'), lastcat.fetchone()) + + def test_executemany(self): + cattable.insert().execute([ + {'id': 89, 'description': 'Python'}, + {'id': 8, 'description': 'Ruby'}, + {'id': 3, 'description': 'Perl'}, + {'id': 1, 'description': 'Java'}, + ]) + + cats = cattable.select().order_by(cattable.c.id).execute() + self.assertEqual([(1, 'Java'), (3, 'Perl'), (8, 'Ruby'), (89, 'Python')], list(cats)) + + cattable.insert().execute([ + {'description': 'PHP'}, + {'description': 'Smalltalk'}, + ]) + + lastcats = cattable.select().order_by(desc(cattable.c.id)).limit(2).execute() + self.assertEqual([(91, 'Smalltalk'), (90, 'PHP')], list(lastcats)) + + </ins><span class="cx"> class ReflectionTest(TestBase): </span><span class="cx"> __only_on__ = 'mssql' </span><span class="cx"> </span></span></pre> </div> </div> </body> </html> |