def init_db(cls, connect_str="sqlite:///:memory:", do_bootstrap=True):
"""
Policy-Engine specific db initialization and setup for testing.
:param connect_str: connection string, defaults to sqllite in-memory if none provided
:return:
"""
conf = cls.setup_engine_config(connect_str)
from anchore_engine.db import (
initialize,
Account,
AccountUser,
AccessCredential,
Anchore,
)
from anchore_engine.db.entities.common import do_create
from anchore_engine.version import version, db_version
initialize(
versions={
"service_version": version,
"db_version": db_version
},
localconfig=conf,
)
do_create(specific_tables=[
Account.__table__,
AccountUser.__table__,
AccessCredential.__table__,
Anchore.__table__,
])
def init_db(connect_str='sqlite:///:memory:', do_bootstrap=True):
"""
Policy-Engine specific db initialization and setup for testing.
:param connect_str: connection string, defaults to sqllite in-memory if none provided
:return:
"""
conf = TestRDBMSArchiveDriver.setup_engine_config(connect_str)
from anchore_engine.db import initialize, ArchiveDocument, Anchore, ObjectStorageRecord
from anchore_engine.db.entities.common import do_create
from anchore_engine.version import version, db_version
initialize(versions={
'service_version': version,
'db_version': db_version
},
localconfig=conf,
specific_tables=[
ArchiveDocument.__table__, Anchore.__table__,
ObjectStorageRecord.__table__
],
bootstrap_db=do_bootstrap)
do_create(specific_tables=[
ArchiveDocument.__table__, Anchore.__table__,
ObjectStorageRecord.__table__
])
def do_create_tables(specific_tables=None):
print("Creating DB Tables")
from anchore_engine.db.entities.common import Base, do_create
try:
with upgrade_context(my_module_upgrade_id) as ctx:
do_create(specific_tables=specific_tables, base=Base)
except Exception as err:
raise err
print("DB Tables created")
return (True)
def init_db(connect_str='sqlite:///:memory:', do_bootstrap=False):
"""
Policy-Engine specific db initialization and setup for testing.
:param connect_str: connection string, defaults to sqllite in-memory if none provided
:return:
"""
conf = setup_engine_config(connect_str)
from anchore_engine.db import initialize
from anchore_engine.db.entities.common import do_create
pol_tables = get_policy_tables()
initialize(localconfig=conf)
if do_bootstrap:
do_create(pol_tables)
init_distro_mappings()
