def test_merge_prefer_merger(self): user_a = User(**self.userdata) user_b = User(name='Hubble Bubbs') auth_a = Auth(**self.authdata) auth_b = Auth(self.authdata['provider'], '8ahf81', '98kn32kafo2') auth_a.user = user_a auth_b.user = user_b for obj in [user_a, user_b, auth_a, auth_b]: db.session.add(obj) db.session.commit() user_a.merge(user_b) self.assertEqual(len(user_a.auths.all()), 1) self.assertEqual(user_a.auths.first().provider_id, self.authdata['provider_id'])
def test_merge(self): user_a = User(**self.userdata) user_b = User(name='Hubble Bubbs') auth_a = Auth(**self.authdata) auth_b = Auth('twitter', '8ahf81', '98kn32kafo2') auth_a.user = user_a auth_b.user = user_b for obj in [user_a, user_b, auth_a, auth_b]: db.session.add(obj) db.session.commit() self.assertEqual(User.query.count(), 2) user_a.merge(user_b) self.assertEqual(len(user_a.auths.all()), 2) self.assertEqual(User.query.count(), 1)