42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
#!flask/bin/python
|
||
|
|
import os
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from config import basedir
|
||
|
|
from app import app, db
|
||
|
|
from app.models import User
|
||
|
|
|
||
|
|
class TestCase(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
app.config['TESTING'] = True
|
||
|
|
app.config['CSRF_ENABLED'] = False
|
||
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join(basedir, 'test.db')
|
||
|
|
db.create_all()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
db.drop_all()
|
||
|
|
|
||
|
|
def test_avatar(self):
|
||
|
|
# create a user
|
||
|
|
u = User(nickname = 'john', email = 'john@example.com')
|
||
|
|
avatar = u.avatar(128)
|
||
|
|
expected = 'http://www.gravatar.com/avatar/d4c74594d841139328695756648b6bd6'
|
||
|
|
assert avatar[0:len(expected)] == expected
|
||
|
|
|
||
|
|
def test_make_unique_nickname(self):
|
||
|
|
# create a user and write it to the database
|
||
|
|
u = User(nickname = 'john', email = 'john@example.com')
|
||
|
|
db.session.add(u)
|
||
|
|
db.session.commit()
|
||
|
|
nickname = User.make_unique_nickname('john')
|
||
|
|
assert nickname != 'john'
|
||
|
|
# make another user with the new nickname
|
||
|
|
u = User(nickname = nickname, email = 'susan@example.com')
|
||
|
|
db.session.add(u)
|
||
|
|
db.session.commit()
|
||
|
|
nickname2 = User.make_unique_nickname('john')
|
||
|
|
assert nickname2 != 'john'
|
||
|
|
assert nickname2 != nickname
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|