mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-28 11:35:19 -04:00
Mike Bayer
70be7312f1
Changed the name of the ``.values`` attribute of the new MySQL INSERT..ON DUPLICATE KEY UPDATE construct to ``.inserted``, as :class:`.Insert` already has a method called :meth:`.Insert.values`. The ``.inserted`` attribute ultimately renders the MySQL ``VALUES()`` function. Change-Id: I8da8e30a3077698385a4b77e2c2032e2d1ff10b2 Fixes: #4072
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
from sqlalchemy.testing.assertions import eq_, assert_raises
|
|
from sqlalchemy.testing import fixtures
|
|
from sqlalchemy import testing
|
|
from sqlalchemy.dialects.mysql import insert
|
|
from sqlalchemy import Table, Column, Integer, String
|
|
|
|
|
|
class OnDuplicateTest(fixtures.TablesTest):
|
|
__only_on__ = 'mysql',
|
|
__backend__ = True
|
|
run_define_tables = 'each'
|
|
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
Table(
|
|
'foos', metadata,
|
|
Column('id', Integer, primary_key=True, autoincrement=True),
|
|
Column('bar', String(10)),
|
|
Column('baz', String(10)),
|
|
)
|
|
|
|
def test_bad_args(self):
|
|
assert_raises(
|
|
ValueError,
|
|
insert(self.tables.foos, values={}).on_duplicate_key_update
|
|
)
|
|
|
|
def test_on_duplicate_key_update(self):
|
|
foos = self.tables.foos
|
|
with testing.db.connect() as conn:
|
|
conn.execute(insert(foos, dict(id=1, bar='b', baz='bz')))
|
|
stmt = insert(foos).values(
|
|
[dict(id=1, bar='ab'), dict(id=2, bar='b')])
|
|
stmt = stmt.on_duplicate_key_update(bar=stmt.inserted.bar)
|
|
result = conn.execute(stmt)
|
|
eq_(result.inserted_primary_key, [2])
|
|
eq_(
|
|
conn.execute(foos.select().where(foos.c.id == 1)).fetchall(),
|
|
[(1, 'ab', 'bz')]
|
|
)
|
|
|
|
def test_last_inserted_id(self):
|
|
foos = self.tables.foos
|
|
with testing.db.connect() as conn:
|
|
stmt = insert(foos).values({"bar": "b", "baz": "bz"})
|
|
result = conn.execute(
|
|
stmt.on_duplicate_key_update(
|
|
bar=stmt.inserted.bar, baz="newbz")
|
|
)
|
|
eq_(result.inserted_primary_key, [1])
|
|
|
|
stmt = insert(foos).values({"id": 1, "bar": "b", "baz": "bz"})
|
|
result = conn.execute(
|
|
stmt.on_duplicate_key_update(
|
|
bar=stmt.inserted.bar, baz="newbz")
|
|
)
|
|
eq_(result.inserted_primary_key, [1])
|
|
|
|
|
|
|