diff options
Diffstat (limited to 'test/orm/test_validators.py')
| -rw-r--r-- | test/orm/test_validators.py | 333 |
1 files changed, 183 insertions, 150 deletions
diff --git a/test/orm/test_validators.py b/test/orm/test_validators.py index cbbf9f7a8..923775a7b 100644 --- a/test/orm/test_validators.py +++ b/test/orm/test_validators.py @@ -1,6 +1,11 @@ from test.orm import _fixtures -from sqlalchemy.testing import fixtures, assert_raises, eq_, ne_, \ - assert_raises_message +from sqlalchemy.testing import ( + fixtures, + assert_raises, + eq_, + ne_, + assert_raises_message, +) from sqlalchemy.orm import mapper, Session, validates, relationship from sqlalchemy.orm import collections from sqlalchemy.testing.mock import Mock, call @@ -13,113 +18,114 @@ class ValidatorTest(_fixtures.FixtureTest): canary = Mock() class User(fixtures.ComparableEntity): - @validates('name') + @validates("name") def validate_name(self, key, name): canary(key, name) - ne_(name, 'fred') - return name + ' modified' + ne_(name, "fred") + return name + " modified" mapper(User, users) sess = Session() - u1 = User(name='ed') - eq_(u1.name, 'ed modified') + u1 = User(name="ed") + eq_(u1.name, "ed modified") assert_raises(AssertionError, setattr, u1, "name", "fred") - eq_(u1.name, 'ed modified') - eq_(canary.mock_calls, [call('name', 'ed'), call('name', 'fred')]) + eq_(u1.name, "ed modified") + eq_(canary.mock_calls, [call("name", "ed"), call("name", "fred")]) sess.add(u1) sess.commit() eq_( - sess.query(User).filter_by(name='ed modified').one(), - User(name='ed') + sess.query(User).filter_by(name="ed modified").one(), + User(name="ed"), ) def test_collection(self): - users, addresses, Address = (self.tables.users, - self.tables.addresses, - self.classes.Address) + users, addresses, Address = ( + self.tables.users, + self.tables.addresses, + self.classes.Address, + ) canary = Mock() class User(fixtures.ComparableEntity): - @validates('addresses') + @validates("addresses") def validate_address(self, key, ad): canary(key, ad) - assert '@' in ad.email_address + assert "@" in ad.email_address return ad - mapper(User, users, properties={ - 'addresses': relationship(Address)} - ) + mapper(User, users, properties={"addresses": relationship(Address)}) mapper(Address, addresses) sess = Session() - u1 = User(name='edward') - a0 = Address(email_address='noemail') + u1 = User(name="edward") + a0 = Address(email_address="noemail") assert_raises(AssertionError, u1.addresses.append, a0) - a1 = Address(id=15, email_address='foo@bar.com') + a1 = Address(id=15, email_address="foo@bar.com") u1.addresses.append(a1) - eq_(canary.mock_calls, [call('addresses', a0), call('addresses', a1)]) + eq_(canary.mock_calls, [call("addresses", a0), call("addresses", a1)]) sess.add(u1) sess.commit() eq_( - sess.query(User).filter_by(name='edward').one(), - User(name='edward', addresses=[ - Address(email_address='foo@bar.com')]) + sess.query(User).filter_by(name="edward").one(), + User( + name="edward", addresses=[Address(email_address="foo@bar.com")] + ), ) def test_validators_dict(self): - users, addresses, Address = (self.tables.users, - self.tables.addresses, - self.classes.Address) + users, addresses, Address = ( + self.tables.users, + self.tables.addresses, + self.classes.Address, + ) class User(fixtures.ComparableEntity): - - @validates('name') + @validates("name") def validate_name(self, key, name): - ne_(name, 'fred') - return name + ' modified' + ne_(name, "fred") + return name + " modified" - @validates('addresses') + @validates("addresses") def validate_address(self, key, ad): - assert '@' in ad.email_address + assert "@" in ad.email_address return ad def simple_function(self, key, value): return key, value - u_m = mapper(User, users, properties={ - 'addresses': relationship(Address)}) + u_m = mapper( + User, users, properties={"addresses": relationship(Address)} + ) mapper(Address, addresses) eq_( dict((k, v[0].__name__) for k, v in list(u_m.validators.items())), - {'name': 'validate_name', - 'addresses': 'validate_address'} + {"name": "validate_name", "addresses": "validate_address"}, ) def test_validator_w_removes(self): - users, addresses, Address = (self.tables.users, - self.tables.addresses, - self.classes.Address) + users, addresses, Address = ( + self.tables.users, + self.tables.addresses, + self.classes.Address, + ) canary = Mock() class User(fixtures.ComparableEntity): - - @validates('name', include_removes=True) + @validates("name", include_removes=True) def validate_name(self, key, item, remove): canary(key, item, remove) return item - @validates('addresses', include_removes=True) + @validates("addresses", include_removes=True) def validate_address(self, key, item, remove): canary(key, item, remove) return item - mapper(User, users, properties={ - 'addresses': relationship(Address) - }) + mapper(User, users, properties={"addresses": relationship(Address)}) mapper(Address, addresses) u1 = User() @@ -132,33 +138,38 @@ class ValidatorTest(_fixtures.FixtureTest): u1.addresses = [a1, a2] u1.addresses = [a2, a3] - eq_(canary.mock_calls, [ - call('name', 'ed', False), - call('name', 'mary', False), - call('name', 'mary', True), - # append a1 - call('addresses', a1, False), - # remove a1 - call('addresses', a1, True), - # set to [a1, a2] - this is two appends - call('addresses', a1, False), call('addresses', a2, False), - # set to [a2, a3] - this is a remove of a1, - # append of a3. the appends are first. - # in 1.2 due to #3896, we also get 'a2' in the - # validates as it is part of the set - call('addresses', a2, False), - call('addresses', a3, False), - call('addresses', a1, True), - ]) + eq_( + canary.mock_calls, + [ + call("name", "ed", False), + call("name", "mary", False), + call("name", "mary", True), + # append a1 + call("addresses", a1, False), + # remove a1 + call("addresses", a1, True), + # set to [a1, a2] - this is two appends + call("addresses", a1, False), + call("addresses", a2, False), + # set to [a2, a3] - this is a remove of a1, + # append of a3. the appends are first. + # in 1.2 due to #3896, we also get 'a2' in the + # validates as it is part of the set + call("addresses", a2, False), + call("addresses", a3, False), + call("addresses", a1, True), + ], + ) def test_validator_bulk_collection_set(self): - users, addresses, Address = (self.tables.users, - self.tables.addresses, - self.classes.Address) + users, addresses, Address = ( + self.tables.users, + self.tables.addresses, + self.classes.Address, + ) class User(fixtures.ComparableEntity): - - @validates('addresses', include_removes=True) + @validates("addresses", include_removes=True) def validate_address(self, key, item, remove): if not remove: assert isinstance(item, str) @@ -167,9 +178,7 @@ class ValidatorTest(_fixtures.FixtureTest): item = Address(email_address=item) return item - mapper(User, users, properties={ - 'addresses': relationship(Address) - }) + mapper(User, users, properties={"addresses": relationship(Address)}) mapper(Address, addresses) u1 = User() @@ -177,22 +186,23 @@ class ValidatorTest(_fixtures.FixtureTest): u1.addresses.append("e2") eq_( u1.addresses, - [Address(email_address="e1"), Address(email_address="e2")] + [Address(email_address="e1"), Address(email_address="e2")], ) u1.addresses = ["e3", "e4"] eq_( u1.addresses, - [Address(email_address="e3"), Address(email_address="e4")] + [Address(email_address="e3"), Address(email_address="e4")], ) def test_validator_bulk_dict_set(self): - users, addresses, Address = (self.tables.users, - self.tables.addresses, - self.classes.Address) + users, addresses, Address = ( + self.tables.users, + self.tables.addresses, + self.classes.Address, + ) class User(fixtures.ComparableEntity): - - @validates('addresses', include_removes=True) + @validates("addresses", include_removes=True) def validate_address(self, key, item, remove): if not remove: assert isinstance(item, str) @@ -201,13 +211,18 @@ class ValidatorTest(_fixtures.FixtureTest): item = Address(email_address=item) return item - mapper(User, users, properties={ - 'addresses': relationship( - Address, - collection_class=collections.attribute_mapped_collection( - "email_address") - ) - }) + mapper( + User, + users, + properties={ + "addresses": relationship( + Address, + collection_class=collections.attribute_mapped_collection( + "email_address" + ), + ) + }, + ) mapper(Address, addresses) u1 = User() @@ -217,16 +232,16 @@ class ValidatorTest(_fixtures.FixtureTest): u1.addresses, { "e1": Address(email_address="e1"), - "e2": Address(email_address="e2") - } + "e2": Address(email_address="e2"), + }, ) u1.addresses = {"e3": "e3", "e4": "e4"} eq_( u1.addresses, { "e3": Address(email_address="e3"), - "e4": Address(email_address="e4") - } + "e4": Address(email_address="e4"), + }, ) def test_validator_multi_warning(self): @@ -245,7 +260,9 @@ class ValidatorTest(_fixtures.FixtureTest): exc.InvalidRequestError, "A validation function for mapped attribute " "'name' on mapper Mapper|Foo|users already exists", - mapper, Foo, users + mapper, + Foo, + users, ) class Bar(object): @@ -261,7 +278,9 @@ class ValidatorTest(_fixtures.FixtureTest): exc.InvalidRequestError, "A validation function for mapped attribute " "'name' on mapper Mapper|Bar|users already exists", - mapper, Bar, users + mapper, + Bar, + users, ) def test_validator_wo_backrefs_wo_removes(self): @@ -277,41 +296,57 @@ class ValidatorTest(_fixtures.FixtureTest): self._test_validator_backrefs(True, True) def _test_validator_backrefs(self, include_backrefs, include_removes): - users, addresses = (self.tables.users, - self.tables.addresses) + users, addresses = (self.tables.users, self.tables.addresses) canary = Mock() class User(fixtures.ComparableEntity): if include_removes: - @validates('addresses', include_removes=True, - include_backrefs=include_backrefs) + + @validates( + "addresses", + include_removes=True, + include_backrefs=include_backrefs, + ) def validate_address(self, key, item, remove): canary(key, item, remove) return item + else: - @validates('addresses', include_removes=False, - include_backrefs=include_backrefs) + + @validates( + "addresses", + include_removes=False, + include_backrefs=include_backrefs, + ) def validate_address(self, key, item): canary(key, item) return item class Address(fixtures.ComparableEntity): if include_removes: - @validates('user', include_backrefs=include_backrefs, - include_removes=True) + + @validates( + "user", + include_backrefs=include_backrefs, + include_removes=True, + ) def validate_user(self, key, item, remove): canary(key, item, remove) return item + else: - @validates('user', include_backrefs=include_backrefs) + + @validates("user", include_backrefs=include_backrefs) def validate_user(self, key, item): canary(key, item) return item - mapper(User, users, properties={ - 'addresses': relationship(Address, backref="user") - }) + mapper( + User, + users, + properties={"addresses": relationship(Address, backref="user")}, + ) mapper(Address, addresses) u1 = User() @@ -331,66 +366,64 @@ class ValidatorTest(_fixtures.FixtureTest): if include_backrefs: if include_removes: - eq_(calls, + eq_( + calls, [ # append #1 - call('addresses', Address(), False), - + call("addresses", Address(), False), # backref for append - call('user', User(addresses=[]), False), - + call("user", User(addresses=[]), False), # append #2 - call('addresses', Address(user=None), False), - + call("addresses", Address(user=None), False), # backref for append - call('user', User(addresses=[]), False), - + call("user", User(addresses=[]), False), # assign a2.user = u2 - call('user', User(addresses=[]), False), - + call("user", User(addresses=[]), False), # backref for u1.addresses.remove(a2) - call('addresses', Address(user=None), True), - + call("addresses", Address(user=None), True), # backref for u2.addresses.append(a2) - call('addresses', Address(user=None), False), - + call("addresses", Address(user=None), False), # del a1.user - call('user', User(addresses=[]), True), - + call("user", User(addresses=[]), True), # backref for u1.addresses.remove(a1) - call('addresses', Address(), True), - + call("addresses", Address(), True), # u2.addresses.remove(a2) - call('addresses', Address(user=None), True), - + call("addresses", Address(user=None), True), # backref for a2.user = None - call('user', None, False) - ]) + call("user", None, False), + ], + ) else: - eq_(calls, + eq_( + calls, [ - call('addresses', Address()), - call('user', User(addresses=[])), - call('addresses', Address(user=None)), - call('user', User(addresses=[])), - call('user', User(addresses=[])), - call('addresses', Address(user=None)), - call('user', None) - ]) + call("addresses", Address()), + call("user", User(addresses=[])), + call("addresses", Address(user=None)), + call("user", User(addresses=[])), + call("user", User(addresses=[])), + call("addresses", Address(user=None)), + call("user", None), + ], + ) else: if include_removes: - eq_(calls, + eq_( + calls, [ - call('addresses', Address(), False), - call('addresses', Address(user=None), False), - call('user', User(addresses=[]), False), - call('user', User(addresses=[]), True), - call('addresses', Address(user=None), True) - ]) + call("addresses", Address(), False), + call("addresses", Address(user=None), False), + call("user", User(addresses=[]), False), + call("user", User(addresses=[]), True), + call("addresses", Address(user=None), True), + ], + ) else: - eq_(calls, + eq_( + calls, [ - call('addresses', Address()), - call('addresses', Address(user=None)), - call('user', User(addresses=[])) - ]) + call("addresses", Address()), + call("addresses", Address(user=None)), + call("user", User(addresses=[])), + ], + ) |