Fix CI issues and Bump dev requirements

astronomer · Jan 24, 2022 · ce17679 · ce17679
1 parent 9a0961a
commit ce17679
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -4,11 +4,10 @@ wheel
 # We need our back-ported fixes to Airflow for our tests to pass
 # We need to specify the astro version here, otherwise it would install the one from pypi repo instead
 --extra-index-url=https://pip.astronomer.io/simple/
---constraint=https://raw.githubusercontent.com/astronomer/ap-airflow/master/1.10.7/buster/include/pip-constraints.txt
-astronomer-certified==1.10.7.*
+--constraint=https://raw.githubusercontent.com/astronomer/ap-airflow/master/2.2.3/bullseye/build-time-pip-constraints.txt
+astronomer-certified==2.2.3.*
 
 -e .[test]
 
-attrs<20.0
-marshmallow-sqlalchemy==0.23.0
-marshmallow<=2.21.0
+marshmallow-sqlalchemy~=0.26.1
+marshmallow~=3.14.0
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -35,10 +35,10 @@ def app():
     os.environ['AIRFLOW__CORE__SQL_ALCHEMY_CONN'] = 'sqlite:///testdb.sqlite'
     os.environ['AIRFLOW__WEBSERVER__RBAC'] = 'True'
     from airflow.utils.db import initdb
-    from airflow.www_rbac.app import create_app
+    from airflow.www.app import create_app
 
-    initdb(rbac=True)
-    app, _ = create_app(testing=True)
+    initdb()
+    app = create_app(testing=True)
     app.test_client_class = FlaskLoginClient
     return app
 

diff --git a/tests/test_plugin.py b/tests/test_plugin.py
@@ -5,6 +5,7 @@
 
 def test_plugin_registered():
     """Verify that the plugin is registered"""
+    plugins_manager.ensure_plugins_loaded()
     assert len(plugins_manager.plugins) == 1
     assert plugins_manager.plugins[0].name == 'astronomer_version_check'
     assert plugins_manager.plugins[0].flask_blueprints != []