def test_valid_email_change_token(self): user1 = User('test1') user1.email = '*****@*****.**' user1.password = '******' db.session.add(user1) db.session.commit() token = user1.generate_email_change_token('*****@*****.**') self.assertTrue(user1.change_email(token)) self.assertTrue(user1.email == '*****@*****.**')
def test_duplicate_email_change_token(self): user1 = User('test1') user1.email = '*****@*****.**' user1.password = '******' user2 = User('test2') user2.email = '*****@*****.**' user2.password = '******' db.session.add(user1) db.session.add(user2) db.session.commit() token = user2.generate_email_change_token('*****@*****.**') self.assertFalse(user2.change_email(token)) self.assertTrue(user2.email == '*****@*****.**')