def test_publicid(self): p = PublicId(idtype=PublicId.TWITTER, account="user1", user_id=self.user.id, verifyed=True) db.session.add(p) db.session.commit() p = PublicId(idtype=PublicId.EMAIL, account="user1", user_id=self.user.id, verifyed=True, code="my_code") db.session.add(p) db.session.commit() old_code = p.code p.refresh_code() assert p.code != old_code
def test_verify_email(self): p = PublicId(idtype=PublicId.EMAIL, account="*****@*****.**", user_id=self.user.id) db.session.add(p) db.session.commit() p.refresh_code() url = '/api/verify?key={}'.format(p.code) result = self.client.get(url) self.assertEqual(result.status_code, 200) data = json.loads(result.data.decode('ascii')) assert data["message"] == "verify completed" assert p.verifyed