def test_connection_form_widgets(self):
provider_manager = ProvidersManager()
connections_form_widgets = list(
provider_manager.connection_form_widgets.keys())
assert CONNECTION_FORM_WIDGETS == connections_form_widgets
def _providers_info(self):
return [(p.data['package-name'], p.version)
for p in ProvidersManager().providers.values()]
def test_hooks(self):
provider_manager = ProvidersManager()
connections_list = list(provider_manager.hooks.keys())
assert CONNECTIONS_LIST == connections_list
def test_field_behaviours(self):
provider_manager = ProvidersManager()
connections_with_field_behaviours = list(provider_manager.field_behaviours.keys())
assert CONNECTIONS_WITH_FIELD_BEHAVIOURS == connections_with_field_behaviours
def test_extra_links(self):
provider_manager = ProvidersManager()
extra_link_class_names = list(provider_manager.extra_links_class_names)
assert EXTRA_LINKS == extra_link_class_names
def on_starting(server):
from airflow.providers_manager import ProvidersManager
# Load providers before forking workers
ProvidersManager().connection_form_widgets
def test_hooks(self):
provider_manager = ProvidersManager()
connections_list = list(provider_manager.hooks.keys())
self.assertEqual(CONNECTIONS_LIST, connections_list)
def hooks_list(args):
"""Lists all hooks at the command line"""
msg = _tabulate_hooks(ProvidersManager().hooks.items(), args.output)
print(msg)
def providers_list(args):
"""Lists all providers at the command line"""
print(
_tabulate_providers(ProvidersManager().providers.values(),
args.output))
def test_providers_are_loaded(self):
provider_manager = ProvidersManager()
provider_list = list(provider_manager.providers.keys())
provider_list.sort()
self.assertEqual(ALL_PROVIDERS, provider_list)