def test_init_rest():
"""Test REST extension initialization."""
app = Flask('testapp')
Babel(app)
Mail(app)
InvenioDB(app)
ext = InvenioAccountsREST(app)
assert 'invenio-accounts' in app.extensions
assert 'security' not in app.blueprints.keys()
app = Flask('testapp')
Babel(app)
Mail(app)
InvenioDB(app)
ext = InvenioAccountsREST()
assert 'invenio-accounts' not in app.extensions
assert 'security' not in app.blueprints.keys()
ext.init_app(app)
assert 'invenio-accounts' in app.extensions
assert 'security' not in app.blueprints.keys()
app = Flask('testapp')
app.config['ACCOUNTS_REGISTER_BLUEPRINT'] = True
Babel(app)
Mail(app)
InvenioDB(app)
ext = InvenioAccountsREST()
assert 'invenio-accounts' not in app.extensions
assert 'security' not in app.blueprints.keys()
ext.init_app(app)
assert 'invenio-accounts' in app.extensions
assert 'security' in app.blueprints.keys()
def app(request):
"""Flask application fixture."""
instance_path = tempfile.mkdtemp()
def init_app(app):
app.config.update(
LOGIN_DISABLED=False,
MAIL_SUPPRESS_SEND=True,
OAUTH2_CACHE_TYPE='simple',
OAUTHLIB_INSECURE_TRANSPORT=True,
SECRET_KEY='CHANGE_ME',
SECURITY_DEPRECATED_PASSWORD_SCHEMES=[],
SECURITY_PASSWORD_HASH='plaintext',
SECURITY_PASSWORD_SALT='CHANGE_ME_ALSO',
SECURITY_PASSWORD_SCHEMES=['plaintext'],
SQLALCHEMY_DATABASE_URI=os.getenv(
'SQLALCHEMY_DATABASE_URI',
'sqlite:///' + os.path.join(instance_path, 'test.db')),
SQLALCHEMY_TRACK_MODIFICATIONS=True,
TESTING=True,
WTF_CSRF_ENABLED=False,
)
Babel(app)
Mail(app)
Menu(app)
Breadcrumbs(app)
InvenioDB(app)
InvenioOAuth2Server(app)
api_app = Flask('testapiapp', instance_path=instance_path)
api_app.config.update(APPLICATION_ROOT='/api',
ACCOUNTS_REGISTER_BLUEPRINT=True)
init_app(api_app)
InvenioAccountsREST(api_app)
InvenioOAuth2ServerREST(api_app)
app = Flask('testapp', instance_path=instance_path)
init_app(app)
InvenioAccountsUI(app)
app.register_blueprint(accounts_blueprint)
app.register_blueprint(server_blueprint)
app.register_blueprint(settings_blueprint)
app.wsgi_app = DispatcherMiddleware(app.wsgi_app,
{'/api': api_app.wsgi_app})
with app.app_context():
if str(db.engine.url) != 'sqlite://' and \
not database_exists(str(db.engine.url)):
create_database(str(db.engine.url))
db.create_all()
def teardown():
with app.app_context():
if str(db.engine.url) != 'sqlite://':
drop_database(str(db.engine.url))
shutil.rmtree(instance_path)
request.addfinalizer(teardown)
return app
def app_with_flexible_registration(request):
"""Flask application fixture with Invenio Accounts."""
from webargs import fields
from invenio_accounts.views.rest import RegisterView, use_kwargs
class MyRegisterView(RegisterView):
post_args = {
**RegisterView.post_args,
'active': fields.Boolean(required=True)
}
@use_kwargs(post_args)
def post(self, **kwargs):
"""Register a user."""
return super(MyRegisterView, self).post(**kwargs)
api_app = _app_factory()
InvenioREST(api_app)
InvenioAccountsREST(api_app)
api_app.config['ACCOUNTS_REST_AUTH_VIEWS']['register'] = MyRegisterView
api_app.register_blueprint(create_blueprint(api_app))
_database_setup(api_app, request)
yield api_app
def base_app():
"""Flask application fixture without InvenioStats."""
from invenio_stats.config import STATS_EVENTS
instance_path = tempfile.mkdtemp()
app_ = Flask('testapp', instance_path=instance_path)
stats_events = {
'file-download': deepcopy(STATS_EVENTS['file-download']),
'record-view': {
'signal': 'invenio_records_ui.signals.record_viewed',
'event_builders': ['invenio_stats.contrib.event_builders'
'.record_view_event_builder']
}
}
stats_events.update({'event_{}'.format(idx): {} for idx in range(5)})
app_.config.update(dict(
CELERY_ALWAYS_EAGER=True,
CELERY_TASK_ALWAYS_EAGER=True,
CELERY_CACHE_BACKEND='memory',
CELERY_EAGER_PROPAGATES_EXCEPTIONS=True,
CELERY_TASK_EAGER_PROPAGATES=True,
CELERY_RESULT_BACKEND='cache',
SQLALCHEMY_DATABASE_URI=os.environ.get(
'SQLALCHEMY_DATABASE_URI', 'sqlite://'),
SQLALCHEMY_TRACK_MODIFICATIONS=True,
TESTING=True,
OAUTH2SERVER_CLIENT_ID_SALT_LEN=64,
OAUTH2SERVER_CLIENT_SECRET_SALT_LEN=60,
OAUTH2SERVER_TOKEN_PERSONAL_SALT_LEN=60,
STATS_MQ_EXCHANGE=Exchange(
'test_events',
type='direct',
delivery_mode='transient', # in-memory queue
durable=True,
),
SECRET_KEY='asecretkey',
SERVER_NAME='localhost',
STATS_QUERIES={'bucket-file-download-histogram': {},
'bucket-file-download-total': {},
'test-query': {},
'test-query2': {}},
STATS_EVENTS=stats_events,
STATS_AGGREGATIONS={'file-download-agg': {}}
))
FlaskCeleryExt(app_)
InvenioAccounts(app_)
InvenioAccountsREST(app_)
InvenioDB(app_)
InvenioRecords(app_)
InvenioFilesREST(app_)
InvenioPIDStore(app_)
InvenioCache(app_)
InvenioQueues(app_)
InvenioOAuth2Server(app_)
InvenioOAuth2ServerREST(app_)
InvenioSearch(app_, entry_point_group=None)
with app_.app_context():
yield app_
shutil.rmtree(instance_path)
def test_init_rest():
"""Test REST extension initialization."""
app = Flask('testapp')
Babel(app)
Mail(app)
InvenioDB(app)
ext = InvenioAccountsREST(app)
assert 'invenio-accounts' in app.extensions
assert 'security' not in app.blueprints.keys()
app = Flask('testapp')
Babel(app)
Mail(app)
InvenioDB(app)
ext = InvenioAccountsREST()
assert 'invenio-accounts' not in app.extensions
assert 'security' not in app.blueprints.keys()
ext.init_app(app)
assert 'invenio-accounts' in app.extensions
assert 'security' not in app.blueprints.keys()
app = Flask('testapp')
app.config['ACCOUNTS_REGISTER_BLUEPRINT'] = True
Babel(app)
Mail(app)
InvenioDB(app)
ext = InvenioAccountsREST()
assert 'invenio-accounts' not in app.extensions
assert 'security' not in app.blueprints.keys()
ext.init_app(app)
assert 'invenio-accounts' in app.extensions
assert 'security' in app.blueprints.keys()
def test_init_rest():
"""Test REST extension initialization."""
app = Flask("testapp")
app.config["SECRET_KEY"] = "CHANGEME"
Babel(app)
Mail(app)
InvenioDB(app)
ext = InvenioAccountsREST(app)
assert "invenio-accounts" in app.extensions
assert "security" not in app.blueprints.keys()
assert "security_email_templates" in app.blueprints.keys()
app = Flask("testapp")
app.config["SECRET_KEY"] = "CHANGEME"
Babel(app)
Mail(app)
InvenioDB(app)
ext = InvenioAccountsREST()
assert "invenio-accounts" not in app.extensions
assert "security" not in app.blueprints.keys()
ext.init_app(app)
assert "invenio-accounts" in app.extensions
assert "security" not in app.blueprints.keys()
assert "security_email_templates" in app.blueprints.keys()
app = Flask("testapp")
app.config["SECRET_KEY"] = "CHANGEME"
app.config["ACCOUNTS_REGISTER_BLUEPRINT"] = True
Babel(app)
Mail(app)
InvenioDB(app)
ext = InvenioAccountsREST()
assert "invenio-accounts" not in app.extensions
assert "security" not in app.blueprints.keys()
ext.init_app(app)
assert "invenio-accounts" in app.extensions
assert "security" in app.blueprints.keys()
assert "security_email_templates" in app.blueprints.keys()
def base_app(events_config, aggregations_config):
"""Flask application fixture without InvenioStats."""
instance_path = tempfile.mkdtemp()
app_ = Flask('testapp', instance_path=instance_path)
app_.config.update(dict(
CELERY_ALWAYS_EAGER=True,
CELERY_TASK_ALWAYS_EAGER=True,
CELERY_CACHE_BACKEND='memory',
CELERY_EAGER_PROPAGATES_EXCEPTIONS=True,
CELERY_TASK_EAGER_PROPAGATES=True,
CELERY_RESULT_BACKEND='cache',
SQLALCHEMY_DATABASE_URI=os.environ.get(
'SQLALCHEMY_DATABASE_URI', 'sqlite://'),
SQLALCHEMY_TRACK_MODIFICATIONS=True,
# Bump the ES client timeout for slower environments (like Travis CI)
SEARCH_CLIENT_CONFIG={'timeout': 30, 'max_retries': 5},
TESTING=True,
OAUTH2SERVER_CLIENT_ID_SALT_LEN=64,
OAUTH2SERVER_CLIENT_SECRET_SALT_LEN=60,
OAUTH2SERVER_TOKEN_PERSONAL_SALT_LEN=60,
STATS_MQ_EXCHANGE=Exchange(
'test_events',
type='direct',
delivery_mode='transient', # in-memory queue
durable=True,
),
SECRET_KEY='asecretkey',
SERVER_NAME='localhost',
STATS_QUERIES={},
STATS_EVENTS=events_config,
STATS_AGGREGATIONS=aggregations_config,
))
FlaskCeleryExt(app_)
InvenioAccounts(app_)
InvenioAccountsREST(app_)
InvenioDB(app_)
InvenioRecords(app_)
InvenioFilesREST(app_)
InvenioPIDStore(app_)
InvenioCache(app_)
InvenioQueues(app_)
InvenioOAuth2Server(app_)
InvenioOAuth2ServerREST(app_)
InvenioSearch(app_, entry_point_group=None)
with app_.app_context():
yield app_
shutil.rmtree(instance_path)
def api(request):
"""Flask application fixture."""
api_app = _app_factory(
dict(
SQLALCHEMY_DATABASE_URI=os.environ.get(
'SQLALCHEMY_DATABASE_URI', 'sqlite:///test.db'),
SERVER_NAME='localhost',
TESTING=True,
))
InvenioREST(api_app)
InvenioAccountsREST(api_app)
api_app.register_blueprint(create_blueprint(api_app))
_database_setup(api_app, request)
yield api_app
('fr', 'French'),
('it', 'Italian'),
],
)
InvenioAssets(app)
InvenioTheme(app)
InvenioI18N(app)
Breadcrumbs(app)
InvenioDB(app)
InvenioAdmin(app)
InvenioAccess(app)
OAuth2Provider(app)
InvenioOAuth2ServerREST(app)
accounts = InvenioAccountsUI(app)
InvenioAccountsREST(app)
app.register_blueprint(blueprint_account)
InvenioOAuth2Server(app)
# Register blueprints
app.register_blueprint(settings_blueprint)
app.register_blueprint(server_blueprint)
app.register_blueprint(blueprint_admin_ui)
with app.app_context():
# Register a test scope
current_oauth2server.register_scope(
Scope('test:scope', help_text='Access to the homepage', group='test'))
# @app.route('/jwt', methods=['GET'])