From 665c22a4ef8e38de6954e8ee47b4c02a24353a73 Mon Sep 17 00:00:00 2001 From: Diego Date: Tue, 7 Mar 2017 23:22:54 +0100 Subject: [PATCH] tests: model signal connection SQLAlchemy 2.2 Signed-off-by: Diego Rodriguez Co-authored-by: Alexander Ioannidis --- tests/test_utils.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index fda6e07..b9ddf6f 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -46,8 +46,11 @@ def remove_schema(sender, json=None, record=None, **kwargs): if '$schema' in json: del json['$schema'] - models_committed.connect(process_models_committed_signal, sender=app) - + with app.app_context(): + # NOTE: We have to use `db.get_app()` because Flask-SQLAlchemy v2.2 + # changed the way it picks a sender for the model signals. + models_committed.connect(process_models_committed_signal, + sender=db.get_app()) with app.app_context(): current_search_client.indices.delete_alias('_all', '_all', @@ -85,7 +88,7 @@ def remove_schema(sender, json=None, record=None, **kwargs): record_indexer = RecordIndexer(queue=queue) result = record_indexer.process_bulk_queue() - assert 2 == len(list(result)) + assert (2, 0) == result response = current_search_client.get( index='records-default-v1.0.0',