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
