Ejemplos de AzureDataFactoryHook._linked_service_exists en Python

Ejemplo n.º 1

def test_update_linked_service_non_existent(hook: AzureDataFactoryHook,
                                            user_args, sdk_args):
    hook._linked_service_exists = Mock(return_value=False)

    with pytest.raises(AirflowException,
                       match=r"Linked service .+ does not exist"):
        hook.update_linked_service(*user_args)

Ejemplo n.º 2

def test_update_linked_service(hook: AzureDataFactoryHook, user_args,
                               sdk_args):
    hook._linked_service_exists = Mock(return_value=True)
    hook.update_linked_service(*user_args)

    hook._conn.linked_services.create_or_update(*sdk_args)