diff options
Diffstat (limited to 'test/ext/test_mutable.py')
| -rw-r--r-- | test/ext/test_mutable.py | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/test/ext/test_mutable.py b/test/ext/test_mutable.py index ed97a0d92..602ff911a 100644 --- a/test/ext/test_mutable.py +++ b/test/ext/test_mutable.py @@ -4,7 +4,7 @@ from sqlalchemy.orm import mapper, Session, composite from sqlalchemy.orm.mapper import Mapper from sqlalchemy.orm.instrumentation import ClassManager from sqlalchemy.testing.schema import Table, Column -from sqlalchemy.testing import eq_, assert_raises_message +from sqlalchemy.testing import eq_, assert_raises_message, assert_raises from sqlalchemy.testing.util import picklers from sqlalchemy.testing import fixtures from sqlalchemy.ext.mutable import MutableComposite @@ -146,6 +146,21 @@ class _MutableDictTestBase(_MutableDictTestFixture): eq_(f1.data.pop('a'), 'b') sess.commit() + assert_raises(KeyError, f1.data.pop, 'g') + + eq_(f1.data, {'c': 'd'}) + + def test_pop_default(self): + sess = Session() + + f1 = Foo(data={'a': 'b', 'c': 'd'}) + sess.add(f1) + sess.commit() + + eq_(f1.data.pop('a', 'q'), 'b') + eq_(f1.data.pop('a', 'q'), 'q') + sess.commit() + eq_(f1.data, {'c': 'd'}) def test_popitem(self): |
