def init(): db.init_app(app) db.drop_all(app=app) db.create_all(app=app) print("init...db") email = app.config.get('ADMINS')[0] admin = User('admin', email, '-+') db.session.add(admin) user2 = User('user2', '*****@*****.**', '-+') db.session.add(user2) uppo = Uppo() uppo.name_p = 'admin P' uppo.id = 2 # uppo.user = user2 db.session.add(uppo) p = KnPost(u'test post') p.html_content = 'this is a test post from manager' db.session.add(p) db.session.commit() print("create...admin")
import sys import os sys.path.insert(0, os.path.join('.', 'pixtch')) import unittest from modules import db from flaskPixtch import create_app db_uri = 'sqlite:///db/unit_test.db' app = create_app() app.init_db(uri=db_uri) app.setup() db.app = app db.init_app(app) db.drop_all() db.create_all() class DatabaseTestCase(unittest.TestCase): def setUp(self): self.app = app.test_client() # app.testing = True app.config['TESTING'] = True print 'setup...' def tearDown(self): # os.close(self.db_fd) pass # os.unlink(flaskr.config['DATABASE'])