def test_dropped_db(client):
"""Check behavior when the db has no tables."""
db.drop_all()
assert b'Sorry' in client.get('/poetry/poem').data
assert "404" in client.get('/poetry/poem/Flea').status
# /add_samples route should not cause server error when db has no tables
assert "404" in client.get('/poetry/add_samples').status
def init_db():
"""
Drops and re-creates the SQL schema
"""
db.drop_all()
db.configure_mappers()
db.create_all()
db.session.commit()
def init_db():
"""
Drops and re-creates the SQL schema
"""
db.drop_all()
db.configure_mappers()
db.create_all()
db.session.commit()
def _test_db(_test_app):
test_app = _test_app
tdb.init_app(test_app)
try:
initilize_test_db(tdb)
yield tdb
finally:
tdb.session.commit()
tdb.drop_all()
def database():
db.session.remove()
db.reflect()
db.drop_all()
db.create_all()
yield db # this is where the testing happens!
db.drop_all()
def client():
""" My custom test client.
"""
app.config['TESTING'] = True
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'
# app.config['SQLALCHEMY_DATABASE_URI'] = \
# 'mysql+mysqlconnector://test:Test12345678!@localhost/iot_test'
with app.app_context():
db.drop_all()
db.create_all()
# Add test user
user = User(username='******', email='*****@*****.**')
user.set_password('test')
db.session.add(user)
db.session.commit()
# Add test device
device = Device(name='test', display_name='test', user=user)
db.session.add(device)
db.session.commit()
schema = {
'test1': ['test1', 2, 1, 0],
'test2': ['test2', 2, 1, 0],
'control': ['control', 3, 1, 1],
}
device.set_schema(schema)
# Add test data
new_data = DeviceData(time=datetime.utcfromtimestamp(1500000000),
data={
'test1': 10.0,
'test2': 12.0,
'control': False,
},
device=device)
db.session.add(new_data)
db.session.commit()
# Get token
access_token = create_access_token(identity=user.username)
refresh_token = create_refresh_token(identity=user.username)
my_client = MyTestClient(access_token,
refresh_token,
default_auth=TokenType.access)
yield my_client
with app.app_context():
db.session.remove()
db.drop_all()
def clear_db(db):
confirm = input('Clear all tables? [yes/no]: ')
if confirm == 'yes':
# drop/create tables
print('#'*60)
print('Dropping all tables')
print('#'*60)
db.drop_all()
print('#'*60)
print('Creating all tables')
print('#'*60)
db.create_all()
print('All tables cleared')
elif confirm == 'no':
exit(1)
else:
print('Please type yes or no')
clear_db(db)
def clear_db(db):
confirm = input('Clear all tables? [yes/no]: ')
if confirm == 'yes':
# drop/create tables
print('#'*60)
print('Dropping all tables')
print('#'*60)
db.drop_all()
print('#'*60)
print('Creating all tables')
print('#'*60)
db.create_all()
print('All tables cleared')
elif confirm == 'no':
exit(1)
else:
print('Please type yes or no')
clear_db(db)
def app():
"""Set up and tear down the test app."""
from run import create_app, db
app = create_app('testing')
app.config['TESTING'] = True
app_context = app.app_context()
app_context.push()
db.create_all()
yield app
db.session.remove()
db.drop_all()
app_context.pop()
def initial_db():
### Drop All tables
db.reflect()
db.drop_all()
### Clear sessions
session.clear()
session.pop('username', None)
### Create Tables
db.create_all()
### Create Roles
administrative_staff = models.Role(name='IdariPersonel')
teacher = models.Role(name='Ogretmen')
student = models.Role(name='Ogrenci')
db.session.add(administrative_staff)
db.session.add(teacher)
db.session.add(student)
db.session.commit()
### Create Role Funcs
func_create_user = models.RoleFuncs(name='CreateUser', role_id=administrative_staff.id)
func_see_all_user = models.RoleFuncs(name='SeeAllUser', role_id=administrative_staff.id)
db.session.add(func_create_user)
db.session.add(func_see_all_user)
db.session.commit()
func_danisman_ata = models.RoleFuncs(name='AssignCons', role_id=administrative_staff.id)
db.session.add(func_danisman_ata)
db.session.commit()
###################### Create Administrative Staff
new_administrative_staff = create_new_user(
uname="admin",
password="******",
fname= "Mustafa",
mname= "Kemal",
sname= "Ataturk",
email= "*****@*****.**",
role_id=administrative_staff.id
)
return '<a href="/">Ok!</a>'
def tearDown(self):
db.session.remove()
db.drop_all()
from run import db from run import Role, User db.drop_all() db.create_all() admin_role = Role(name='Admin') user_john = User(username='******', role=admin_role) db.session.add(admin_role) db.session.commit()
def api_delete_all():
db.drop_all()
db.create_all()
create_test_data()
return {'status': 'Successfully deleted all database data'}, 200
def setUp(self):
db.drop_all()
db.create_all()
db.session.commit()
def dropall():
db.drop_all()
print("all tables dropped! remember to delete directory: migrations")
