Python DB.drop_all Examples

Example #1

File: test_lifegroup_service.py Project: KennethLikeCoding/HopeTransport

    def setUp(self):
        APP.config['TESTING'] = True
        APP.config['DEBUG'] = False
        APP.config[
            'SQLALCHEMY_DATABASE_URI'] = 'postgres://localhost/transport_test'

        self.app = APP.test_client()

        DB.drop_all()
        DB.create_all()

        self.lifegoup1 = {
            'name': 'uq1',
            'password': '******',
            'email': '*****@*****.**'
        }
        self.lifegoup2 = {
            'name': 'uq2',
            'password': '******',
            'email': '*****@*****.**'
        }
        self.lifegoup3 = {
            'name': 'uq3',
            'password': '******',
            'email': '*****@*****.**'
        }

        DB.session.add(LifegroupModel(**self.lifegoup1))
        DB.session.add(LifegroupModel(**self.lifegoup2))
        DB.session.add(LifegroupModel(**self.lifegoup3))
        DB.session.commit()

        self.member1 = {
            'id': '1',
            'lifegroup': 'uq1',
            'name': 'Kenneth Guo',
            'suburb': 'Sunnybank QLD, 4109'
        }
        self.member2 = {
            'id': '2',
            'lifegroup': 'uq2',
            'name': 'Ken Guo',
            'suburb': 'Sunnybank QLD, 4109'
        }
        self.note1 = {
            'id': '1',
            'lifegroup': 'uq1',
            'text': "A and B stay at the same place"
        }
        self.note2 = {
            'id': '2',
            'lifegroup': 'uq2',
            'text': "C and D stay at the same place"
        }
        DB.session.add(MemberModel(**self.member1))
        DB.session.add(MemberModel(**self.member2))
        DB.session.add(NoteModel(**self.note1))
        DB.session.add(NoteModel(**self.note2))
        DB.session.commit()
        self.assertFalse(APP.debug)

Example #2

File: test_auth_service.py Project: KennethLikeCoding/HopeTransport

    def setUp(self):
        APP.config['TESTING'] = True
        APP.config['DEBUG'] = False
        APP.config[
            'SQLALCHEMY_DATABASE_URI'] = 'postgres://localhost/transport_test'

        self.app = APP.test_client()

        DB.drop_all()
        DB.create_all()

        self.lifegoup1 = {
            'name': 'uq1',
            'password': '******',
            'email': '*****@*****.**'
        }
        self.lifegoup2 = {
            'name': 'uq2',
            'password': '******',
            'email': '*****@*****.**'
        }
        self.lifegoup3 = {
            'name': 'uq3',
            'password': '******',
            'email': '*****@*****.**'
        }

        DB.session.add(LifegroupModel(**self.lifegoup1))
        DB.session.add(LifegroupModel(**self.lifegoup2))
        DB.session.add(LifegroupModel(**self.lifegoup3))
        DB.session.commit()

        self.assertFalse(APP.debug)

Example #3

File: conftest.py Project: ecclesianuernberg/genesis

def db():
    ''' creates tables and drops them for every test '''
    DB.create_all()

    yield

    DB.session.commit()
    DB.drop_all()

Example #4

File: conftest.py Project: TheRealCoop/very-good-sms

def app():
    environ['FLASK_ENV'] = 'test'
    _app = create_app()
    with _app.app_context():
        DB.create_all()
        yield _app
        DB.session.remove()
        DB.drop_all()
    return _app

Example #5

 def setUp(self):
     """setting up all variables"""
     APP.config['TESTING'] = True
     APP.config['WTF_CSRF_ENABLED'] = False
     APP.config['DEBUG'] = False
     APP.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
     APP.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + \
         os.path.join(BASEDIR, TEST_DB)
     self.app = APP.test_client()
     DB.drop_all()
     DB.create_all()
     insert_jokes_categories()

Example #6

File: test_member_model.py Project: KennethLikeCoding/HopeTransport

    def setUp(self):
        APP.config['TESTING'] = True
        APP.config['DEBUG'] = False
        APP.config['SQLALCHEMY_DATABASE_URI'] = 'postgres://localhost/transport_test'

        self.app = APP.test_client()

        self.member = {'id': '123', 'name': 'Kenneth', 'suburb': 'Sunnybank, QLD 4109'}

        DB.drop_all()
        DB.create_all()

        self.assertFalse(APP.debug)

Example #7

    def setUp(self):
        APP.config['TESTING'] = True
        APP.config['DEBUG'] = False
        APP.config[
            'SQLALCHEMY_DATABASE_URI'] = 'postgres://localhost/transport_test'

        self.app = APP.test_client()

        DB.drop_all()
        DB.create_all()
        self.lifegoup1 = {
            'name': 'uq8',
            'password': '******',
            'email': '*****@*****.**'
        }
        self.lifegoup2 = {
            'name': 'uq6',
            'password': '******',
            'email': '*****@*****.**'
        }
        self.member1 = {
            'id': '1',
            'lifegroup': 'uq8',
            'name': 'Kenneth Guo',
            'suburb': 'Sunnybank QLD, 4109'
        }
        self.member2 = {
            'id': '2',
            'lifegroup': 'uq8',
            'name': 'Ezmond Cheung',
            'seats': 4,
            'suburb': 'Taringa QLD, 1234'
        }
        self.member3 = {
            'id': '3',
            'lifegroup': 'uq6',
            'name': 'Bruno Cheung',
            'suburb': 'xxxx QLD, 4321'
        }

        DB.session.add(LifegroupModel(**self.lifegoup1))
        DB.session.add(LifegroupModel(**self.lifegoup2))
        DB.session.commit()
        DB.session.add(MemberModel(**self.member1))
        DB.session.add(MemberModel(**self.member2))
        DB.session.add(MemberModel(**self.member3))
        DB.session.commit()

        self.assertFalse(APP.debug)

Example #8

    def setUp(self):
        '''Declare initial variables'''
        self.maxDiff = None
        self.app = create_app('testing')
        self.client = self.app.test_client()
        self.app_context = self.app.app_context()
        self.app_context.push()

        self.database = DB
        DB.drop_all()
        DB.create_all()
        self.today = datetime.utcnow().date()
        self.test_user = {
            'username': '******',
            'email': '*****@*****.**',
            'password': '******'
        }
        self.admin_user = dict(username='******',
                               email='*****@*****.**',
                               password='******',
                               admin=True)
        self.meal_model = Meal
        self.order_model = Order
        self.menu_model = Menu
        self.order_model = Order
        self.user_model = User
        self.user1 = User(email='*****@*****.**',
                          password='******',
                          username='******')
        self.user2 = User(email='*****@*****.**',
                          username='******',
                          password='******')

        self.menu = Menu()
        self.menu1 = Menu(date=datetime(year=2019, month=4, day=18))
        self.menu2 = Menu(date=datetime(year=2019, month=4, day=19))

        self.meal1 = Meal(name='Rice & Beef',
                          price=100.00,
                          description='Rice with beef. Yummy.')
        self.meal2 = Meal(name='Ugali Fish',
                          price=150.00,
                          description='Ugali and fish, Nyanza tings!')
        self.meal3 = Meal(name='Muthokoi',
                          price=100.00,
                          description='Kamba tributes')

Example #9

File: test_note_service.py Project: KennethLikeCoding/HopeTransport

    def setUp(self):
        APP.config['TESTING'] = True
        APP.config['DEBUG'] = False
        APP.config[
            'SQLALCHEMY_DATABASE_URI'] = 'postgres://localhost/transport_test'

        self.app = APP.test_client()

        DB.drop_all()
        DB.create_all()
        self.lifegoup1 = {
            'name': 'uq8',
            'password': '******',
            'email': '*****@*****.**'
        }
        self.lifegoup2 = {
            'name': 'uq7',
            'password': '******',
            'email': '*****@*****.**'
        }
        DB.session.add(LifegroupModel(**self.lifegoup1))
        DB.session.add(LifegroupModel(**self.lifegoup2))
        DB.session.commit()

        self.note1 = {
            'id': '1',
            'lifegroup': 'uq8',
            'text': "A and B stay at the same place"
        }
        self.note2 = {
            'id': '2',
            'lifegroup': 'uq8',
            'text': "C and D stay close to each other"
        }
        self.note3 = {
            'id': '3',
            'lifegroup': 'uq7',
            'text': "E and F want to be in the same car"
        }
        DB.session.add(NoteModel(**self.note1))
        DB.session.add(NoteModel(**self.note2))
        DB.session.add(NoteModel(**self.note3))
        DB.session.commit()

        self.assertFalse(APP.debug)

Example #10

    def setUp(self):
        APP.config['TESTING'] = True
        APP.config['DEBUG'] = False
        APP.config[
            'SQLALCHEMY_DATABASE_URI'] = 'postgres://localhost/transport_test'

        self.app = APP.test_client()

        self.note = {
            'id': '1',
            'lifegroup': 'uq8',
            'body': 'Haha make sure everyone can go home safely'
        }

        DB.drop_all()
        DB.create_all()

        self.assertFalse(APP.debug)

Example #11

    def setUp(self):
        APP.config['TESTING'] = True
        APP.config['DEBUG'] = False
        APP.config[
            'SQLALCHEMY_DATABASE_URI'] = 'postgres://localhost/transport_test'

        self.app = APP.test_client()

        self.lifegroup = {
            'name': 'uq8',
            'password': '******',
            'email': '*****@*****.**'
        }

        DB.drop_all()
        DB.create_all()

        self.assertFalse(APP.debug)

Example #12

File: conftest.py Project: erikdeirdre/people

def init_database(request):
    """Initializes the database """
    DB.drop_all()
    DB.create_all()

    base_dir = join(abspath(dirname(__file__)), '..')

    for fixture_file in glob(join(base_dir, 'seed', '*.json')):
        fixtures = JSONLoader().load(fixture_file)
        load_fixtures(DB, fixtures)

    for fixture_file in sorted(glob(join(base_dir, 'seed', 'demo', '*.json'))):
        fixtures = JSONLoader().load(fixture_file)
        load_fixtures(DB, fixtures)

    request.cls.DB = DB

    yield DB

    close_all_sessions()

Example #13

 def tearDown(self):
     """Tears down all test_db data and resets db to empty state"""
     with self.app.app_context():
         db.session.remove()
         db.drop_all()

Example #14

 def tearDown(self):
     # reset all database entries to empty dicts
     DB.session.remove()
     DB.drop_all()
     self.app_context.pop()

Example #15

 def tearDown(self):
     with APP.app_context():
         DB.session.remove()
         DB.drop_all()

Example #16

File: test_basics.py Project: kzkz221/flasky

 def tearDown(self):
     DB.session.remove()
     DB.drop_all()
     self.app_context.pop()

Example #17

 def tearDown(self):
     """ A method that is called after each test. """
     with self.app.app_context():
         DB.session.remove()
         DB.drop_all()

Example #18

File: seed_db.py Project: Skrewtape/tom

#!./venv/bin/python
"""Set up DB tables and add some core data"""
from app import DB

import app.types

from re import compile, UNICODE

_strip_pattern = compile('[\W_]+', UNICODE)

DB.drop_all()
DB.create_all()

ROLE_MAP = {}

ALL_ROLES = ['admin', 'scorekeeper']

for role_name in ALL_ROLES:
    role = app.types.Role(name=role_name)
    DB.session.add(role)
    ROLE_MAP[role_name] = role

for info in [
    ['avi', 'finkel.org', ALL_ROLES],
    ['elizabeth', 'papa.org', ALL_ROLES],
    ['admin', 'papa.org', ['admin']],
    ['scorekeeper', 'papa.org', ['scorekeeper']],
]:
    user = app.types.User(
        username=info[0],
        email=info[0] + '@' + info[1],

Example #19

 def tearDown(self):
     DB.drop_all()

Example #20

 def tearDown(self):
     """teardown all initialized variables."""
     with self.app.app_context():
         # drop all tables
         DB.session.remove()
         DB.drop_all()

Example #21

 def tearDown(self):
     DB.drop_all(app=self.app)

Example #22

 def tearDown(self):
     db.session.remove()
     db.drop_all()
     self.client = None
     self.app_context.pop()