예제 #1
0
파일: manager.py 프로젝트: solpie/Pixtch
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")
예제 #2
0
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'])