def addsampledata():
"""Add sample data"""
OrganizationGroup._OrganizationGroup__insert_defaults()
ReportType._ReportType__insert_defaults()
Role._Role__insert_defaults()
Country._Country__insert_defaults()
MembershipRole._MembershipRole__insert_defaults()
adm = Role.query.filter_by(permissions=0xff).first()
o = Organization(
abbreviation="CERT-EU",
full_name="Computer Emergency Response Team for EU "
"Institutions Agencies and Bodies",
ip_ranges=['212.8.189.16/28'],
abuse_emails=['*****@*****.**'],
contact_emails=[ContactEmail(email='*****@*****.**')],
asns=[5400],
fqdns=['cert.europa.eu'])
db.session.add(o)
db.session.commit()
user = User(name='testadmin',
email='*****@*****.**',
password='******',
role=adm)
db.session.add(user)
db.session.commit()
click.echo('Done')
def db(request, app):
"""Create test database tables"""
_db.reflect()
_db.drop_all()
# Create the tables based on the current model
_db.create_all()
MembershipRole._MembershipRole__insert_defaults()
Country._Country__insert_defaults()
# user = User.create_test_user()
testfixture.testdata.addyaml()
user = User.query.filter_by(name="certmaster").first()
TestClient.test_user = user
TestClient._api_user = user
TestClient.test_user.organization_id = Organization.query.filter_by(abbreviation='cert').first().id
app.test_client_class = TestClient
app.response_class = TestResponse
_db.session.commit()
def initializedb():
""" fill lookup tables """
Country._Country__insert_defaults()
MembershipRole._MembershipRole__insert_defaults()
click.echo('Done')
