You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@airflow.apache.org by po...@apache.org on 2024/10/15 02:43:15 UTC
(airflow) branch main updated: Move tests_common from "dev" to top-level. (#42985)
This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new cd1e9f5a04 Move tests_common from "dev" to top-level. (#42985)
cd1e9f5a04 is described below
commit cd1e9f5a0488ec53b382f1f639d80a3eba8ae307
Author: Jarek Potiuk <ja...@potiuk.com>
AuthorDate: Tue Oct 15 04:43:07 2024 +0200
Move tests_common from "dev" to top-level. (#42985)
* Move tests common without changes
* Fix docstrings in tests_common
* Move tests_common from "dev" to top-level.
Follow-up after #42505 fixing teething problem with tests_common.
Originally in #42505 common test code was moved to "dev" folder, but
the "dev" folder is really dedicated to "build" scripts and the
problem with moving "tests_common" to the folder was that the
whole "dev" folder is replaced (for non-committer PRs) with
the content from the target branch.
This is done for security reasons, because we can accidentally
use any of the scripts from dev in the CI build scripts and we
might not notice, which will open us to a security issue where
a file in "dev" coming from PR could be accidentally executed
during the "pull_request_target" workflow - which would expose
our secrets and GitHub Package write permissions to a
contributor coming from a fork.
This change moves the files, fixes pre-commit specification and
docs, also fixes a number of "doc" issues detected by "ruff" in
the tests_common folder as they were detected after the move.
The tests_common folder is added to folders mounted when
breeze is executed with local folders mounted (in order to
avoid accidental mounting of randomly generated files to
inside the breeze container).
All imports for the common tests were updated to reflect this
move.
---
.dockerignore | 1 +
.pre-commit-config.yaml | 8 ++--
Dockerfile.ci | 1 +
contributing-docs/testing/system_tests.rst | 2 +-
contributing-docs/testing/unit_tests.rst | 6 +--
.../airflow_breeze/utils/docker_command_utils.py | 1 +
.../src/airflow_breeze/utils/selective_checks.py | 2 +-
dev/breeze/tests/test_selective_checks.py | 4 +-
dev/perf/scheduler_dag_execution_timing.py | 2 +-
helm_tests/airflow_aux/test_pod_template_file.py | 1 +
.../alibaba/cloud/log/test_oss_task_handler.py | 5 +--
.../amazon/aws/auth_manager/avp/test_facade.py | 3 +-
.../aws/auth_manager/cli/test_avp_commands.py | 5 +--
.../test_aws_security_manager_override.py | 3 +-
.../aws/auth_manager/test_aws_auth_manager.py | 7 ++-
.../amazon/aws/auth_manager/views/test_auth.py | 5 +--
.../aws/executors/batch/test_batch_executor.py | 5 +--
.../amazon/aws/executors/ecs/test_ecs_executor.py | 7 ++-
providers/tests/amazon/aws/hooks/test_base_aws.py | 3 +-
providers/tests/amazon/aws/hooks/test_s3.py | 3 +-
providers/tests/amazon/aws/links/test_base_aws.py | 5 +--
.../amazon/aws/log/test_cloudwatch_task_handler.py | 3 +-
.../tests/amazon/aws/log/test_s3_task_handler.py | 3 +-
.../amazon/aws/secrets/test_systems_manager.py | 3 +-
.../amazon/aws/transfers/test_redshift_to_s3.py | 3 +-
.../amazon/aws/transfers/test_s3_to_redshift.py | 3 +-
.../tests/amazon/aws/transfers/test_s3_to_sftp.py | 3 +-
.../tests/amazon/aws/transfers/test_sftp_to_s3.py | 3 +-
providers/tests/apache/hive/hooks/test_hive.py | 2 +-
providers/tests/apache/livy/hooks/test_livy.py | 3 +-
.../tests/apache/spark/hooks/test_spark_sql.py | 3 +-
providers/tests/atlassian/jira/hooks/test_jira.py | 3 +-
.../tests/atlassian/jira/operators/test_jira.py | 3 +-
.../tests/atlassian/jira/sensors/test_jira.py | 3 +-
providers/tests/celery/cli/test_celery_command.py | 5 +--
.../tests/celery/executors/test_celery_executor.py | 7 ++-
.../tests/celery/log_handlers/test_log_handlers.py | 5 +--
.../cncf/kubernetes/cli/test_kubernetes_command.py | 3 +-
.../executors/test_kubernetes_executor.py | 5 +--
.../tests/cncf/kubernetes/hooks/test_kubernetes.py | 5 +--
.../kubernetes/log_handlers/test_log_handlers.py | 5 +--
.../tests/cncf/kubernetes/operators/test_pod.py | 3 +-
.../kubernetes/operators/test_spark_kubernetes.py | 3 +-
providers/tests/cncf/kubernetes/test_client.py | 3 +-
.../cncf/kubernetes/test_template_rendering.py | 3 +-
.../common/io/operators/test_file_transfer.py | 4 +-
providers/tests/common/io/xcom/test_backend.py | 11 +++--
providers/tests/common/sql/hooks/test_dbapi.py | 3 +-
providers/tests/common/sql/hooks/test_sql.py | 2 +-
providers/tests/common/sql/hooks/test_sqlparse.py | 3 +-
providers/tests/common/sql/operators/test_sql.py | 3 +-
.../tests/common/sql/operators/test_sql_execute.py | 3 +-
providers/tests/common/sql/sensors/test_sql.py | 3 +-
providers/tests/common/sql/test_utils.py | 3 +-
providers/tests/conftest.py | 2 +-
.../databricks/plugins/test_databricks_workflow.py | 3 +-
.../edge/api_endpoints/test_rpc_api_endpoint.py | 5 +--
providers/tests/edge/cli/test_edge_command.py | 3 +-
.../edge/plugins/test_edge_executor_plugin.py | 3 +-
.../elasticsearch/log/test_es_task_handler.py | 4 +-
.../api/auth/backend/test_basic_auth.py | 3 +-
.../api/auth/backend/test_kerberos_auth.py | 2 +-
.../auth_manager/api/auth/backend/test_session.py | 3 +-
.../api_endpoints/api_connexion_utils.py | 2 +-
.../api_endpoints/test_asset_endpoint.py | 6 +--
.../fab/auth_manager/api_endpoints/test_auth.py | 11 +++--
.../api_endpoints/test_backfill_endpoint.py | 14 +++---
.../fab/auth_manager/api_endpoints/test_cors.py | 7 ++-
.../api_endpoints/test_dag_endpoint.py | 6 +--
.../api_endpoints/test_dag_run_endpoint.py | 4 +-
.../api_endpoints/test_dag_source_endpoint.py | 4 +-
.../api_endpoints/test_dag_warning_endpoint.py | 4 +-
.../api_endpoints/test_event_log_endpoint.py | 4 +-
.../api_endpoints/test_import_error_endpoint.py | 6 +--
.../test_role_and_permission_endpoint.py | 4 +-
.../api_endpoints/test_task_instance_endpoint.py | 4 +-
.../api_endpoints/test_user_endpoint.py | 6 +--
.../api_endpoints/test_variable_endpoint.py | 4 +-
.../api_endpoints/test_xcom_endpoint.py | 4 +-
.../auth_manager/cli_commands/test_definition.py | 2 +-
.../auth_manager/cli_commands/test_role_command.py | 5 +--
.../cli_commands/test_sync_perm_command.py | 3 +-
.../auth_manager/cli_commands/test_user_command.py | 3 +-
.../fab/auth_manager/cli_commands/test_utils.py | 3 +-
providers/tests/fab/auth_manager/conftest.py | 5 +--
.../tests/fab/auth_manager/decorators/test_auth.py | 3 +-
.../fab/auth_manager/models/test_anonymous_user.py | 2 +-
.../fab/auth_manager/schemas/test_user_schema.py | 2 +-
.../security_manager/test_constants.py | 2 +-
.../auth_manager/security_manager/test_override.py | 2 +-
.../fab/auth_manager/test_fab_auth_manager.py | 2 +-
providers/tests/fab/auth_manager/test_models.py | 3 +-
providers/tests/fab/auth_manager/test_security.py | 12 ++---
.../fab/auth_manager/views/test_permissions.py | 4 +-
.../fab/auth_manager/views/test_roles_list.py | 4 +-
.../tests/fab/auth_manager/views/test_user.py | 4 +-
.../tests/fab/auth_manager/views/test_user_edit.py | 4 +-
.../fab/auth_manager/views/test_user_stats.py | 4 +-
.../google/cloud/hooks/test_bigquery_system.py | 2 +-
.../tests/google/cloud/hooks/test_kms_system.py | 2 +-
.../cloud/hooks/test_secret_manager_system.py | 2 +-
.../google/cloud/log/test_gcs_task_handler.py | 5 +--
.../cloud/log/test_gcs_task_handler_system.py | 14 +++---
.../cloud/log/test_stackdriver_task_handler.py | 7 ++-
.../log/test_stackdriver_task_handler_system.py | 14 +++---
.../tests/google/cloud/operators/test_bigquery.py | 13 +++---
.../google/cloud/operators/test_bigquery_dts.py | 3 +-
.../google/cloud/operators/test_dataprep_system.py | 7 ++-
.../tests/google/cloud/operators/test_dataproc.py | 5 +--
.../cloud/operators/test_datastore_system.py | 4 +-
.../tests/google/cloud/operators/test_looker.py | 3 +-
.../cloud/secrets/test_secret_manager_system.py | 2 +-
.../transfers/test_facebook_ads_to_gcs_system.py | 10 ++---
.../transfers/test_salesforce_to_gcs_system.py | 12 ++---
.../tests/google/cloud/utils/gcp_authenticator.py | 6 +--
.../common/auth_backend/test_google_openid.py | 9 ++--
.../operators/test_display_video_system.py | 12 ++---
providers/tests/hashicorp/hooks/test_vault.py | 3 +-
providers/tests/imap/hooks/test_imap.py | 3 +-
.../microsoft/azure/log/test_wasb_task_handler.py | 5 +--
providers/tests/mongo/hooks/test_mongo.py | 3 +-
providers/tests/mysql/hooks/test_mysql.py | 4 +-
.../tests/openlineage/extractors/test_bash.py | 3 +-
.../tests/openlineage/extractors/test_manager.py | 3 +-
.../tests/openlineage/extractors/test_python.py | 3 +-
.../tests/openlineage/plugins/test_adapter.py | 5 +--
.../tests/openlineage/plugins/test_execution.py | 5 +--
.../tests/openlineage/plugins/test_listener.py | 5 +--
.../tests/openlineage/plugins/test_openlineage.py | 5 +--
providers/tests/openlineage/plugins/test_utils.py | 3 +-
providers/tests/openlineage/test_conf.py | 3 +-
providers/tests/openlineage/utils/test_utils.py | 8 ++--
.../tests/opensearch/log/test_os_task_handler.py | 4 +-
.../tests/redis/log/test_redis_task_handler.py | 3 +-
providers/tests/sftp/operators/test_sftp.py | 3 +-
providers/tests/smtp/hooks/test_smtp.py | 3 +-
providers/tests/smtp/notifications/test_smtp.py | 5 +--
providers/tests/ssh/operators/test_ssh.py | 3 +-
providers/tests/standard/operators/test_bash.py | 3 +-
.../tests/standard/operators/test_datetime.py | 3 +-
providers/tests/standard/operators/test_weekday.py | 3 +-
providers/tests/standard/sensors/test_weekday.py | 3 +-
.../system/airbyte/example_airbyte_trigger_job.py | 2 +-
.../system/alibaba/example_adb_spark_batch.py | 4 +-
.../tests/system/alibaba/example_adb_spark_sql.py | 4 +-
.../tests/system/alibaba/example_oss_bucket.py | 4 +-
.../tests/system/alibaba/example_oss_object.py | 4 +-
providers/tests/system/amazon/CONTRIBUTING.md | 2 +-
.../tests/system/amazon/aws/example_appflow.py | 4 +-
.../tests/system/amazon/aws/example_appflow_run.py | 4 +-
.../tests/system/amazon/aws/example_athena.py | 4 +-
.../system/amazon/aws/example_azure_blob_to_s3.py | 4 +-
providers/tests/system/amazon/aws/example_batch.py | 4 +-
.../tests/system/amazon/aws/example_bedrock.py | 4 +-
.../aws/example_bedrock_retrieve_and_generate.py | 4 +-
.../system/amazon/aws/example_cloudformation.py | 4 +-
.../tests/system/amazon/aws/example_comprehend.py | 4 +-
.../aws/example_comprehend_document_classifier.py | 4 +-
.../tests/system/amazon/aws/example_datasync.py | 4 +-
providers/tests/system/amazon/aws/example_dms.py | 4 +-
.../tests/system/amazon/aws/example_dynamodb.py | 4 +-
.../system/amazon/aws/example_dynamodb_to_s3.py | 4 +-
providers/tests/system/amazon/aws/example_ec2.py | 4 +-
providers/tests/system/amazon/aws/example_ecs.py | 4 +-
.../tests/system/amazon/aws/example_ecs_fargate.py | 4 +-
.../system/amazon/aws/example_eks_templated.py | 4 +-
.../aws/example_eks_with_fargate_in_one_step.py | 4 +-
.../amazon/aws/example_eks_with_fargate_profile.py | 4 +-
.../aws/example_eks_with_nodegroup_in_one_step.py | 4 +-
.../amazon/aws/example_eks_with_nodegroups.py | 4 +-
providers/tests/system/amazon/aws/example_emr.py | 4 +-
.../tests/system/amazon/aws/example_emr_eks.py | 4 +-
.../amazon/aws/example_emr_notebook_execution.py | 4 +-
.../system/amazon/aws/example_emr_serverless.py | 4 +-
.../tests/system/amazon/aws/example_eventbridge.py | 2 +-
.../tests/system/amazon/aws/example_ftp_to_s3.py | 4 +-
.../tests/system/amazon/aws/example_gcs_to_s3.py | 4 +-
.../system/amazon/aws/example_glacier_to_gcs.py | 4 +-
providers/tests/system/amazon/aws/example_glue.py | 4 +-
.../system/amazon/aws/example_glue_data_quality.py | 4 +-
...xample_glue_data_quality_with_recommendation.py | 4 +-
.../system/amazon/aws/example_glue_databrew.py | 4 +-
.../amazon/aws/example_google_api_sheets_to_s3.py | 4 +-
.../amazon/aws/example_google_api_youtube_to_s3.py | 4 +-
.../system/amazon/aws/example_hive_to_dynamodb.py | 4 +-
.../tests/system/amazon/aws/example_http_to_s3.py | 4 +-
.../amazon/aws/example_imap_attachment_to_s3.py | 4 +-
.../system/amazon/aws/example_kinesis_analytics.py | 4 +-
.../tests/system/amazon/aws/example_lambda.py | 4 +-
.../tests/system/amazon/aws/example_local_to_s3.py | 4 +-
.../tests/system/amazon/aws/example_mongo_to_s3.py | 4 +-
.../tests/system/amazon/aws/example_neptune.py | 4 +-
.../tests/system/amazon/aws/example_quicksight.py | 4 +-
.../tests/system/amazon/aws/example_rds_event.py | 4 +-
.../tests/system/amazon/aws/example_rds_export.py | 4 +-
.../system/amazon/aws/example_rds_instance.py | 4 +-
.../system/amazon/aws/example_rds_snapshot.py | 4 +-
.../tests/system/amazon/aws/example_redshift.py | 4 +-
.../amazon/aws/example_redshift_s3_transfers.py | 4 +-
providers/tests/system/amazon/aws/example_s3.py | 4 +-
.../system/amazon/aws/example_s3_to_dynamodb.py | 4 +-
.../tests/system/amazon/aws/example_s3_to_ftp.py | 4 +-
.../tests/system/amazon/aws/example_s3_to_sftp.py | 4 +-
.../tests/system/amazon/aws/example_s3_to_sql.py | 5 ++-
.../tests/system/amazon/aws/example_sagemaker.py | 4 +-
.../amazon/aws/example_sagemaker_endpoint.py | 4 +-
.../amazon/aws/example_sagemaker_notebook.py | 4 +-
.../amazon/aws/example_sagemaker_pipeline.py | 4 +-
.../system/amazon/aws/example_salesforce_to_s3.py | 4 +-
.../tests/system/amazon/aws/example_sftp_to_s3.py | 4 +-
providers/tests/system/amazon/aws/example_sns.py | 4 +-
.../tests/system/amazon/aws/example_sql_to_s3.py | 4 +-
providers/tests/system/amazon/aws/example_sqs.py | 4 +-
.../system/amazon/aws/example_step_functions.py | 4 +-
.../amazon/aws/tests/test_aws_auth_manager.py | 4 +-
providers/tests/system/apache/beam/example_beam.py | 2 +-
.../system/apache/beam/example_beam_java_flink.py | 2 +-
.../system/apache/beam/example_beam_java_spark.py | 2 +-
providers/tests/system/apache/beam/example_go.py | 2 +-
.../system/apache/beam/example_go_dataflow.py | 2 +-
.../system/apache/beam/example_java_dataflow.py | 2 +-
.../tests/system/apache/beam/example_python.py | 2 +-
.../system/apache/beam/example_python_async.py | 2 +-
.../system/apache/beam/example_python_dataflow.py | 2 +-
.../apache/cassandra/example_cassandra_dag.py | 2 +-
.../tests/system/apache/drill/example_drill_dag.py | 2 +-
.../tests/system/apache/druid/example_druid_dag.py | 2 +-
.../system/apache/hive/example_twitter_dag.py | 4 +-
.../tests/system/apache/iceberg/example_iceberg.py | 2 +-
.../apache/kafka/example_dag_event_listener.py | 2 +-
.../system/apache/kafka/example_dag_hello_kafka.py | 2 +-
.../tests/system/apache/kylin/example_kylin_dag.py | 4 +-
providers/tests/system/apache/livy/example_livy.py | 4 +-
providers/tests/system/apache/pig/example_pig.py | 2 +-
.../tests/system/apache/pinot/example_pinot_dag.py | 2 +-
.../tests/system/apache/spark/example_pyspark.py | 2 +-
.../tests/system/apache/spark/example_spark_dag.py | 2 +-
providers/tests/system/asana/example_asana.py | 4 +-
.../system/cncf/kubernetes/example_kubernetes.py | 4 +-
.../cncf/kubernetes/example_kubernetes_async.py | 4 +-
.../kubernetes/example_kubernetes_decorator.py | 2 +-
.../cncf/kubernetes/example_kubernetes_job.py | 4 +-
.../cncf/kubernetes/example_kubernetes_resource.py | 4 +-
.../cncf/kubernetes/example_spark_kubernetes.py | 4 +-
.../cohere/example_cohere_embedding_operator.py | 2 +-
.../common/io/example_file_transfer_local_to_s3.py | 4 +-
.../common/sql/example_sql_column_table_check.py | 2 +-
.../system/common/sql/example_sql_execute_query.py | 2 +-
.../tests/system/databricks/example_databricks.py | 4 +-
.../system/databricks/example_databricks_repos.py | 4 +-
.../databricks/example_databricks_sensors.py | 4 +-
.../system/databricks/example_databricks_sql.py | 4 +-
.../databricks/example_databricks_workflow.py | 4 +-
.../tests/system/dbt/cloud/example_dbt_cloud.py | 8 ++--
.../tests/system/dingding/example_dingding.py | 4 +-
providers/tests/system/docker/example_docker.py | 2 +-
.../system/docker/example_docker_copy_data.py | 2 +-
.../tests/system/docker/example_docker_swarm.py | 2 +-
.../example_taskflow_api_docker_virtualenv.py | 8 +++-
.../elasticsearch/example_elasticsearch_query.py | 2 +-
providers/tests/system/ftp/example_ftp.py | 4 +-
providers/tests/system/github/example_github.py | 2 +-
providers/tests/system/google/ads/example_ads.py | 4 +-
.../google/cloud/automl/example_automl_dataset.py | 4 +-
.../cloud/automl/example_automl_translation.py | 4 +-
.../automl/example_automl_video_classification.py | 4 +-
.../cloud/automl/example_automl_video_tracking.py | 4 +-
.../automl/example_automl_vision_classification.py | 4 +-
.../example_automl_vision_object_detection.py | 4 +-
.../cloud/azure/example_azure_blob_to_gcs.py | 4 +-
.../cloud/azure/example_azure_fileshare_to_gcs.py | 4 +-
.../cloud/bigquery/example_bigquery_dataset.py | 4 +-
.../google/cloud/bigquery/example_bigquery_dts.py | 4 +-
.../cloud/bigquery/example_bigquery_operations.py | 4 +-
.../example_bigquery_operations_location.py | 4 +-
.../cloud/bigquery/example_bigquery_queries.py | 4 +-
.../bigquery/example_bigquery_queries_async.py | 4 +-
.../cloud/bigquery/example_bigquery_sensors.py | 4 +-
.../cloud/bigquery/example_bigquery_tables.py | 4 +-
.../cloud/bigquery/example_bigquery_to_bigquery.py | 4 +-
.../cloud/bigquery/example_bigquery_to_gcs.py | 4 +-
.../bigquery/example_bigquery_to_gcs_async.py | 4 +-
.../cloud/bigquery/example_bigquery_to_mssql.py | 4 +-
.../cloud/bigquery/example_bigquery_to_mysql.py | 4 +-
.../cloud/bigquery/example_bigquery_to_postgres.py | 4 +-
.../cloud/bigquery/example_bigquery_transfer.py | 4 +-
.../cloud/bigquery/example_bigquery_value_check.py | 6 +--
.../google/cloud/bigtable/example_bigtable.py | 4 +-
.../cloud/cloud_batch/example_cloud_batch.py | 4 +-
.../cloud/cloud_build/example_cloud_build.py | 4 +-
.../cloud_build/example_cloud_build_trigger.py | 4 +-
.../cloud/cloud_functions/example_functions.py | 4 +-
.../example_cloud_memorystore_memcached.py | 4 +-
.../example_cloud_memorystore_redis.py | 4 +-
.../google/cloud/cloud_run/example_cloud_run.py | 4 +-
.../cloud/cloud_run/example_cloud_run_service.py | 4 +-
.../google/cloud/cloud_sql/example_cloud_sql.py | 4 +-
.../cloud/cloud_sql/example_cloud_sql_query.py | 4 +-
.../cloud/cloud_sql/example_cloud_sql_query_ssl.py | 4 +-
.../cloud/composer/example_cloud_composer.py | 4 +-
.../system/google/cloud/compute/example_compute.py | 4 +-
.../google/cloud/compute/example_compute_igm.py | 4 +-
.../google/cloud/compute/example_compute_ssh.py | 4 +-
.../cloud/compute/example_compute_ssh_os_login.py | 4 +-
.../cloud/compute/example_compute_ssh_parallel.py | 4 +-
.../example_dlp_deidentify_content.py | 4 +-
.../data_loss_prevention/example_dlp_info_types.py | 4 +-
.../example_dlp_inspect_template.py | 4 +-
.../cloud/data_loss_prevention/example_dlp_job.py | 4 +-
.../example_dlp_job_trigger.py | 4 +-
.../google/cloud/dataflow/example_dataflow_go.py | 2 +-
.../cloud/dataflow/example_dataflow_native_java.py | 4 +-
.../dataflow/example_dataflow_native_python.py | 4 +-
.../example_dataflow_native_python_async.py | 4 +-
.../cloud/dataflow/example_dataflow_pipeline.py | 4 +-
.../example_dataflow_sensors_deferrable.py | 4 +-
.../google/cloud/dataflow/example_dataflow_sql.py | 4 +-
.../dataflow/example_dataflow_streaming_python.py | 4 +-
.../cloud/dataflow/example_dataflow_template.py | 4 +-
.../google/cloud/dataflow/example_dataflow_yaml.py | 4 +-
.../google/cloud/dataform/example_dataform.py | 4 +-
.../google/cloud/datafusion/example_datafusion.py | 4 +-
.../cloud/datapipelines/example_datapipeline.py | 4 +-
.../google/cloud/dataplex/example_dataplex.py | 4 +-
.../google/cloud/dataplex/example_dataplex_dp.py | 4 +-
.../google/cloud/dataplex/example_dataplex_dq.py | 4 +-
.../google/cloud/dataprep/example_dataprep.py | 4 +-
.../cloud/dataproc/example_dataproc_batch.py | 4 +-
.../dataproc/example_dataproc_batch_deferrable.py | 4 +-
.../dataproc/example_dataproc_batch_persistent.py | 4 +-
...proc_cluster_create_existing_stopped_cluster.py | 4 +-
.../example_dataproc_cluster_deferrable.py | 4 +-
.../dataproc/example_dataproc_cluster_diagnose.py | 4 +-
.../dataproc/example_dataproc_cluster_generator.py | 4 +-
.../example_dataproc_cluster_start_stop.py | 4 +-
.../dataproc/example_dataproc_cluster_update.py | 4 +-
.../cloud/dataproc/example_dataproc_flink.py | 4 +-
.../google/cloud/dataproc/example_dataproc_gke.py | 4 +-
.../cloud/dataproc/example_dataproc_hadoop.py | 4 +-
.../google/cloud/dataproc/example_dataproc_hive.py | 4 +-
.../google/cloud/dataproc/example_dataproc_pig.py | 4 +-
.../cloud/dataproc/example_dataproc_presto.py | 4 +-
.../cloud/dataproc/example_dataproc_pyspark.py | 4 +-
.../cloud/dataproc/example_dataproc_spark.py | 4 +-
.../cloud/dataproc/example_dataproc_spark_async.py | 4 +-
.../dataproc/example_dataproc_spark_deferrable.py | 4 +-
.../cloud/dataproc/example_dataproc_spark_sql.py | 4 +-
.../cloud/dataproc/example_dataproc_sparkr.py | 4 +-
.../cloud/dataproc/example_dataproc_trino.py | 4 +-
.../cloud/dataproc/example_dataproc_workflow.py | 4 +-
.../example_dataproc_workflow_deferrable.py | 4 +-
.../example_dataproc_metastore.py | 4 +-
.../example_dataproc_metastore_backup.py | 4 +-
...ple_dataproc_metastore_hive_partition_sensor.py | 4 +-
.../cloud/datastore/example_datastore_commit.py | 4 +-
.../cloud/datastore/example_datastore_query.py | 4 +-
.../cloud/datastore/example_datastore_rollback.py | 4 +-
.../google/cloud/gcs/example_calendar_to_gcs.py | 4 +-
.../system/google/cloud/gcs/example_firestore.py | 4 +-
.../system/google/cloud/gcs/example_gcs_acl.py | 4 +-
.../google/cloud/gcs/example_gcs_copy_delete.py | 4 +-
.../system/google/cloud/gcs/example_gcs_sensor.py | 4 +-
.../google/cloud/gcs/example_gcs_to_bigquery.py | 4 +-
.../cloud/gcs/example_gcs_to_bigquery_async.py | 4 +-
.../system/google/cloud/gcs/example_gcs_to_gcs.py | 4 +-
.../google/cloud/gcs/example_gcs_to_gdrive.py | 4 +-
.../google/cloud/gcs/example_gcs_to_sheets.py | 4 +-
.../google/cloud/gcs/example_gcs_transform.py | 4 +-
.../cloud/gcs/example_gcs_transform_timespan.py | 4 +-
.../cloud/gcs/example_gcs_upload_download.py | 4 +-
.../google/cloud/gcs/example_gdrive_to_gcs.py | 4 +-
.../google/cloud/gcs/example_mssql_to_gcs.py | 4 +-
.../google/cloud/gcs/example_mysql_to_gcs.py | 4 +-
.../google/cloud/gcs/example_oracle_to_gcs.py | 4 +-
.../system/google/cloud/gcs/example_s3_to_gcs.py | 4 +-
.../system/google/cloud/gcs/example_sftp_to_gcs.py | 4 +-
.../system/google/cloud/gcs/example_sheets.py | 4 +-
.../google/cloud/gcs/example_sheets_to_gcs.py | 4 +-
.../google/cloud/gcs/example_trino_to_gcs.py | 4 +-
.../kubernetes_engine/example_kubernetes_engine.py | 4 +-
.../example_kubernetes_engine_async.py | 4 +-
.../example_kubernetes_engine_job.py | 4 +-
.../example_kubernetes_engine_kueue.py | 4 +-
.../example_kubernetes_engine_resource.py | 4 +-
.../cloud/life_sciences/example_life_sciences.py | 4 +-
.../google/cloud/ml_engine/example_mlengine.py | 4 +-
.../natural_language/example_natural_language.py | 4 +-
.../system/google/cloud/pubsub/example_pubsub.py | 4 +-
.../cloud/pubsub/example_pubsub_deferrable.py | 4 +-
.../system/google/cloud/spanner/example_spanner.py | 4 +-
.../cloud/speech_to_text/example_speech_to_text.py | 4 +-
.../cloud/sql_to_sheets/example_sql_to_sheets.py | 4 +-
.../cloud/stackdriver/example_stackdriver.py | 4 +-
.../example_cloud_storage_transfer_service_aws.py | 4 +-
.../example_cloud_storage_transfer_service_gcp.py | 4 +-
...le_cloud_storage_transfer_service_gcs_to_gcs.py | 4 +-
.../system/google/cloud/tasks/example_queue.py | 4 +-
.../system/google/cloud/tasks/example_tasks.py | 4 +-
.../cloud/text_to_speech/example_text_to_speech.py | 4 +-
.../google/cloud/transfers/example_gcs_to_sftp.py | 4 +-
.../cloud/transfers/example_gdrive_to_local.py | 4 +-
.../cloud/transfers/example_postgres_to_gcs.py | 4 +-
.../google/cloud/translate/example_translate.py | 4 +-
.../translate_speech/example_translate_speech.py | 4 +-
...ample_vertex_ai_auto_ml_forecasting_training.py | 4 +-
.../example_vertex_ai_auto_ml_image_training.py | 4 +-
.../example_vertex_ai_auto_ml_list_training.py | 4 +-
.../example_vertex_ai_auto_ml_tabular_training.py | 4 +-
.../example_vertex_ai_auto_ml_video_training.py | 4 +-
.../example_vertex_ai_batch_prediction_job.py | 4 +-
.../example_vertex_ai_custom_container.py | 4 +-
.../vertex_ai/example_vertex_ai_custom_job.py | 4 +-
.../example_vertex_ai_custom_job_python_package.py | 4 +-
.../cloud/vertex_ai/example_vertex_ai_dataset.py | 4 +-
.../cloud/vertex_ai/example_vertex_ai_endpoint.py | 4 +-
.../example_vertex_ai_generative_model.py | 4 +-
.../example_vertex_ai_generative_model_tuning.py | 4 +-
.../example_vertex_ai_hyperparameter_tuning_job.py | 4 +-
.../example_vertex_ai_list_custom_jobs.py | 4 +-
.../vertex_ai/example_vertex_ai_model_service.py | 4 +-
.../vertex_ai/example_vertex_ai_pipeline_job.py | 4 +-
.../example_video_intelligence.py | 4 +-
.../cloud/vision/example_vision_annotate_image.py | 4 +-
.../cloud/vision/example_vision_autogenerated.py | 4 +-
.../google/cloud/vision/example_vision_explicit.py | 4 +-
.../google/cloud/workflows/example_workflows.py | 4 +-
.../datacatalog/example_datacatalog_entries.py | 4 +-
.../example_datacatalog_search_catalog.py | 4 +-
.../example_datacatalog_tag_templates.py | 4 +-
.../google/datacatalog/example_datacatalog_tags.py | 4 +-
.../tests/system/google/leveldb/example_leveldb.py | 4 +-
.../marketing_platform/example_analytics_admin.py | 4 +-
.../marketing_platform/example_campaign_manager.py | 4 +-
.../marketing_platform/example_search_ads.py | 2 +-
.../system/google/suite/example_local_to_drive.py | 4 +-
providers/tests/system/http/example_http.py | 2 +-
.../tests/system/influxdb/example_influxdb.py | 4 +-
.../system/influxdb/example_influxdb_query.py | 2 +-
.../tests/system/jdbc/example_jdbc_queries.py | 4 +-
.../system/jenkins/example_jenkins_job_trigger.py | 2 +-
.../microsoft/azure/example_adf_run_pipeline.py | 4 +-
.../system/microsoft/azure/example_adls_create.py | 4 +-
.../system/microsoft/azure/example_adls_delete.py | 4 +-
.../system/microsoft/azure/example_adls_list.py | 4 +-
.../azure/example_azure_batch_operator.py | 2 +-
.../azure/example_azure_container_instances.py | 2 +-
.../microsoft/azure/example_azure_cosmosdb.py | 4 +-
.../microsoft/azure/example_azure_service_bus.py | 4 +-
.../microsoft/azure/example_azure_synapse.py | 2 +-
.../system/microsoft/azure/example_fileshare.py | 4 +-
.../microsoft/azure/example_local_to_adls.py | 4 +-
.../microsoft/azure/example_local_to_wasb.py | 4 +-
.../system/microsoft/azure/example_msfabric.py | 4 +-
.../system/microsoft/azure/example_msgraph.py | 4 +-
.../system/microsoft/azure/example_powerbi.py | 4 +-
.../azure/example_powerbi_dataset_refresh.py | 4 +-
.../system/microsoft/azure/example_s3_to_wasb.py | 4 +-
.../system/microsoft/azure/example_sftp_to_wasb.py | 4 +-
.../azure/example_synapse_run_pipeline.py | 4 +-
.../system/microsoft/azure/example_wasb_sensors.py | 2 +-
.../tests/system/microsoft/mssql/example_mssql.py | 4 +-
.../tests/system/microsoft/winrm/example_winrm.py | 4 +-
providers/tests/system/mysql/example_mysql.py | 2 +-
providers/tests/system/neo4j/example_neo4j.py | 2 +-
providers/tests/system/openai/example_openai.py | 2 +-
.../openai/example_trigger_batch_operator.py | 2 +-
.../tests/system/opensearch/example_opensearch.py | 4 +-
.../system/opsgenie/example_opsgenie_alert.py | 2 +-
.../system/opsgenie/example_opsgenie_notifier.py | 2 +-
.../tests/system/papermill/example_papermill.py | 2 +-
.../papermill/example_papermill_remote_verify.py | 2 +-
.../system/papermill/example_papermill_verify.py | 2 +-
.../tests/system/pgvector/example_pgvector.py | 2 +-
.../system/pgvector/example_pgvector_openai.py | 2 +-
.../system/pinecone/example_create_pod_index.py | 2 +-
.../pinecone/example_create_serverless_index.py | 2 +-
.../tests/system/pinecone/example_dag_pinecone.py | 2 +-
.../system/pinecone/example_pinecone_cohere.py | 2 +-
.../system/pinecone/example_pinecone_openai.py | 2 +-
.../tests/system/postgres/example_postgres.py | 4 +-
.../tests/system/presto/example_gcs_to_presto.py | 2 +-
.../tests/system/qdrant/example_dag_qdrant.py | 2 +-
.../tests/system/redis/example_redis_publish.py | 4 +-
providers/tests/system/salesforce/example_bulk.py | 2 +-
.../salesforce/example_salesforce_apex_rest.py | 2 +-
.../tests/system/samba/example_gcs_to_samba.py | 4 +-
providers/tests/system/sftp/example_sftp_sensor.py | 4 +-
.../system/singularity/example_singularity.py | 2 +-
providers/tests/system/slack/example_slack.py | 2 +-
.../tests/system/slack/example_slack_webhook.py | 2 +-
.../tests/system/slack/example_sql_to_slack.py | 2 +-
.../system/slack/example_sql_to_slack_webhook.py | 2 +-
.../snowflake/example_copy_into_snowflake.py | 2 +-
.../tests/system/snowflake/example_snowflake.py | 2 +-
.../system/snowflake/example_snowpark_decorator.py | 2 +-
.../system/snowflake/example_snowpark_operator.py | 2 +-
providers/tests/system/sqlite/example_sqlite.py | 4 +-
providers/tests/system/tableau/example_tableau.py | 2 +-
.../tests/system/telegram/example_telegram.py | 2 +-
.../example_azure_blob_to_teradata_transfer.py | 4 +-
.../teradata/example_s3_to_teradata_transfer.py | 4 +-
.../tests/system/teradata/example_ssl_teradata.py | 4 +-
.../tests/system/teradata/example_teradata.py | 4 +-
.../system/teradata/example_teradata_call_sp.py | 4 +-
.../teradata/example_teradata_compute_cluster.py | 4 +-
.../example_teradata_to_teradata_transfer.py | 4 +-
.../tests/system/trino/example_gcs_to_trino.py | 2 +-
providers/tests/system/trino/example_trino.py | 2 +-
.../system/weaviate/example_weaviate_cohere.py | 2 +-
.../example_weaviate_dynamic_mapping_dag.py | 2 +-
.../system/weaviate/example_weaviate_openai.py | 2 +-
.../system/weaviate/example_weaviate_operator.py | 2 +-
.../system/weaviate/example_weaviate_using_hook.py | 2 +-
.../weaviate/example_weaviate_vectorizer_dag.py | 2 +-
.../example_weaviate_without_vectorizer_dag.py | 2 +-
.../tests/system/yandex/example_yandexcloud.py | 7 ++-
.../system/yandex/example_yandexcloud_dataproc.py | 8 ++--
.../example_yandexcloud_dataproc_lightweight.py | 8 ++--
.../tests/system/yandex/example_yandexcloud_yq.py | 8 ++--
providers/tests/system/ydb/example_ydb.py | 4 +-
.../system/zendesk/example_zendesk_custom_get.py | 2 +-
providers/tests/yandex/hooks/test_yandex.py | 4 +-
providers/tests/yandex/links/test_yq.py | 5 +--
providers/tests/yandex/operators/test_yq.py | 3 +-
pyproject.toml | 6 +--
scripts/ci/docker-compose/local.yml | 3 ++
scripts/ci/pre_commit/check_system_tests.py | 8 ++--
scripts/in_container/install_devel_deps.py | 2 +-
task_sdk/tests/conftest.py | 2 +-
tests/always/test_connection.py | 3 +-
tests/always/test_example_dags.py | 3 +-
tests/always/test_secrets.py | 5 +--
tests/always/test_secrets_backends.py | 3 +-
tests/always/test_secrets_local_filesystem.py | 3 +-
tests/api_connexion/conftest.py | 7 ++-
.../api_connexion/endpoints/test_asset_endpoint.py | 11 +++--
.../endpoints/test_backfill_endpoint.py | 5 +--
.../endpoints/test_config_endpoint.py | 4 +-
.../endpoints/test_connection_endpoint.py | 9 ++--
tests/api_connexion/endpoints/test_dag_endpoint.py | 9 ++--
tests/api_connexion/endpoints/test_dag_parsing.py | 5 +--
.../endpoints/test_dag_run_endpoint.py | 11 +++--
.../endpoints/test_dag_source_endpoint.py | 5 +--
.../endpoints/test_dag_stats_endpoint.py | 5 +--
.../endpoints/test_dag_warning_endpoint.py | 5 +--
.../endpoints/test_event_log_endpoint.py | 7 ++-
.../endpoints/test_extra_link_endpoint.py | 11 +++--
.../endpoints/test_import_error_endpoint.py | 9 ++--
tests/api_connexion/endpoints/test_log_endpoint.py | 5 +--
.../test_mapped_task_instance_endpoint.py | 7 ++-
.../endpoints/test_plugin_endpoint.py | 9 ++--
.../api_connexion/endpoints/test_pool_endpoint.py | 9 ++--
.../endpoints/test_provider_endpoint.py | 3 +-
.../api_connexion/endpoints/test_task_endpoint.py | 5 +--
.../endpoints/test_task_instance_endpoint.py | 7 ++-
.../endpoints/test_variable_endpoint.py | 9 ++--
.../api_connexion/endpoints/test_xcom_endpoint.py | 7 ++-
.../schemas/test_connection_schema.py | 3 +-
tests/api_connexion/schemas/test_dag_run_schema.py | 5 +--
tests/api_connexion/schemas/test_dataset_schema.py | 3 +-
tests/api_connexion/schemas/test_error_schema.py | 5 +--
tests/api_connexion/schemas/test_plugin_schema.py | 3 +-
tests/api_connexion/schemas/test_pool_schemas.py | 3 +-
tests/api_connexion/schemas/test_xcom_schema.py | 3 +-
tests/api_connexion/test_auth.py | 7 ++-
tests/api_connexion/test_parameters.py | 3 +-
tests/api_connexion/test_security.py | 2 +-
tests/api_fastapi/views/public/test_connections.py | 3 +-
tests/api_fastapi/views/public/test_dag_run.py | 3 +-
tests/api_fastapi/views/public/test_dags.py | 3 +-
tests/api_fastapi/views/public/test_variables.py | 3 +-
tests/api_fastapi/views/ui/test_assets.py | 3 +-
tests/api_fastapi/views/ui/test_dashboard.py | 3 +-
.../endpoints/test_rpc_api_endpoint.py | 5 +--
tests/api_internal/test_internal_api_call.py | 3 +-
tests/assets/test_manager.py | 3 +-
tests/assets/tests_asset.py | 2 +-
tests/auth/managers/simple/views/test_auth.py | 3 +-
tests/cli/commands/test_backfill_command.py | 3 +-
tests/cli/commands/test_celery_command.py | 3 +-
tests/cli/commands/test_config_command.py | 3 +-
tests/cli/commands/test_connection_command.py | 3 +-
tests/cli/commands/test_dag_command.py | 6 +--
tests/cli/commands/test_dag_processor_command.py | 3 +-
tests/cli/commands/test_fastapi_api_command.py | 1 +
tests/cli/commands/test_info_command.py | 3 +-
tests/cli/commands/test_internal_api_command.py | 4 +-
tests/cli/commands/test_jobs_command.py | 3 +-
tests/cli/commands/test_kerberos_command.py | 3 +-
tests/cli/commands/test_kubernetes_command.py | 3 +-
tests/cli/commands/test_plugins_command.py | 16 +++----
.../cli/commands/test_rotate_fernet_key_command.py | 5 +--
tests/cli/commands/test_scheduler_command.py | 3 +-
tests/cli/commands/test_task_command.py | 7 ++-
tests/cli/commands/test_variable_command.py | 3 +-
tests/cli/commands/test_webserver_command.py | 4 +-
tests/cli/conftest.py | 3 +-
tests/cli/test_cli_parser.py | 3 +-
tests/conftest.py | 4 +-
tests/core/test_configuration.py | 6 +--
tests/core/test_core.py | 3 +-
tests/core/test_example_dags_system.py | 5 +--
tests/core/test_impersonation_tests.py | 3 +-
tests/core/test_logging_config.py | 3 +-
tests/core/test_otel_tracer.py | 3 +-
tests/core/test_sentry.py | 3 +-
tests/core/test_settings.py | 3 +-
tests/core/test_sqlalchemy_config.py | 3 +-
tests/core/test_stats.py | 3 +-
tests/dag_processing/test_job_runner.py | 12 ++---
tests/dag_processing/test_processor.py | 11 +++--
.../test_external_task_sensor_check_existense.py | 1 +
tests/dags/test_miscellaneous.py | 3 +-
tests/dags/test_sensor.py | 3 +-
tests/decorators/test_bash.py | 3 +-
tests/decorators/test_mapped.py | 1 +
tests/decorators/test_python.py | 4 +-
tests/executors/test_executor_loader.py | 3 +-
.../cli/commands/test_celery_command.py | 3 +-
.../integration/executors/test_celery_executor.py | 3 +-
tests/integration/security/test_kerberos.py | 3 +-
tests/jobs/test_base_job.py | 4 +-
tests/jobs/test_local_task_job.py | 11 +++--
tests/jobs/test_scheduler_job.py | 22 +++++-----
tests/jobs/test_triggerer_job.py | 4 +-
tests/jobs/test_triggerer_job_logging.py | 3 +-
tests/lineage/test_hook.py | 3 +-
tests/lineage/test_lineage.py | 3 +-
tests/listeners/class_listener.py | 3 +-
tests/listeners/test_asset_listener.py | 1 +
tests/listeners/test_dag_import_error_listener.py | 10 ++---
tests/listeners/test_listeners.py | 1 +
tests/models/test_backfill.py | 3 +-
tests/models/test_base.py | 3 +-
tests/models/test_baseoperator.py | 4 +-
tests/models/test_cleartasks.py | 6 +--
tests/models/test_dag.py | 26 +++++------
tests/models/test_dagbag.py | 8 ++--
tests/models/test_dagcode.py | 3 +-
tests/models/test_dagrun.py | 10 ++---
tests/models/test_dagwarning.py | 3 +-
tests/models/test_mappedoperator.py | 8 ++--
tests/models/test_param.py | 3 +-
tests/models/test_pool.py | 3 +-
tests/models/test_renderedtifields.py | 5 +--
tests/models/test_serialized_dag.py | 5 +--
tests/models/test_skipmixin.py | 3 +-
tests/models/test_taskinstance.py | 12 ++---
tests/models/test_timestamp.py | 3 +-
tests/models/test_trigger.py | 3 +-
tests/models/test_variable.py | 5 +--
tests/models/test_xcom.py | 5 +--
tests/models/test_xcom_arg.py | 3 +-
tests/operators/test_branch_operator.py | 3 +-
tests/operators/test_email.py | 3 +-
tests/operators/test_latest_only_operator.py | 5 +--
tests/operators/test_python.py | 7 ++-
tests/plugins/test_plugin.py | 8 ++--
tests/plugins/test_plugins_manager.py | 5 +--
tests/secrets/test_cache.py | 3 +-
tests/security/test_kerberos.py | 3 +-
tests/sensors/test_base.py | 3 +-
tests/sensors/test_external_task_sensor.py | 8 ++--
tests/sensors/test_python.py | 1 +
tests/serialization/test_dag_serialization.py | 37 ++++++++--------
tests/serialization/test_pydantic_models.py | 4 +-
tests/serialization/test_serde.py | 3 +-
tests/serialization/test_serialized_objects.py | 4 +-
.../core/example_external_task_child_deferrable.py | 2 +-
.../example_external_task_parent_deferrable.py | 4 +-
tests/system/example_empty.py | 4 +-
.../task/task_runner/test_standard_task_runner.py | 6 +--
.../ti_deps/deps/test_pool_slots_available_dep.py | 3 +-
tests/ti_deps/deps/test_prev_dagrun_dep.py | 5 +--
tests/ti_deps/deps/test_ready_to_reschedule_dep.py | 3 +-
tests/utils/log/test_log_reader.py | 5 +--
tests/utils/log/test_secrets_masker.py | 3 +-
tests/utils/log/test_task_context_logger.py | 5 +--
tests/utils/test_dag_cycle.py | 1 +
tests/utils/test_db.py | 3 +-
tests/utils/test_db_cleanup.py | 3 +-
tests/utils/test_db_manager.py | 3 +-
tests/utils/test_dot_renderer.py | 5 +--
tests/utils/test_email.py | 3 +-
tests/utils/test_file.py | 4 +-
tests/utils/test_helpers.py | 5 +--
tests/utils/test_log_handlers.py | 5 +--
tests/utils/test_net.py | 3 +-
tests/utils/test_serve_logs.py | 3 +-
tests/utils/test_sqlalchemy.py | 3 +-
tests/utils/test_state.py | 4 +-
tests/utils/test_task_group.py | 6 +--
.../test_task_handler_with_custom_formatter.py | 7 ++-
tests/utils/test_types.py | 4 +-
tests/www/test_app.py | 5 +--
tests/www/test_utils.py | 3 +-
tests/www/views/conftest.py | 9 ++--
tests/www/views/test_anonymous_as_admin_role.py | 3 +-
tests/www/views/test_session.py | 5 +--
tests/www/views/test_views.py | 15 +++----
tests/www/views/test_views_acl.py | 10 ++---
tests/www/views/test_views_base.py | 7 ++-
tests/www/views/test_views_cluster_activity.py | 3 +-
tests/www/views/test_views_configuration.py | 5 +--
tests/www/views/test_views_connection.py | 3 +-
tests/www/views/test_views_custom_user_views.py | 4 +-
tests/www/views/test_views_dagrun.py | 8 ++--
tests/www/views/test_views_dataset.py | 5 +--
tests/www/views/test_views_decorators.py | 7 ++-
tests/www/views/test_views_extra_links.py | 15 ++++---
tests/www/views/test_views_grid.py | 7 ++-
tests/www/views/test_views_home.py | 8 ++--
tests/www/views/test_views_log.py | 11 +++--
tests/www/views/test_views_mount.py | 3 +-
tests/www/views/test_views_paused.py | 3 +-
tests/www/views/test_views_pool.py | 3 +-
tests/www/views/test_views_rate_limit.py | 7 ++-
tests/www/views/test_views_rendered.py | 7 ++-
tests/www/views/test_views_robots.py | 2 +-
tests/www/views/test_views_task_norun.py | 2 +-
tests/www/views/test_views_tasks.py | 10 ++---
tests/www/views/test_views_trigger_dag.py | 4 +-
tests/www/views/test_views_variable.py | 4 +-
{dev/tests_common => tests_common}/__init__.py | 0
.../_internals/__init__.py | 0
.../_internals/capture_warnings.py | 4 +-
.../_internals/forbidden_warnings.py | 2 +-
{dev/tests_common => tests_common}/pyproject.toml | 2 +-
.../tests_common => tests_common}/pytest_plugin.py | 51 +++++++++++-----------
.../test_utils/README.md | 0
.../test_utils/__init__.py | 2 +-
.../test_utils/api_connexion_utils.py | 12 +++--
.../test_utils/asserts.py | 9 +++-
.../test_utils/azure_system_helpers.py | 16 +++----
.../test_utils/compat.py | 2 +-
.../test_utils/config.py | 0
.../tests_common => tests_common}/test_utils/db.py | 8 ++--
.../test_utils/decorators.py | 0
.../test_utils/fake_datetime.py | 4 +-
.../test_utils/gcp_system_helpers.py | 17 ++++----
.../test_utils/get_all_tests.py | 8 ++--
.../test_utils/hdfs_utils.py | 2 +
.../test_utils/logging_command_executor.py | 6 ++-
.../test_utils/mapping.py | 0
.../test_utils/mock_executor.py | 6 +--
.../test_utils/mock_operators.py | 51 ++++++++++++----------
.../test_utils/mock_plugins.py | 0
.../test_utils/mock_security_manager.py | 4 +-
.../test_utils/operators/__init__.py | 0
.../operators/postgres_local_executor.cfg | 0
.../test_utils/perf/__init__.py | 0
.../test_utils/perf/perf_kit/__init__.py | 5 +--
.../test_utils/perf/perf_kit/memory.py | 4 +-
.../test_utils/perf/perf_kit/python.py | 15 +++----
.../test_utils/perf/perf_kit/repeat_and_time.py | 12 ++---
.../test_utils/perf/perf_kit/sqlalchemy.py | 15 +++----
.../test_utils/permissions.py | 2 +
.../test_utils/providers.py | 4 +-
.../test_utils/remote_user_api_auth_backend.py | 2 +-
.../test_utils/reset_warning_registry.py | 4 +-
.../test_utils/salesforce_system_helpers.py | 0
.../test_utils/sftp_system_helpers.py | 2 +-
.../test_utils/system_tests.py | 0
.../test_utils/system_tests_class.py | 13 +++---
.../test_utils/terraform.py | 4 +-
.../test_utils/timetables.py | 2 +
.../test_utils/watcher.py | 8 +++-
.../test_utils/www.py | 0
768 files changed, 1560 insertions(+), 1752 deletions(-)
diff --git a/.dockerignore b/.dockerignore
index 2d70030fd8..e913ed4f43 100644
--- a/.dockerignore
+++ b/.dockerignore
@@ -42,6 +42,7 @@
# Add tests and kubernetes_tests to context.
!tests
+!tests_common
!kubernetes_tests
!helm_tests
!docker_tests
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index a2f42e6519..fe4315fafc 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -364,7 +364,7 @@ repos:
args: []
require_serial: true
additional_dependencies: ["ruff==0.5.5"]
- exclude: ^.*/.*_vendor/|^tests/dags/test_imports.py|^airflow/contrib/
+ exclude: ^.*/.*_vendor/|^tests/dags/test_imports.py$
- id: replace-bad-characters
name: Replace bad characters
entry: ./scripts/ci/pre_commit/replace_bad_characters.py
@@ -736,7 +736,7 @@ repos:
name: Verify usage of Airflow deprecation classes in core
entry: category=DeprecationWarning|category=PendingDeprecationWarning
files: \.py$
- exclude: ^airflow/configuration\.py$|^providers/src/airflow/providers/|^scripts/in_container/verify_providers\.py$|^(providers/)?tests/.*$|^dev/tests_common/
+ exclude: ^airflow/configuration\.py$|^providers/src/airflow/providers/|^scripts/in_container/verify_providers\.py$|^(providers/)?tests/.*$|^tests_common/
pass_filenames: true
- id: check-provide-create-sessions-imports
language: pygrep
@@ -1184,7 +1184,9 @@ repos:
^(providers/)?tests/ |
^dev/.*\.py$ |
^scripts/.*\.py$ |
- ^\w+_tests/ |
+ ^docker_tests/.*$ |
+ ^helm_tests/.*$ |
+ ^tests_common/.*$ |
^docs/.*\.py$ |
^hatch_build.py$
- id: check-provider-docs-valid
diff --git a/Dockerfile.ci b/Dockerfile.ci
index df1cdf8c27..0cdf989979 100644
--- a/Dockerfile.ci
+++ b/Dockerfile.ci
@@ -1369,6 +1369,7 @@ COPY pyproject.toml ${AIRFLOW_SOURCES}/pyproject.toml
COPY providers/pyproject.toml ${AIRFLOW_SOURCES}/providers/pyproject.toml
COPY task_sdk/pyproject.toml ${AIRFLOW_SOURCES}/task_sdk/pyproject.toml
COPY airflow/__init__.py ${AIRFLOW_SOURCES}/airflow/
+COPY tests_common/ ${AIRFLOW_SOURCES}/tests_common/
COPY generated/* ${AIRFLOW_SOURCES}/generated/
COPY constraints/* ${AIRFLOW_SOURCES}/constraints/
COPY LICENSE ${AIRFLOW_SOURCES}/LICENSE
diff --git a/contributing-docs/testing/system_tests.rst b/contributing-docs/testing/system_tests.rst
index cc64d25e90..fe010f76bd 100644
--- a/contributing-docs/testing/system_tests.rst
+++ b/contributing-docs/testing/system_tests.rst
@@ -35,7 +35,7 @@ Airflow system tests are pretty special because they serve three purposes:
Old System Tests
----------------
-The system tests derive from the ``dev.tests_common.test_utils.system_test_class.SystemTests`` class.
+The system tests derive from the ``tests_common.test_utils.system_test_class.SystemTests`` class.
Old versions of System tests should also be marked with ``@pytest.marker.system(SYSTEM)`` where ``system``
designates the system to be tested (for example, ``google.cloud``). These tests are skipped by default.
diff --git a/contributing-docs/testing/unit_tests.rst b/contributing-docs/testing/unit_tests.rst
index 468f10b68d..dc4e27f10d 100644
--- a/contributing-docs/testing/unit_tests.rst
+++ b/contributing-docs/testing/unit_tests.rst
@@ -1171,7 +1171,7 @@ are not part of the public API. We deal with it in one of the following ways:
1) If the whole provider is supposed to only work for later airflow version, we remove the whole provider
by excluding it from compatibility test configuration (see below)
-2) Some compatibility shims are defined in ``dev.tests_common.test_utils/compat.py`` - and they can be used to make the
+2) Some compatibility shims are defined in ``tests_common.test_utils/compat.py`` - and they can be used to make the
tests compatible - for example importing ``ParseImportError`` after the exception has been renamed from
``ImportError`` and it would fail in Airflow 2.9, but we have a fallback import in ``compat.py`` that
falls back to old import automatically, so all tests testing / expecting ``ParseImportError`` should import
@@ -1184,7 +1184,7 @@ are not part of the public API. We deal with it in one of the following ways:
.. code-block:: python
- from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
+ from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
@pytest.mark.skipif(not AIRFLOW_V_2_8_PLUS, reason="The tests should be skipped for Airflow < 2.8")
@@ -1197,7 +1197,7 @@ are not part of the public API. We deal with it in one of the following ways:
.. code-block:: python
- from dev.tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
+ from tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
@pytest.mark.skipif(
diff --git a/dev/breeze/src/airflow_breeze/utils/docker_command_utils.py b/dev/breeze/src/airflow_breeze/utils/docker_command_utils.py
index 164a1583e1..2352bcda53 100644
--- a/dev/breeze/src/airflow_breeze/utils/docker_command_utils.py
+++ b/dev/breeze/src/airflow_breeze/utils/docker_command_utils.py
@@ -97,6 +97,7 @@ VOLUMES_FOR_SELECTED_MOUNTS = [
("pyproject.toml", "/opt/airflow/pyproject.toml"),
("scripts", "/opt/airflow/scripts"),
("scripts/docker/entrypoint_ci.sh", "/entrypoint"),
+ ("tests_common", "/opt/airflow/tests_common"),
("tests", "/opt/airflow/tests"),
("helm_tests", "/opt/airflow/helm_tests"),
("kubernetes_tests", "/opt/airflow/kubernetes_tests"),
diff --git a/dev/breeze/src/airflow_breeze/utils/selective_checks.py b/dev/breeze/src/airflow_breeze/utils/selective_checks.py
index cf4e124ff9..74a856f369 100644
--- a/dev/breeze/src/airflow_breeze/utils/selective_checks.py
+++ b/dev/breeze/src/airflow_breeze/utils/selective_checks.py
@@ -245,7 +245,7 @@ CI_FILE_GROUP_MATCHES = HashableDict(
],
FileGroupForCi.TESTS_UTILS_FILES: [
r"^tests/utils/",
- r"^dev/tests_common/.*\.py$",
+ r"^tests_common/.*\.py$",
],
FileGroupForCi.TASK_SDK_FILES: [
r"^task_sdk/src/airflow/sdk/.*\.py$",
diff --git a/dev/breeze/tests/test_selective_checks.py b/dev/breeze/tests/test_selective_checks.py
index 619dc8eec2..ec0769c0a8 100644
--- a/dev/breeze/tests/test_selective_checks.py
+++ b/dev/breeze/tests/test_selective_checks.py
@@ -778,7 +778,7 @@ def assert_outputs_are_printed(expected_outputs: dict[str, str], stderr: str):
),
(
pytest.param(
- ("dev/tests_common/__init__.py",),
+ ("tests_common/__init__.py",),
{
"affected-providers-list-as-string": ALL_PROVIDERS_AFFECTED,
"all-python-versions": "['3.9']",
@@ -799,7 +799,7 @@ def assert_outputs_are_printed(expected_outputs: dict[str, str], stderr: str):
"needs-mypy": "true",
"mypy-folders": "['airflow', 'providers', 'docs', 'dev']",
},
- id="All tests should be run when dev/tests_common/ change",
+ id="All tests should be run when tests_common/ change",
)
),
],
diff --git a/dev/perf/scheduler_dag_execution_timing.py b/dev/perf/scheduler_dag_execution_timing.py
index fc5c21bda7..1956c8023b 100755
--- a/dev/perf/scheduler_dag_execution_timing.py
+++ b/dev/perf/scheduler_dag_execution_timing.py
@@ -107,7 +107,7 @@ def get_executor_under_test(dotted_path):
from airflow.executors.executor_loader import ExecutorLoader
if dotted_path == "MockExecutor":
- from dev.tests_common.test_utils.mock_executor import MockExecutor as executor
+ from tests_common.test_utils.mock_executor import MockExecutor as executor
else:
executor = ExecutorLoader.load_executor(dotted_path)
diff --git a/helm_tests/airflow_aux/test_pod_template_file.py b/helm_tests/airflow_aux/test_pod_template_file.py
index 8c6c8c21f9..5a507b7ee6 100644
--- a/helm_tests/airflow_aux/test_pod_template_file.py
+++ b/helm_tests/airflow_aux/test_pod_template_file.py
@@ -24,6 +24,7 @@ import jmespath
import pytest
from helm_tests.airflow_aux.test_container_lifecycle import CONTAINER_LIFECYCLE_PARAMETERS
+
from tests.charts.helm_template_generator import render_chart
diff --git a/providers/tests/alibaba/cloud/log/test_oss_task_handler.py b/providers/tests/alibaba/cloud/log/test_oss_task_handler.py
index 18abe57aa0..ff65dd89ce 100644
--- a/providers/tests/alibaba/cloud/log/test_oss_task_handler.py
+++ b/providers/tests/alibaba/cloud/log/test_oss_task_handler.py
@@ -22,14 +22,13 @@ from unittest import mock
from unittest.mock import PropertyMock
import pytest
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs
from airflow.providers.alibaba.cloud.log.oss_task_handler import OSSTaskHandler
from airflow.utils.state import TaskInstanceState
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs
-
pytestmark = pytest.mark.db_test
OSS_TASK_HANDLER_STRING = "airflow.providers.alibaba.cloud.log.oss_task_handler.{}"
diff --git a/providers/tests/amazon/aws/auth_manager/avp/test_facade.py b/providers/tests/amazon/aws/auth_manager/avp/test_facade.py
index 3d2a019503..af1149aec4 100644
--- a/providers/tests/amazon/aws/auth_manager/avp/test_facade.py
+++ b/providers/tests/amazon/aws/auth_manager/avp/test_facade.py
@@ -21,6 +21,7 @@ from typing import TYPE_CHECKING
from unittest.mock import Mock
import pytest
+from tests_common.test_utils.config import conf_vars
from airflow.exceptions import AirflowException
from airflow.providers.amazon.aws.auth_manager.avp.entities import AvpEntities, get_action_id, get_entity_type
@@ -28,8 +29,6 @@ from airflow.providers.amazon.aws.auth_manager.avp.facade import AwsAuthManagerA
from airflow.providers.amazon.aws.auth_manager.user import AwsAuthManagerUser
from airflow.utils.helpers import prune_dict
-from dev.tests_common.test_utils.config import conf_vars
-
if TYPE_CHECKING:
from airflow.auth.managers.base_auth_manager import ResourceMethod
diff --git a/providers/tests/amazon/aws/auth_manager/cli/test_avp_commands.py b/providers/tests/amazon/aws/auth_manager/cli/test_avp_commands.py
index 6122079fee..c523beec56 100644
--- a/providers/tests/amazon/aws/auth_manager/cli/test_avp_commands.py
+++ b/providers/tests/amazon/aws/auth_manager/cli/test_avp_commands.py
@@ -20,13 +20,12 @@ import importlib
from unittest.mock import ANY, Mock, patch
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
+from tests_common.test_utils.config import conf_vars
from airflow.cli import cli_parser
from airflow.providers.amazon.aws.auth_manager.cli.avp_commands import init_avp, update_schema
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
mock_boto3 = Mock()
pytestmark = [
diff --git a/providers/tests/amazon/aws/auth_manager/security_manager/test_aws_security_manager_override.py b/providers/tests/amazon/aws/auth_manager/security_manager/test_aws_security_manager_override.py
index b6071aac95..1c57bc14d8 100644
--- a/providers/tests/amazon/aws/auth_manager/security_manager/test_aws_security_manager_override.py
+++ b/providers/tests/amazon/aws/auth_manager/security_manager/test_aws_security_manager_override.py
@@ -20,8 +20,7 @@ from unittest.mock import Mock, patch
import pytest
from flask import Flask
-
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
python3_saml = pytest.importorskip("python3-saml")
diff --git a/providers/tests/amazon/aws/auth_manager/test_aws_auth_manager.py b/providers/tests/amazon/aws/auth_manager/test_aws_auth_manager.py
index 47e8a4cbcb..ba93b2105a 100644
--- a/providers/tests/amazon/aws/auth_manager/test_aws_auth_manager.py
+++ b/providers/tests/amazon/aws/auth_manager/test_aws_auth_manager.py
@@ -22,6 +22,9 @@ from unittest.mock import ANY, Mock, patch
import pytest
from flask import Flask, session
from flask_appbuilder.menu import MenuItem
+from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS, AIRFLOW_V_2_9_PLUS
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.www import check_content_in_response
from airflow.providers.amazon.aws.auth_manager.avp.entities import AvpEntities
from airflow.providers.amazon.aws.auth_manager.avp.facade import AwsAuthManagerAmazonVerifiedPermissionsFacade
@@ -39,10 +42,6 @@ from airflow.security.permissions import (
from airflow.www import app as application
from airflow.www.extensions.init_appbuilder import init_appbuilder
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS, AIRFLOW_V_2_9_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.www import check_content_in_response
-
try:
from airflow.auth.managers.models.resource_details import (
AccessView,
diff --git a/providers/tests/amazon/aws/auth_manager/views/test_auth.py b/providers/tests/amazon/aws/auth_manager/views/test_auth.py
index 9b2eec6918..919eb59ead 100644
--- a/providers/tests/amazon/aws/auth_manager/views/test_auth.py
+++ b/providers/tests/amazon/aws/auth_manager/views/test_auth.py
@@ -20,13 +20,12 @@ from unittest.mock import Mock, patch
import pytest
from flask import session, url_for
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
+from tests_common.test_utils.config import conf_vars
from airflow.exceptions import AirflowException
from airflow.www import app as application
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = [
pytest.mark.skipif(not AIRFLOW_V_2_9_PLUS, reason="Test requires Airflow 2.9+"),
pytest.mark.skip_if_database_isolation_mode,
diff --git a/providers/tests/amazon/aws/executors/batch/test_batch_executor.py b/providers/tests/amazon/aws/executors/batch/test_batch_executor.py
index e7037bd16c..85272186b3 100644
--- a/providers/tests/amazon/aws/executors/batch/test_batch_executor.py
+++ b/providers/tests/amazon/aws/executors/batch/test_batch_executor.py
@@ -26,6 +26,8 @@ import pytest
import yaml
from botocore.exceptions import ClientError, NoCredentialsError
from semver import VersionInfo
+from tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
+from tests_common.test_utils.config import conf_vars
from airflow.exceptions import AirflowException
from airflow.executors.base_executor import BaseExecutor
@@ -46,9 +48,6 @@ from airflow.utils.helpers import convert_camel_to_snake
from airflow.utils.state import State
from airflow.version import version as airflow_version_str
-from dev.tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
-from dev.tests_common.test_utils.config import conf_vars
-
airflow_version = VersionInfo(*map(int, airflow_version_str.split(".")[:3]))
ARN1 = "arn1"
diff --git a/providers/tests/amazon/aws/executors/ecs/test_ecs_executor.py b/providers/tests/amazon/aws/executors/ecs/test_ecs_executor.py
index 50cdb58038..80b7d85a9e 100644
--- a/providers/tests/amazon/aws/executors/ecs/test_ecs_executor.py
+++ b/providers/tests/amazon/aws/executors/ecs/test_ecs_executor.py
@@ -32,6 +32,9 @@ import yaml
from botocore.exceptions import ClientError
from inflection import camelize
from semver import VersionInfo
+from tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
+from tests_common.test_utils.config import conf_vars
from airflow.exceptions import AirflowException
from airflow.executors.base_executor import BaseExecutor
@@ -57,10 +60,6 @@ from airflow.utils.state import State, TaskInstanceState
from airflow.utils.timezone import utcnow
from airflow.version import version as airflow_version_str
-from dev.tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
airflow_version = VersionInfo(*map(int, airflow_version_str.split(".")[:3]))
diff --git a/providers/tests/amazon/aws/hooks/test_base_aws.py b/providers/tests/amazon/aws/hooks/test_base_aws.py
index c58993d748..8919203da0 100644
--- a/providers/tests/amazon/aws/hooks/test_base_aws.py
+++ b/providers/tests/amazon/aws/hooks/test_base_aws.py
@@ -38,6 +38,7 @@ from botocore.exceptions import NoCredentialsError
from botocore.utils import FileWebIdentityTokenLoader
from moto import mock_aws
from moto.core import DEFAULT_ACCOUNT_ID
+from tests_common.test_utils.config import conf_vars
from airflow.exceptions import AirflowException
from airflow.models.connection import Connection
@@ -50,8 +51,6 @@ from airflow.providers.amazon.aws.hooks.base_aws import (
)
from airflow.providers.amazon.aws.utils.connection_wrapper import AwsConnectionWrapper
-from dev.tests_common.test_utils.config import conf_vars
-
pytest.importorskip("aiobotocore")
MOCK_AWS_CONN_ID = "mock-conn-id"
diff --git a/providers/tests/amazon/aws/hooks/test_s3.py b/providers/tests/amazon/aws/hooks/test_s3.py
index 2e9a010006..89d29dae02 100644
--- a/providers/tests/amazon/aws/hooks/test_s3.py
+++ b/providers/tests/amazon/aws/hooks/test_s3.py
@@ -30,6 +30,7 @@ import boto3
import pytest
from botocore.exceptions import ClientError
from moto import mock_aws
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
from airflow.exceptions import AirflowException
from airflow.models import Connection
@@ -43,8 +44,6 @@ from airflow.providers.amazon.aws.hooks.s3 import (
)
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
-
@pytest.fixture
def mocked_s3_res():
diff --git a/providers/tests/amazon/aws/links/test_base_aws.py b/providers/tests/amazon/aws/links/test_base_aws.py
index 7862287080..738fb6acc4 100644
--- a/providers/tests/amazon/aws/links/test_base_aws.py
+++ b/providers/tests/amazon/aws/links/test_base_aws.py
@@ -21,14 +21,13 @@ from typing import TYPE_CHECKING, NamedTuple
from unittest import mock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.mock_operators import MockOperator
from airflow.models.xcom import XCom
from airflow.providers.amazon.aws.links.base_aws import BaseAwsLink
from airflow.serialization.serialized_objects import SerializedDAG
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.mock_operators import MockOperator
-
if TYPE_CHECKING:
from airflow.models.taskinstance import TaskInstance
diff --git a/providers/tests/amazon/aws/log/test_cloudwatch_task_handler.py b/providers/tests/amazon/aws/log/test_cloudwatch_task_handler.py
index c78fab89e1..d07699d9a7 100644
--- a/providers/tests/amazon/aws/log/test_cloudwatch_task_handler.py
+++ b/providers/tests/amazon/aws/log/test_cloudwatch_task_handler.py
@@ -26,6 +26,7 @@ from unittest.mock import ANY, Mock, call
import boto3
import pytest
from moto import mock_aws
+from tests_common.test_utils.config import conf_vars
from watchtower import CloudWatchLogHandler
from airflow.models import DAG, DagRun, TaskInstance
@@ -36,8 +37,6 @@ from airflow.providers.amazon.aws.utils import datetime_to_epoch_utc_ms
from airflow.utils.state import State
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-
def get_time_str(time_in_milliseconds):
dt_time = dt.fromtimestamp(time_in_milliseconds / 1000.0, tz=timezone.utc)
diff --git a/providers/tests/amazon/aws/log/test_s3_task_handler.py b/providers/tests/amazon/aws/log/test_s3_task_handler.py
index 9819cf95e9..a471bf0075 100644
--- a/providers/tests/amazon/aws/log/test_s3_task_handler.py
+++ b/providers/tests/amazon/aws/log/test_s3_task_handler.py
@@ -26,6 +26,7 @@ import boto3
import pytest
from botocore.exceptions import ClientError
from moto import mock_aws
+from tests_common.test_utils.config import conf_vars
from airflow.models import DAG, DagRun, TaskInstance
from airflow.operators.empty import EmptyOperator
@@ -34,8 +35,6 @@ from airflow.providers.amazon.aws.log.s3_task_handler import S3TaskHandler
from airflow.utils.state import State, TaskInstanceState
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-
@pytest.fixture(autouse=True)
def s3mock():
diff --git a/providers/tests/amazon/aws/secrets/test_systems_manager.py b/providers/tests/amazon/aws/secrets/test_systems_manager.py
index c1b35a799d..e1c035ab42 100644
--- a/providers/tests/amazon/aws/secrets/test_systems_manager.py
+++ b/providers/tests/amazon/aws/secrets/test_systems_manager.py
@@ -21,12 +21,11 @@ from unittest import mock
import pytest
from moto import mock_aws
+from tests_common.test_utils.config import conf_vars
from airflow.configuration import initialize_secrets_backends
from airflow.providers.amazon.aws.secrets.systems_manager import SystemsManagerParameterStoreBackend
-from dev.tests_common.test_utils.config import conf_vars
-
URI_CONNECTION = pytest.param(
"postgres://my-login:my-pass@my-host:5432/my-schema?param1=val1¶m2=val2", id="uri-connection"
)
diff --git a/providers/tests/amazon/aws/transfers/test_redshift_to_s3.py b/providers/tests/amazon/aws/transfers/test_redshift_to_s3.py
index e27ff2d0aa..605705a2b6 100644
--- a/providers/tests/amazon/aws/transfers/test_redshift_to_s3.py
+++ b/providers/tests/amazon/aws/transfers/test_redshift_to_s3.py
@@ -22,14 +22,13 @@ from unittest import mock
import pytest
from boto3.session import Session
+from tests_common.test_utils.asserts import assert_equal_ignore_multiple_spaces
from airflow.exceptions import AirflowException
from airflow.models.connection import Connection
from airflow.providers.amazon.aws.transfers.redshift_to_s3 import RedshiftToS3Operator
from airflow.providers.amazon.aws.utils.redshift import build_credentials_block
-from dev.tests_common.test_utils.asserts import assert_equal_ignore_multiple_spaces
-
class TestRedshiftToS3Transfer:
@pytest.mark.parametrize("table_as_file_name, expected_s3_key", [[True, "key/table_"], [False, "key"]])
diff --git a/providers/tests/amazon/aws/transfers/test_s3_to_redshift.py b/providers/tests/amazon/aws/transfers/test_s3_to_redshift.py
index 6e300791e0..c9d9ca337d 100644
--- a/providers/tests/amazon/aws/transfers/test_s3_to_redshift.py
+++ b/providers/tests/amazon/aws/transfers/test_s3_to_redshift.py
@@ -22,6 +22,7 @@ from unittest import mock
import pytest
from boto3.session import Session
+from tests_common.test_utils.asserts import assert_equal_ignore_multiple_spaces
from airflow.exceptions import AirflowException
from airflow.models.connection import Connection
@@ -34,8 +35,6 @@ from airflow.providers.common.compat.openlineage.facet import (
SchemaDatasetFacetFields,
)
-from dev.tests_common.test_utils.asserts import assert_equal_ignore_multiple_spaces
-
class TestS3ToRedshiftTransfer:
@mock.patch("airflow.providers.amazon.aws.hooks.s3.S3Hook.get_connection")
diff --git a/providers/tests/amazon/aws/transfers/test_s3_to_sftp.py b/providers/tests/amazon/aws/transfers/test_s3_to_sftp.py
index 545398d966..bcab8b2909 100644
--- a/providers/tests/amazon/aws/transfers/test_s3_to_sftp.py
+++ b/providers/tests/amazon/aws/transfers/test_s3_to_sftp.py
@@ -20,6 +20,7 @@ from __future__ import annotations
import boto3
import pytest
from moto import mock_aws
+from tests_common.test_utils.config import conf_vars
from airflow.models import DAG
from airflow.providers.amazon.aws.hooks.s3 import S3Hook
@@ -28,8 +29,6 @@ from airflow.providers.ssh.hooks.ssh import SSHHook
from airflow.providers.ssh.operators.ssh import SSHOperator
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/amazon/aws/transfers/test_sftp_to_s3.py b/providers/tests/amazon/aws/transfers/test_sftp_to_s3.py
index e4afe5c8c1..980c8a19da 100644
--- a/providers/tests/amazon/aws/transfers/test_sftp_to_s3.py
+++ b/providers/tests/amazon/aws/transfers/test_sftp_to_s3.py
@@ -20,6 +20,7 @@ from __future__ import annotations
import boto3
import pytest
from moto import mock_aws
+from tests_common.test_utils.config import conf_vars
from airflow.models import DAG
from airflow.providers.amazon.aws.hooks.s3 import S3Hook
@@ -28,8 +29,6 @@ from airflow.providers.ssh.hooks.ssh import SSHHook
from airflow.providers.ssh.operators.ssh import SSHOperator
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
BUCKET = "test-bucket"
diff --git a/providers/tests/apache/hive/hooks/test_hive.py b/providers/tests/apache/hive/hooks/test_hive.py
index aee09db280..d975ce50fb 100644
--- a/providers/tests/apache/hive/hooks/test_hive.py
+++ b/providers/tests/apache/hive/hooks/test_hive.py
@@ -25,6 +25,7 @@ from unittest import mock
import pandas as pd
import pytest
from hmsclient import HMSClient
+from tests_common.test_utils.asserts import assert_equal_ignore_multiple_spaces
from airflow.exceptions import AirflowException
from airflow.models.connection import Connection
@@ -34,7 +35,6 @@ from airflow.secrets.environment_variables import CONN_ENV_PREFIX
from airflow.utils import timezone
from airflow.utils.operator_helpers import AIRFLOW_VAR_NAME_FORMAT_MAPPING
-from dev.tests_common.test_utils.asserts import assert_equal_ignore_multiple_spaces
from providers.tests.apache.hive import (
BaseMockConnectionCursor,
InvalidHiveCliHook,
diff --git a/providers/tests/apache/livy/hooks/test_livy.py b/providers/tests/apache/livy/hooks/test_livy.py
index 0fca347b06..d2968591f9 100644
--- a/providers/tests/apache/livy/hooks/test_livy.py
+++ b/providers/tests/apache/livy/hooks/test_livy.py
@@ -25,14 +25,13 @@ import pytest
import requests
from aiohttp import ClientResponseError, RequestInfo
from requests.exceptions import RequestException
+from tests_common.test_utils.db import clear_db_connections
from airflow.exceptions import AirflowException
from airflow.models import Connection
from airflow.providers.apache.livy.hooks.livy import BatchState, LivyAsyncHook, LivyHook
from airflow.utils import db
-from dev.tests_common.test_utils.db import clear_db_connections
-
pytestmark = pytest.mark.skip_if_database_isolation_mode
LIVY_CONN_ID = LivyHook.default_conn_name
diff --git a/providers/tests/apache/spark/hooks/test_spark_sql.py b/providers/tests/apache/spark/hooks/test_spark_sql.py
index 10f63cce99..179e75ffc7 100644
--- a/providers/tests/apache/spark/hooks/test_spark_sql.py
+++ b/providers/tests/apache/spark/hooks/test_spark_sql.py
@@ -22,14 +22,13 @@ from io import StringIO
from unittest.mock import call, patch
import pytest
+from tests_common.test_utils.db import clear_db_connections
from airflow.exceptions import AirflowException
from airflow.models import Connection
from airflow.providers.apache.spark.hooks.spark_sql import SparkSqlHook
from airflow.utils import db
-from dev.tests_common.test_utils.db import clear_db_connections
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/atlassian/jira/hooks/test_jira.py b/providers/tests/atlassian/jira/hooks/test_jira.py
index e2cf938947..fbb7913d12 100644
--- a/providers/tests/atlassian/jira/hooks/test_jira.py
+++ b/providers/tests/atlassian/jira/hooks/test_jira.py
@@ -20,13 +20,12 @@ from __future__ import annotations
from unittest import mock
import pytest
+from tests_common.test_utils.compat import connection_as_json
from airflow.exceptions import AirflowProviderDeprecationWarning
from airflow.models import Connection
from airflow.providers.atlassian.jira.hooks.jira import JiraHook
-from dev.tests_common.test_utils.compat import connection_as_json
-
@pytest.fixture
def mocked_jira_client():
diff --git a/providers/tests/atlassian/jira/operators/test_jira.py b/providers/tests/atlassian/jira/operators/test_jira.py
index 614c1a4939..67c63a60d6 100644
--- a/providers/tests/atlassian/jira/operators/test_jira.py
+++ b/providers/tests/atlassian/jira/operators/test_jira.py
@@ -20,13 +20,12 @@ from __future__ import annotations
from unittest import mock
import pytest
+from tests_common.test_utils.compat import connection_as_json
from airflow.models import Connection
from airflow.providers.atlassian.jira.operators.jira import JiraOperator
from airflow.utils import timezone
-from dev.tests_common.test_utils.compat import connection_as_json
-
DEFAULT_DATE = timezone.datetime(2017, 1, 1)
MINIMAL_TEST_TICKET = {
"id": "911539",
diff --git a/providers/tests/atlassian/jira/sensors/test_jira.py b/providers/tests/atlassian/jira/sensors/test_jira.py
index 9cb2572969..ca027b7b59 100644
--- a/providers/tests/atlassian/jira/sensors/test_jira.py
+++ b/providers/tests/atlassian/jira/sensors/test_jira.py
@@ -20,13 +20,12 @@ from __future__ import annotations
from unittest import mock
import pytest
+from tests_common.test_utils.compat import connection_as_json
from airflow.models import Connection
from airflow.providers.atlassian.jira.sensors.jira import JiraTicketSensor
from airflow.utils import timezone
-from dev.tests_common.test_utils.compat import connection_as_json
-
DEFAULT_DATE = timezone.datetime(2017, 1, 1)
MINIMAL_TEST_TICKET = {
"id": "911539",
diff --git a/providers/tests/celery/cli/test_celery_command.py b/providers/tests/celery/cli/test_celery_command.py
index 03f5d60dab..0d0310f745 100644
--- a/providers/tests/celery/cli/test_celery_command.py
+++ b/providers/tests/celery/cli/test_celery_command.py
@@ -24,6 +24,8 @@ from unittest import mock
import pytest
import sqlalchemy
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
+from tests_common.test_utils.config import conf_vars
import airflow
from airflow.cli import cli_parser
@@ -31,9 +33,6 @@ from airflow.configuration import conf
from airflow.executors import executor_loader
from airflow.providers.celery.cli import celery_command
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/celery/executors/test_celery_executor.py b/providers/tests/celery/executors/test_celery_executor.py
index 261a11f837..f72ee35b0f 100644
--- a/providers/tests/celery/executors/test_celery_executor.py
+++ b/providers/tests/celery/executors/test_celery_executor.py
@@ -32,6 +32,9 @@ import time_machine
from celery import Celery
from celery.result import AsyncResult
from kombu.asynchronous import set_event_loop
+from tests_common.test_utils import db
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
+from tests_common.test_utils.config import conf_vars
from airflow.configuration import conf
from airflow.models.baseoperator import BaseOperator
@@ -42,10 +45,6 @@ from airflow.providers.celery.executors.celery_executor import CeleryExecutor
from airflow.utils import timezone
from airflow.utils.state import State
-from dev.tests_common.test_utils import db
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/celery/log_handlers/test_log_handlers.py b/providers/tests/celery/log_handlers/test_log_handlers.py
index 9eb9e33e2a..5d7d4d473d 100644
--- a/providers/tests/celery/log_handlers/test_log_handlers.py
+++ b/providers/tests/celery/log_handlers/test_log_handlers.py
@@ -23,6 +23,8 @@ from importlib import reload
from unittest import mock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.config import conf_vars
from airflow.config_templates.airflow_local_settings import DEFAULT_LOGGING_CONFIG
from airflow.executors import executor_loader
@@ -36,9 +38,6 @@ from airflow.utils.state import TaskInstanceState
from airflow.utils.timezone import datetime
from airflow.utils.types import DagRunType
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
if AIRFLOW_V_3_0_PLUS:
pass
diff --git a/providers/tests/cncf/kubernetes/cli/test_kubernetes_command.py b/providers/tests/cncf/kubernetes/cli/test_kubernetes_command.py
index 62019a8d2f..4d277c3b51 100644
--- a/providers/tests/cncf/kubernetes/cli/test_kubernetes_command.py
+++ b/providers/tests/cncf/kubernetes/cli/test_kubernetes_command.py
@@ -24,13 +24,12 @@ from unittest.mock import MagicMock, call
import kubernetes
import pytest
from dateutil.parser import parse
+from tests_common.test_utils.config import conf_vars
from airflow.cli import cli_parser
from airflow.executors import executor_loader
from airflow.providers.cncf.kubernetes.cli import kubernetes_command
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/cncf/kubernetes/executors/test_kubernetes_executor.py b/providers/tests/cncf/kubernetes/executors/test_kubernetes_executor.py
index 07b42ee3fc..85077ba177 100644
--- a/providers/tests/cncf/kubernetes/executors/test_kubernetes_executor.py
+++ b/providers/tests/cncf/kubernetes/executors/test_kubernetes_executor.py
@@ -26,6 +26,8 @@ import pytest
import yaml
from kubernetes.client import models as k8s
from kubernetes.client.rest import ApiException
+from tests_common.test_utils.compat import BashOperator
+from tests_common.test_utils.config import conf_vars
from urllib3 import HTTPResponse
from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
@@ -55,9 +57,6 @@ from airflow.providers.cncf.kubernetes.pod_generator import PodGenerator
from airflow.utils import timezone
from airflow.utils.state import State, TaskInstanceState
-from dev.tests_common.test_utils.compat import BashOperator
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.skip_if_database_isolation_mode
diff --git a/providers/tests/cncf/kubernetes/hooks/test_kubernetes.py b/providers/tests/cncf/kubernetes/hooks/test_kubernetes.py
index a6847e9c2f..d90ed05044 100644
--- a/providers/tests/cncf/kubernetes/hooks/test_kubernetes.py
+++ b/providers/tests/cncf/kubernetes/hooks/test_kubernetes.py
@@ -29,6 +29,8 @@ import pytest
from kubernetes.client.rest import ApiException
from kubernetes.config import ConfigException
from sqlalchemy.orm import make_transient
+from tests_common.test_utils.db import clear_db_connections
+from tests_common.test_utils.providers import get_provider_min_airflow_version
from airflow.exceptions import AirflowException, AirflowNotFoundException
from airflow.hooks.base import BaseHook
@@ -37,9 +39,6 @@ from airflow.providers.cncf.kubernetes.hooks.kubernetes import AsyncKubernetesHo
from airflow.utils import db
from airflow.utils.db import merge_conn
-from dev.tests_common.test_utils.db import clear_db_connections
-from dev.tests_common.test_utils.providers import get_provider_min_airflow_version
-
pytestmark = pytest.mark.db_test
KUBE_CONFIG_PATH = os.getenv("KUBECONFIG", "~/.kube/config")
diff --git a/providers/tests/cncf/kubernetes/log_handlers/test_log_handlers.py b/providers/tests/cncf/kubernetes/log_handlers/test_log_handlers.py
index b02f16ac92..7ef025d7f2 100644
--- a/providers/tests/cncf/kubernetes/log_handlers/test_log_handlers.py
+++ b/providers/tests/cncf/kubernetes/log_handlers/test_log_handlers.py
@@ -26,6 +26,8 @@ from unittest.mock import patch
import pytest
from kubernetes.client import models as k8s
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.config import conf_vars
from airflow.config_templates.airflow_local_settings import DEFAULT_LOGGING_CONFIG
from airflow.executors import executor_loader
@@ -42,9 +44,6 @@ from airflow.utils.state import State, TaskInstanceState
from airflow.utils.timezone import datetime
from airflow.utils.types import DagRunType
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
if AIRFLOW_V_3_0_PLUS:
from airflow.utils.types import DagRunTriggeredByType
diff --git a/providers/tests/cncf/kubernetes/operators/test_pod.py b/providers/tests/cncf/kubernetes/operators/test_pod.py
index ac50641370..a272a5c460 100644
--- a/providers/tests/cncf/kubernetes/operators/test_pod.py
+++ b/providers/tests/cncf/kubernetes/operators/test_pod.py
@@ -27,6 +27,7 @@ import pendulum
import pytest
from kubernetes.client import ApiClient, V1Pod, V1PodSecurityContext, V1PodStatus, models as k8s
from kubernetes.client.exceptions import ApiException
+from tests_common.test_utils import db
from urllib3 import HTTPResponse
from airflow.exceptions import (
@@ -51,8 +52,6 @@ from airflow.utils import timezone
from airflow.utils.session import create_session
from airflow.utils.types import DagRunType
-from dev.tests_common.test_utils import db
-
pytestmark = [pytest.mark.db_test, pytest.mark.skip_if_database_isolation_mode]
diff --git a/providers/tests/cncf/kubernetes/operators/test_spark_kubernetes.py b/providers/tests/cncf/kubernetes/operators/test_spark_kubernetes.py
index 18bce53688..ea15480ba1 100644
--- a/providers/tests/cncf/kubernetes/operators/test_spark_kubernetes.py
+++ b/providers/tests/cncf/kubernetes/operators/test_spark_kubernetes.py
@@ -27,6 +27,7 @@ import pendulum
import pytest
import yaml
from kubernetes.client import models as k8s
+from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
from airflow import DAG
from airflow.models import Connection, DagRun, TaskInstance
@@ -34,8 +35,6 @@ from airflow.providers.cncf.kubernetes.operators.spark_kubernetes import SparkKu
from airflow.utils import db, timezone
from airflow.utils.types import DagRunType
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
-
@patch("airflow.providers.cncf.kubernetes.operators.spark_kubernetes.KubernetesHook")
def test_spark_kubernetes_operator(mock_kubernetes_hook, data_file):
diff --git a/providers/tests/cncf/kubernetes/test_client.py b/providers/tests/cncf/kubernetes/test_client.py
index 269717ffaa..ce3a961eef 100644
--- a/providers/tests/cncf/kubernetes/test_client.py
+++ b/providers/tests/cncf/kubernetes/test_client.py
@@ -21,6 +21,7 @@ from unittest import mock
import pytest
from kubernetes.client import Configuration
+from tests_common.test_utils.config import conf_vars
from urllib3.connection import HTTPConnection, HTTPSConnection
from airflow.providers.cncf.kubernetes.kube_client import (
@@ -29,8 +30,6 @@ from airflow.providers.cncf.kubernetes.kube_client import (
get_kube_client,
)
-from dev.tests_common.test_utils.config import conf_vars
-
class TestClient:
@mock.patch("airflow.providers.cncf.kubernetes.kube_client.config")
diff --git a/providers/tests/cncf/kubernetes/test_template_rendering.py b/providers/tests/cncf/kubernetes/test_template_rendering.py
index 180b7a1e2d..bf67059ed1 100644
--- a/providers/tests/cncf/kubernetes/test_template_rendering.py
+++ b/providers/tests/cncf/kubernetes/test_template_rendering.py
@@ -21,6 +21,7 @@ from unittest import mock
import pytest
from sqlalchemy.orm import make_transient
+from tests_common.test_utils.compat import BashOperator
from airflow.models.renderedtifields import RenderedTaskInstanceFields, RenderedTaskInstanceFields as RTIF
from airflow.providers.cncf.kubernetes.template_rendering import get_rendered_k8s_spec, render_k8s_pod_yaml
@@ -28,8 +29,6 @@ from airflow.utils import timezone
from airflow.utils.session import create_session
from airflow.version import version
-from dev.tests_common.test_utils.compat import BashOperator
-
pytestmark = [pytest.mark.db_test, pytest.mark.skip_if_database_isolation_mode]
DEFAULT_DATE = timezone.datetime(2021, 9, 9)
diff --git a/providers/tests/common/io/operators/test_file_transfer.py b/providers/tests/common/io/operators/test_file_transfer.py
index 55a196fa89..4ce23de045 100644
--- a/providers/tests/common/io/operators/test_file_transfer.py
+++ b/providers/tests/common/io/operators/test_file_transfer.py
@@ -19,9 +19,9 @@ from __future__ import annotations
from unittest import mock
-from airflow.providers.common.compat.openlineage.facet import Dataset
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from airflow.providers.common.compat.openlineage.facet import Dataset
with ignore_provider_compatibility_error("2.8.0", __file__):
from airflow.providers.common.io.operators.file_transfer import FileTransferOperator
diff --git a/providers/tests/common/io/xcom/test_backend.py b/providers/tests/common/io/xcom/test_backend.py
index ae45a2f863..a60fd872bc 100644
--- a/providers/tests/common/io/xcom/test_backend.py
+++ b/providers/tests/common/io/xcom/test_backend.py
@@ -18,9 +18,8 @@
from __future__ import annotations
import pytest
-
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS, ignore_provider_compatibility_error
-from dev.tests_common.test_utils.db import is_db_isolation_mode
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS, ignore_provider_compatibility_error
+from tests_common.test_utils.db import is_db_isolation_mode
pytestmark = [
pytest.mark.db_test,
@@ -35,12 +34,12 @@ from airflow.operators.empty import EmptyOperator
with ignore_provider_compatibility_error("2.8.0", __file__):
from airflow.providers.common.io.xcom.backend import XComObjectStorageBackend
+from tests_common.test_utils import db
+from tests_common.test_utils.config import conf_vars
+
from airflow.utils import timezone
from airflow.utils.xcom import XCOM_RETURN_KEY
-from dev.tests_common.test_utils import db
-from dev.tests_common.test_utils.config import conf_vars
-
@pytest.fixture(autouse=True)
def reset_db():
diff --git a/providers/tests/common/sql/hooks/test_dbapi.py b/providers/tests/common/sql/hooks/test_dbapi.py
index 5017a5f347..0b4d7dc72e 100644
--- a/providers/tests/common/sql/hooks/test_dbapi.py
+++ b/providers/tests/common/sql/hooks/test_dbapi.py
@@ -24,14 +24,13 @@ from unittest import mock
import pytest
from pyodbc import Cursor
+from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
from airflow.config_templates.airflow_local_settings import DEFAULT_LOGGING_CONFIG
from airflow.hooks.base import BaseHook
from airflow.models import Connection
from airflow.providers.common.sql.hooks.sql import DbApiHook, fetch_all_handler, fetch_one_handler
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
-
pytestmark = [
pytest.mark.skipif(not AIRFLOW_V_2_8_PLUS, reason="Tests for Airflow 2.8.0+ only"),
]
diff --git a/providers/tests/common/sql/hooks/test_sql.py b/providers/tests/common/sql/hooks/test_sql.py
index 87673f6de6..9908a512f8 100644
--- a/providers/tests/common/sql/hooks/test_sql.py
+++ b/providers/tests/common/sql/hooks/test_sql.py
@@ -24,6 +24,7 @@ import warnings
from unittest.mock import MagicMock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
from airflow.config_templates.airflow_local_settings import DEFAULT_LOGGING_CONFIG
from airflow.exceptions import AirflowProviderDeprecationWarning
@@ -31,7 +32,6 @@ from airflow.models import Connection
from airflow.providers.common.sql.hooks.sql import DbApiHook, fetch_all_handler
from airflow.utils.session import provide_session
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
from providers.tests.common.sql.test_utils import mock_hook
pytestmark = [
diff --git a/providers/tests/common/sql/hooks/test_sqlparse.py b/providers/tests/common/sql/hooks/test_sqlparse.py
index b3622bd8ef..a32f5ee208 100644
--- a/providers/tests/common/sql/hooks/test_sqlparse.py
+++ b/providers/tests/common/sql/hooks/test_sqlparse.py
@@ -17,11 +17,10 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
from airflow.providers.common.sql.hooks.sql import DbApiHook
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
-
pytestmark = [
pytest.mark.skipif(not AIRFLOW_V_2_8_PLUS, reason="Tests for Airflow 2.8.0+ only"),
]
diff --git a/providers/tests/common/sql/operators/test_sql.py b/providers/tests/common/sql/operators/test_sql.py
index c2e4bc6f2e..d2a371311c 100644
--- a/providers/tests/common/sql/operators/test_sql.py
+++ b/providers/tests/common/sql/operators/test_sql.py
@@ -22,6 +22,7 @@ from unittest import mock
from unittest.mock import MagicMock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS, AIRFLOW_V_3_0_PLUS
from airflow import DAG
from airflow.exceptions import AirflowException
@@ -44,8 +45,6 @@ from airflow.utils import timezone
from airflow.utils.session import create_session
from airflow.utils.state import State
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS, AIRFLOW_V_3_0_PLUS
-
if AIRFLOW_V_3_0_PLUS:
from airflow.utils.types import DagRunTriggeredByType
diff --git a/providers/tests/common/sql/operators/test_sql_execute.py b/providers/tests/common/sql/operators/test_sql_execute.py
index bb42cbaba2..f03c80866c 100644
--- a/providers/tests/common/sql/operators/test_sql_execute.py
+++ b/providers/tests/common/sql/operators/test_sql_execute.py
@@ -22,6 +22,7 @@ from unittest import mock
from unittest.mock import MagicMock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
from airflow.models import Connection
from airflow.providers.common.compat.openlineage.facet import (
@@ -34,8 +35,6 @@ from airflow.providers.common.sql.hooks.sql import DbApiHook, fetch_all_handler
from airflow.providers.common.sql.operators.sql import SQLExecuteQueryOperator
from airflow.providers.openlineage.extractors.base import OperatorLineage
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
-
pytestmark = [
pytest.mark.skipif(not AIRFLOW_V_2_8_PLUS, reason="Tests for Airflow 2.8.0+ only"),
]
diff --git a/providers/tests/common/sql/sensors/test_sql.py b/providers/tests/common/sql/sensors/test_sql.py
index 33e8fe6399..c44bbf1d9d 100644
--- a/providers/tests/common/sql/sensors/test_sql.py
+++ b/providers/tests/common/sql/sensors/test_sql.py
@@ -20,6 +20,7 @@ from __future__ import annotations
from unittest import mock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
from airflow.exceptions import AirflowException
from airflow.models.dag import DAG
@@ -27,8 +28,6 @@ from airflow.providers.common.sql.hooks.sql import DbApiHook
from airflow.providers.common.sql.sensors.sql import SqlSensor
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-
pytestmark = [
pytest.mark.skipif(not AIRFLOW_V_2_9_PLUS, reason="Tests for Airflow 2.8.0+ only"),
pytest.mark.skip_if_database_isolation_mode,
diff --git a/providers/tests/common/sql/test_utils.py b/providers/tests/common/sql/test_utils.py
index 19b7bcc339..50a217c87a 100644
--- a/providers/tests/common/sql/test_utils.py
+++ b/providers/tests/common/sql/test_utils.py
@@ -21,11 +21,10 @@ from typing import TYPE_CHECKING
from unittest import mock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
from airflow.models import Connection
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_8_PLUS
-
pytestmark = [
pytest.mark.skipif(not AIRFLOW_V_2_8_PLUS, reason="Tests for Airflow 2.8.0+ only"),
]
diff --git a/providers/tests/conftest.py b/providers/tests/conftest.py
index 6ca7990e60..85ffd8f1d9 100644
--- a/providers/tests/conftest.py
+++ b/providers/tests/conftest.py
@@ -22,7 +22,7 @@ from unittest import mock
import pytest
-pytest_plugins = "dev.tests_common.pytest_plugin"
+pytest_plugins = "tests_common.pytest_plugin"
@pytest.hookimpl(tryfirst=True)
diff --git a/providers/tests/databricks/plugins/test_databricks_workflow.py b/providers/tests/databricks/plugins/test_databricks_workflow.py
index 695466c62d..72d6b3f321 100644
--- a/providers/tests/databricks/plugins/test_databricks_workflow.py
+++ b/providers/tests/databricks/plugins/test_databricks_workflow.py
@@ -21,6 +21,7 @@ from unittest.mock import MagicMock, Mock, patch
import pytest
from flask import url_for
+from tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
from airflow.exceptions import AirflowException
from airflow.models.dagrun import DagRun
@@ -41,8 +42,6 @@ from airflow.providers.databricks.plugins.databricks_workflow import (
)
from airflow.www.app import create_app
-from dev.tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
-
DAG_ID = "test_dag"
TASK_ID = "test_task"
RUN_ID = "test_run_1"
diff --git a/providers/tests/edge/api_endpoints/test_rpc_api_endpoint.py b/providers/tests/edge/api_endpoints/test_rpc_api_endpoint.py
index 9a9b4a7dd8..4a185971e6 100644
--- a/providers/tests/edge/api_endpoints/test_rpc_api_endpoint.py
+++ b/providers/tests/edge/api_endpoints/test_rpc_api_endpoint.py
@@ -21,6 +21,8 @@ from typing import TYPE_CHECKING, Generator
from unittest import mock
import pytest
+from tests_common.test_utils.decorators import dont_initialize_flask_app_submodules
+from tests_common.test_utils.mock_plugins import mock_plugin_manager
from airflow.api_connexion.exceptions import PermissionDenied
from airflow.configuration import conf
@@ -41,9 +43,6 @@ from airflow.utils.jwt_signer import JWTSigner
from airflow.utils.state import State
from airflow.www import app
-from dev.tests_common.test_utils.decorators import dont_initialize_flask_app_submodules
-from dev.tests_common.test_utils.mock_plugins import mock_plugin_manager
-
# Note: Sounds a bit strange to disable internal API tests in isolation mode but...
# As long as the test is modelled to run its own internal API endpoints, it is conflicting
# to the test setup with a dedicated internal API server.
diff --git a/providers/tests/edge/cli/test_edge_command.py b/providers/tests/edge/cli/test_edge_command.py
index af3e2c00e2..e6e07d032b 100644
--- a/providers/tests/edge/cli/test_edge_command.py
+++ b/providers/tests/edge/cli/test_edge_command.py
@@ -23,6 +23,7 @@ from unittest.mock import patch
import pytest
import time_machine
+from tests_common.test_utils.config import conf_vars
from airflow.exceptions import AirflowException
from airflow.providers.edge.cli.edge_command import (
@@ -34,8 +35,6 @@ from airflow.providers.edge.models.edge_job import EdgeJob
from airflow.providers.edge.models.edge_worker import EdgeWorker, EdgeWorkerState
from airflow.utils.state import TaskInstanceState
-from dev.tests_common.test_utils.config import conf_vars
-
pytest.importorskip("pydantic", minversion="2.0.0")
# Ignore the following error for mocking
diff --git a/providers/tests/edge/plugins/test_edge_executor_plugin.py b/providers/tests/edge/plugins/test_edge_executor_plugin.py
index d0c5a40770..bdd4847865 100644
--- a/providers/tests/edge/plugins/test_edge_executor_plugin.py
+++ b/providers/tests/edge/plugins/test_edge_executor_plugin.py
@@ -19,12 +19,11 @@ from __future__ import annotations
import importlib
import pytest
+from tests_common.test_utils.config import conf_vars
from airflow.plugins_manager import AirflowPlugin
from airflow.providers.edge.plugins import edge_executor_plugin
-from dev.tests_common.test_utils.config import conf_vars
-
def test_plugin_inactive():
with conf_vars({("edge", "api_enabled"): "false"}):
diff --git a/providers/tests/elasticsearch/log/test_es_task_handler.py b/providers/tests/elasticsearch/log/test_es_task_handler.py
index abde5daf8b..adb1009035 100644
--- a/providers/tests/elasticsearch/log/test_es_task_handler.py
+++ b/providers/tests/elasticsearch/log/test_es_task_handler.py
@@ -31,6 +31,8 @@ from urllib.parse import quote
import elasticsearch
import pendulum
import pytest
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs
from airflow.configuration import conf
from airflow.providers.elasticsearch.log.es_response import ElasticSearchResponse
@@ -44,8 +46,6 @@ from airflow.utils import timezone
from airflow.utils.state import DagRunState, TaskInstanceState
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs
from providers.tests.elasticsearch.log.elasticmock import elasticmock
from providers.tests.elasticsearch.log.elasticmock.utilities import SearchFailedException
diff --git a/providers/tests/fab/auth_manager/api/auth/backend/test_basic_auth.py b/providers/tests/fab/auth_manager/api/auth/backend/test_basic_auth.py
index 2c783e2046..65db6a654a 100644
--- a/providers/tests/fab/auth_manager/api/auth/backend/test_basic_auth.py
+++ b/providers/tests/fab/auth_manager/api/auth/backend/test_basic_auth.py
@@ -21,12 +21,11 @@ from unittest.mock import Mock, patch
import pytest
from flask import Response
from flask_appbuilder.const import AUTH_LDAP
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
from airflow.providers.fab.auth_manager.api.auth.backend.basic_auth import requires_authentication
from airflow.www import app as application
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-
pytestmark = [
pytest.mark.skipif(not AIRFLOW_V_2_9_PLUS, reason="Tests for Airflow 2.9.0+ only"),
]
diff --git a/providers/tests/fab/auth_manager/api/auth/backend/test_kerberos_auth.py b/providers/tests/fab/auth_manager/api/auth/backend/test_kerberos_auth.py
index 7d13a89e8d..e57f34ce4b 100644
--- a/providers/tests/fab/auth_manager/api/auth/backend/test_kerberos_auth.py
+++ b/providers/tests/fab/auth_manager/api/auth/backend/test_kerberos_auth.py
@@ -16,7 +16,7 @@
# under the License.
from __future__ import annotations
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.api.auth.backend.kerberos_auth import init_app
diff --git a/providers/tests/fab/auth_manager/api/auth/backend/test_session.py b/providers/tests/fab/auth_manager/api/auth/backend/test_session.py
index 513e8ec2aa..405eafe11d 100644
--- a/providers/tests/fab/auth_manager/api/auth/backend/test_session.py
+++ b/providers/tests/fab/auth_manager/api/auth/backend/test_session.py
@@ -20,12 +20,11 @@ from unittest.mock import Mock, patch
import pytest
from flask import Response
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
from airflow.providers.fab.auth_manager.api.auth.backend.session import requires_authentication
from airflow.www import app as application
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-
pytestmark = [
pytest.mark.skipif(not AIRFLOW_V_2_9_PLUS, reason="Tests for Airflow 2.9.0+ only"),
]
diff --git a/providers/tests/fab/auth_manager/api_endpoints/api_connexion_utils.py b/providers/tests/fab/auth_manager/api_endpoints/api_connexion_utils.py
index e4cbe93c9d..b208b84509 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/api_connexion_utils.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/api_connexion_utils.py
@@ -18,7 +18,7 @@ from __future__ import annotations
from contextlib import contextmanager
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.security_manager.override import EXISTING_ROLES
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_asset_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_asset_endpoint.py
index 5c22e5ba7f..104e7943f4 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_asset_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_asset_endpoint.py
@@ -20,14 +20,14 @@ from typing import Generator
import pytest
import time_machine
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import clear_db_assets, clear_db_runs
+from tests_common.test_utils.www import _check_last_log
from airflow.api_connexion.exceptions import EXCEPTIONS_LINK_MAP
from airflow.security import permissions
from airflow.utils import timezone
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import clear_db_assets, clear_db_runs
-from dev.tests_common.test_utils.www import _check_last_log
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
try:
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_auth.py b/providers/tests/fab/auth_manager/api_endpoints/test_auth.py
index 630ce7050b..fd29120070 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_auth.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_auth.py
@@ -20,12 +20,11 @@ from base64 import b64encode
import pytest
from flask_login import current_user
-
-from dev.tests_common.test_utils.api_connexion_utils import assert_401
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_pools
-from dev.tests_common.test_utils.www import client_with_login
+from tests_common.test_utils.api_connexion_utils import assert_401
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_pools
+from tests_common.test_utils.www import client_with_login
pytestmark = [
pytest.mark.db_test,
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_backfill_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_backfill_endpoint.py
index f8015ff907..4d7a5fc087 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_backfill_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_backfill_endpoint.py
@@ -22,6 +22,13 @@ from unittest import mock
import pendulum
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import (
+ clear_db_backfills,
+ clear_db_dags,
+ clear_db_runs,
+ clear_db_serialized_dags,
+)
from airflow.models import DagBag, DagModel
from airflow.models.dag import DAG
@@ -31,13 +38,6 @@ from airflow.security import permissions
from airflow.utils import timezone
from airflow.utils.session import provide_session
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import (
- clear_db_backfills,
- clear_db_dags,
- clear_db_runs,
- clear_db_serialized_dags,
-)
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
try:
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_cors.py b/providers/tests/fab/auth_manager/api_endpoints/test_cors.py
index 8dbc4f964e..3741d71fb8 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_cors.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_cors.py
@@ -19,10 +19,9 @@ from __future__ import annotations
from base64 import b64encode
import pytest
-
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_pools
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_pools
pytestmark = [
pytest.mark.db_test,
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_dag_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_dag_endpoint.py
index e93425591f..f2bd83efa3 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_dag_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_dag_endpoint.py
@@ -21,6 +21,9 @@ from datetime import datetime
import pendulum
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs, clear_db_serialized_dags
+from tests_common.test_utils.www import _check_last_log
from airflow.api_connexion.exceptions import EXCEPTIONS_LINK_MAP
from airflow.models import DagBag, DagModel
@@ -29,9 +32,6 @@ from airflow.operators.empty import EmptyOperator
from airflow.security import permissions
from airflow.utils.session import provide_session
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs, clear_db_serialized_dags
-from dev.tests_common.test_utils.www import _check_last_log
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_dag_run_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_dag_run_endpoint.py
index fa09df1be7..f822cddf49 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_dag_run_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_dag_run_endpoint.py
@@ -19,6 +19,8 @@ from __future__ import annotations
from datetime import timedelta
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs, clear_db_serialized_dags
from airflow.models.dag import DAG, DagModel
from airflow.models.dagrun import DagRun
@@ -28,8 +30,6 @@ from airflow.utils import timezone
from airflow.utils.session import create_session
from airflow.utils.state import DagRunState
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs, clear_db_serialized_dags
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import (
create_user,
delete_roles,
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_dag_source_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_dag_source_endpoint.py
index 27e4113655..8cfb6c58f5 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_dag_source_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_dag_source_endpoint.py
@@ -21,12 +21,12 @@ import os
from typing import TYPE_CHECKING
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import clear_db_dag_code, clear_db_dags, clear_db_serialized_dags
from airflow.models import DagBag
from airflow.security import permissions
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import clear_db_dag_code, clear_db_dags, clear_db_serialized_dags
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_dag_warning_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_dag_warning_endpoint.py
index 01a2c68091..ec8a72d74e 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_dag_warning_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_dag_warning_endpoint.py
@@ -17,14 +17,14 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import clear_db_dag_warnings, clear_db_dags
from airflow.models.dag import DagModel
from airflow.models.dagwarning import DagWarning
from airflow.security import permissions
from airflow.utils.session import create_session
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import clear_db_dag_warnings, clear_db_dags
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_event_log_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_event_log_endpoint.py
index 4794893241..7d78685829 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_event_log_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_event_log_endpoint.py
@@ -17,13 +17,13 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import clear_db_logs
from airflow.models import Log
from airflow.security import permissions
from airflow.utils import timezone
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import clear_db_logs
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_import_error_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_import_error_endpoint.py
index 110e8e6302..b9b8d4ff9a 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_import_error_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_import_error_endpoint.py
@@ -17,14 +17,14 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS, ParseImportError
+from tests_common.test_utils.db import clear_db_dags, clear_db_import_errors
+from tests_common.test_utils.permissions import _resource_name
from airflow.models.dag import DagModel
from airflow.security import permissions
from airflow.utils import timezone
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS, ParseImportError
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_import_errors
-from dev.tests_common.test_utils.permissions import _resource_name
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_role_and_permission_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_role_and_permission_endpoint.py
index b72c6fe661..986b55e5ff 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_role_and_permission_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_role_and_permission_endpoint.py
@@ -17,12 +17,12 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.api_connexion_utils import assert_401
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
from airflow.api_connexion.exceptions import EXCEPTIONS_LINK_MAP
from airflow.security import permissions
-from dev.tests_common.test_utils.api_connexion_utils import assert_401
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import (
create_role,
create_user,
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_task_instance_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_task_instance_endpoint.py
index 7d379b6f8c..6aea6b86f7 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_task_instance_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_task_instance_endpoint.py
@@ -20,6 +20,8 @@ import datetime as dt
import urllib
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import clear_db_runs, clear_db_sla_miss, clear_rendered_ti_fields
from airflow.api_connexion.exceptions import EXCEPTIONS_LINK_MAP
from airflow.models import DagRun, TaskInstance
@@ -29,8 +31,6 @@ from airflow.utils.state import State
from airflow.utils.timezone import datetime
from airflow.utils.types import DagRunType
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import clear_db_runs, clear_db_sla_miss, clear_rendered_ti_fields
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import (
create_user,
delete_roles,
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_user_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_user_endpoint.py
index c7a20cb59c..29f84850d9 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_user_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_user_endpoint.py
@@ -20,15 +20,15 @@ import unittest.mock
import pytest
from sqlalchemy.sql.functions import count
+from tests_common.test_utils.api_connexion_utils import assert_401
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.config import conf_vars
from airflow.api_connexion.exceptions import EXCEPTIONS_LINK_MAP
from airflow.security import permissions
from airflow.utils import timezone
from airflow.utils.session import create_session
-from dev.tests_common.test_utils.api_connexion_utils import assert_401
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
-from dev.tests_common.test_utils.config import conf_vars
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import (
create_user,
delete_role,
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_variable_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_variable_endpoint.py
index 802eb48240..038dea9e72 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_variable_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_variable_endpoint.py
@@ -17,12 +17,12 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import clear_db_variables
from airflow.models import Variable
from airflow.security import permissions
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import clear_db_variables
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/api_endpoints/test_xcom_endpoint.py b/providers/tests/fab/auth_manager/api_endpoints/test_xcom_endpoint.py
index 06e8ee4847..f45cf58de6 100644
--- a/providers/tests/fab/auth_manager/api_endpoints/test_xcom_endpoint.py
+++ b/providers/tests/fab/auth_manager/api_endpoints/test_xcom_endpoint.py
@@ -19,6 +19,8 @@ from __future__ import annotations
from datetime import timedelta
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs, clear_db_xcom
from airflow.models.dag import DagModel
from airflow.models.dagrun import DagRun
@@ -30,8 +32,6 @@ from airflow.utils.dates import parse_execution_date
from airflow.utils.session import create_session
from airflow.utils.types import DagRunType
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs, clear_db_xcom
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/cli_commands/test_definition.py b/providers/tests/fab/auth_manager/cli_commands/test_definition.py
index de906bef1b..572cbee05e 100644
--- a/providers/tests/fab/auth_manager/cli_commands/test_definition.py
+++ b/providers/tests/fab/auth_manager/cli_commands/test_definition.py
@@ -16,7 +16,7 @@
# under the License.
from __future__ import annotations
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.cli_commands.definition import (
diff --git a/providers/tests/fab/auth_manager/cli_commands/test_role_command.py b/providers/tests/fab/auth_manager/cli_commands/test_role_command.py
index fd176702c7..d07cfc6124 100644
--- a/providers/tests/fab/auth_manager/cli_commands/test_role_command.py
+++ b/providers/tests/fab/auth_manager/cli_commands/test_role_command.py
@@ -23,12 +23,11 @@ from io import StringIO
from typing import TYPE_CHECKING
import pytest
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.config import conf_vars
from airflow.cli import cli_parser
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
-from dev.tests_common.test_utils.config import conf_vars
-
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.cli_commands import role_command
from airflow.providers.fab.auth_manager.cli_commands.utils import get_application_builder
diff --git a/providers/tests/fab/auth_manager/cli_commands/test_sync_perm_command.py b/providers/tests/fab/auth_manager/cli_commands/test_sync_perm_command.py
index e0e4a70f4a..a0345909dc 100644
--- a/providers/tests/fab/auth_manager/cli_commands/test_sync_perm_command.py
+++ b/providers/tests/fab/auth_manager/cli_commands/test_sync_perm_command.py
@@ -20,11 +20,10 @@ from __future__ import annotations
from unittest import mock
import pytest
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
from airflow.cli import cli_parser
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
-
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.cli_commands import sync_perm_command
diff --git a/providers/tests/fab/auth_manager/cli_commands/test_user_command.py b/providers/tests/fab/auth_manager/cli_commands/test_user_command.py
index 5f2f66b086..6ccd4c9971 100644
--- a/providers/tests/fab/auth_manager/cli_commands/test_user_command.py
+++ b/providers/tests/fab/auth_manager/cli_commands/test_user_command.py
@@ -24,11 +24,10 @@ from contextlib import redirect_stdout
from io import StringIO
import pytest
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
from airflow.cli import cli_parser
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
-
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.cli_commands import user_command
from airflow.providers.fab.auth_manager.cli_commands.utils import get_application_builder
diff --git a/providers/tests/fab/auth_manager/cli_commands/test_utils.py b/providers/tests/fab/auth_manager/cli_commands/test_utils.py
index f52defae6b..193317249e 100644
--- a/providers/tests/fab/auth_manager/cli_commands/test_utils.py
+++ b/providers/tests/fab/auth_manager/cli_commands/test_utils.py
@@ -17,8 +17,7 @@
from __future__ import annotations
import pytest
-
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.cli_commands.utils import get_application_builder
diff --git a/providers/tests/fab/auth_manager/conftest.py b/providers/tests/fab/auth_manager/conftest.py
index 9102f5d0f6..5818ecf435 100644
--- a/providers/tests/fab/auth_manager/conftest.py
+++ b/providers/tests/fab/auth_manager/conftest.py
@@ -17,12 +17,11 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.decorators import dont_initialize_flask_app_submodules
from airflow.www import app
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.decorators import dont_initialize_flask_app_submodules
-
@pytest.fixture(scope="session")
def minimal_app_for_auth_api():
diff --git a/providers/tests/fab/auth_manager/decorators/test_auth.py b/providers/tests/fab/auth_manager/decorators/test_auth.py
index 202f0d6227..d6f60bf42f 100644
--- a/providers/tests/fab/auth_manager/decorators/test_auth.py
+++ b/providers/tests/fab/auth_manager/decorators/test_auth.py
@@ -19,11 +19,10 @@ from __future__ import annotations
from unittest.mock import Mock, patch
import pytest
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
from airflow.security.permissions import ACTION_CAN_READ, RESOURCE_DAG
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
-
permissions = [(ACTION_CAN_READ, RESOURCE_DAG)]
with ignore_provider_compatibility_error("2.9.0+", __file__):
diff --git a/providers/tests/fab/auth_manager/models/test_anonymous_user.py b/providers/tests/fab/auth_manager/models/test_anonymous_user.py
index 419d17aa9f..eaf6b357f9 100644
--- a/providers/tests/fab/auth_manager/models/test_anonymous_user.py
+++ b/providers/tests/fab/auth_manager/models/test_anonymous_user.py
@@ -17,7 +17,7 @@
# under the License.
from __future__ import annotations
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.models.anonymous_user import AnonymousUser
diff --git a/providers/tests/fab/auth_manager/schemas/test_user_schema.py b/providers/tests/fab/auth_manager/schemas/test_user_schema.py
index 648372d3ba..9f8dbd3742 100644
--- a/providers/tests/fab/auth_manager/schemas/test_user_schema.py
+++ b/providers/tests/fab/auth_manager/schemas/test_user_schema.py
@@ -17,10 +17,10 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
from airflow.utils import timezone
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_role, delete_role
with ignore_provider_compatibility_error("2.9.0+", __file__):
diff --git a/providers/tests/fab/auth_manager/security_manager/test_constants.py b/providers/tests/fab/auth_manager/security_manager/test_constants.py
index a6566fd987..dbe592c59d 100644
--- a/providers/tests/fab/auth_manager/security_manager/test_constants.py
+++ b/providers/tests/fab/auth_manager/security_manager/test_constants.py
@@ -16,7 +16,7 @@
# under the License.
from __future__ import annotations
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.security_manager.constants import EXISTING_ROLES
diff --git a/providers/tests/fab/auth_manager/security_manager/test_override.py b/providers/tests/fab/auth_manager/security_manager/test_override.py
index 2733b688d1..6ba1ccda29 100644
--- a/providers/tests/fab/auth_manager/security_manager/test_override.py
+++ b/providers/tests/fab/auth_manager/security_manager/test_override.py
@@ -19,7 +19,7 @@ from __future__ import annotations
from unittest import mock
from unittest.mock import Mock
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.security_manager.override import FabAirflowSecurityManagerOverride
diff --git a/providers/tests/fab/auth_manager/test_fab_auth_manager.py b/providers/tests/fab/auth_manager/test_fab_auth_manager.py
index 064c8e0dfd..0d111f0453 100644
--- a/providers/tests/fab/auth_manager/test_fab_auth_manager.py
+++ b/providers/tests/fab/auth_manager/test_fab_auth_manager.py
@@ -31,7 +31,7 @@ try:
except ImportError:
pass
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.fab_auth_manager import FabAuthManager
diff --git a/providers/tests/fab/auth_manager/test_models.py b/providers/tests/fab/auth_manager/test_models.py
index 6f03be3731..76b69c3dea 100644
--- a/providers/tests/fab/auth_manager/test_models.py
+++ b/providers/tests/fab/auth_manager/test_models.py
@@ -19,8 +19,7 @@ from __future__ import annotations
from unittest import mock
from sqlalchemy import Column, MetaData, String, Table
-
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.models import (
diff --git a/providers/tests/fab/auth_manager/test_security.py b/providers/tests/fab/auth_manager/test_security.py
index 8c9e221b71..8156999f01 100644
--- a/providers/tests/fab/auth_manager/test_security.py
+++ b/providers/tests/fab/auth_manager/test_security.py
@@ -31,19 +31,23 @@ import time_machine
from flask_appbuilder import SQLA, Model, expose, has_access
from flask_appbuilder.views import BaseView, ModelView
from sqlalchemy import Column, Date, Float, Integer, String
+from tests_common.test_utils.compat import ignore_provider_compatibility_error
from airflow.configuration import initialize_config
from airflow.exceptions import AirflowException
from airflow.models import DagModel
from airflow.models.dag import DAG
-from dev.tests_common.test_utils.compat import ignore_provider_compatibility_error
-
with ignore_provider_compatibility_error("2.9.0+", __file__):
from airflow.providers.fab.auth_manager.fab_auth_manager import FabAuthManager
from airflow.providers.fab.auth_manager.models import assoc_permission_role
from airflow.providers.fab.auth_manager.models.anonymous_user import AnonymousUser
+from tests_common.test_utils.asserts import assert_queries_count
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs
+from tests_common.test_utils.mock_security_manager import MockSecurityManager
+from tests_common.test_utils.permissions import _resource_name
+
from airflow.security import permissions
from airflow.security.permissions import ACTION_CAN_READ
from airflow.www import app as application
@@ -51,10 +55,6 @@ from airflow.www.auth import get_access_denied_message
from airflow.www.extensions.init_auth_manager import get_auth_manager
from airflow.www.utils import CustomSQLAInterface
-from dev.tests_common.test_utils.asserts import assert_queries_count
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs
-from dev.tests_common.test_utils.mock_security_manager import MockSecurityManager
-from dev.tests_common.test_utils.permissions import _resource_name
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import (
create_user,
create_user_scope,
diff --git a/providers/tests/fab/auth_manager/views/test_permissions.py b/providers/tests/fab/auth_manager/views/test_permissions.py
index 2ac26ffe45..eeb005b47e 100644
--- a/providers/tests/fab/auth_manager/views/test_permissions.py
+++ b/providers/tests/fab/auth_manager/views/test_permissions.py
@@ -18,12 +18,12 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
+from tests_common.test_utils.www import client_with_login
from airflow.security import permissions
from airflow.www import app as application
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-from dev.tests_common.test_utils.www import client_with_login
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/views/test_roles_list.py b/providers/tests/fab/auth_manager/views/test_roles_list.py
index 3c1509d753..f94907c3e0 100644
--- a/providers/tests/fab/auth_manager/views/test_roles_list.py
+++ b/providers/tests/fab/auth_manager/views/test_roles_list.py
@@ -18,12 +18,12 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
+from tests_common.test_utils.www import client_with_login
from airflow.security import permissions
from airflow.www import app as application
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-from dev.tests_common.test_utils.www import client_with_login
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/views/test_user.py b/providers/tests/fab/auth_manager/views/test_user.py
index ee068f7f31..140d7cd331 100644
--- a/providers/tests/fab/auth_manager/views/test_user.py
+++ b/providers/tests/fab/auth_manager/views/test_user.py
@@ -18,12 +18,12 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
+from tests_common.test_utils.www import client_with_login
from airflow.security import permissions
from airflow.www import app as application
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-from dev.tests_common.test_utils.www import client_with_login
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/views/test_user_edit.py b/providers/tests/fab/auth_manager/views/test_user_edit.py
index 7cdc1a493b..fb994dbd66 100644
--- a/providers/tests/fab/auth_manager/views/test_user_edit.py
+++ b/providers/tests/fab/auth_manager/views/test_user_edit.py
@@ -18,12 +18,12 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
+from tests_common.test_utils.www import client_with_login
from airflow.security import permissions
from airflow.www import app as application
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-from dev.tests_common.test_utils.www import client_with_login
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/fab/auth_manager/views/test_user_stats.py b/providers/tests/fab/auth_manager/views/test_user_stats.py
index e50bc87535..bb7d627fda 100644
--- a/providers/tests/fab/auth_manager/views/test_user_stats.py
+++ b/providers/tests/fab/auth_manager/views/test_user_stats.py
@@ -18,12 +18,12 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
+from tests_common.test_utils.www import client_with_login
from airflow.security import permissions
from airflow.www import app as application
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-from dev.tests_common.test_utils.www import client_with_login
from providers.tests.fab.auth_manager.api_endpoints.api_connexion_utils import create_user, delete_user
pytestmark = [
diff --git a/providers/tests/google/cloud/hooks/test_bigquery_system.py b/providers/tests/google/cloud/hooks/test_bigquery_system.py
index 676ab35035..454ab63079 100644
--- a/providers/tests/google/cloud/hooks/test_bigquery_system.py
+++ b/providers/tests/google/cloud/hooks/test_bigquery_system.py
@@ -20,10 +20,10 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.gcp_system_helpers import GoogleSystemTest
from airflow.providers.google.cloud.hooks import bigquery as hook
-from dev.tests_common.test_utils.gcp_system_helpers import GoogleSystemTest
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_BIGQUERY_KEY
diff --git a/providers/tests/google/cloud/hooks/test_kms_system.py b/providers/tests/google/cloud/hooks/test_kms_system.py
index 374fabb524..36c5bfa207 100644
--- a/providers/tests/google/cloud/hooks/test_kms_system.py
+++ b/providers/tests/google/cloud/hooks/test_kms_system.py
@@ -21,10 +21,10 @@ import os
from tempfile import TemporaryDirectory
import pytest
+from tests_common.test_utils.gcp_system_helpers import GoogleSystemTest, provide_gcp_context
from airflow.providers.google.cloud.hooks.kms import CloudKMSHook
-from dev.tests_common.test_utils.gcp_system_helpers import GoogleSystemTest, provide_gcp_context
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_KMS_KEY
# To prevent resource name collisions, key ring and key resources CANNOT be deleted, so
diff --git a/providers/tests/google/cloud/hooks/test_secret_manager_system.py b/providers/tests/google/cloud/hooks/test_secret_manager_system.py
index d13305f52b..9f512f5f23 100644
--- a/providers/tests/google/cloud/hooks/test_secret_manager_system.py
+++ b/providers/tests/google/cloud/hooks/test_secret_manager_system.py
@@ -19,10 +19,10 @@ from __future__ import annotations
import os
import pytest
+from tests_common.test_utils.gcp_system_helpers import GoogleSystemTest, provide_gcp_context
from airflow.providers.google.cloud.hooks.secret_manager import SecretsManagerHook
-from dev.tests_common.test_utils.gcp_system_helpers import GoogleSystemTest, provide_gcp_context
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_SECRET_MANAGER_KEY
TEST_SECRET_ID = os.environ.get("GCP_SECRET_MANAGER_SECRET_ID", "test-secret")
diff --git a/providers/tests/google/cloud/log/test_gcs_task_handler.py b/providers/tests/google/cloud/log/test_gcs_task_handler.py
index 9f27c618de..369f4b8880 100644
--- a/providers/tests/google/cloud/log/test_gcs_task_handler.py
+++ b/providers/tests/google/cloud/log/test_gcs_task_handler.py
@@ -23,14 +23,13 @@ from unittest import mock
from unittest.mock import MagicMock
import pytest
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs
from airflow.providers.google.cloud.log.gcs_task_handler import GCSTaskHandler
from airflow.utils.state import TaskInstanceState
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs
-
@pytest.mark.db_test
class TestGCSTaskHandler:
diff --git a/providers/tests/google/cloud/log/test_gcs_task_handler_system.py b/providers/tests/google/cloud/log/test_gcs_task_handler_system.py
index eb1ce08d14..e3e2bd2caf 100644
--- a/providers/tests/google/cloud/log/test_gcs_task_handler_system.py
+++ b/providers/tests/google/cloud/log/test_gcs_task_handler_system.py
@@ -23,6 +23,13 @@ import subprocess
from unittest import mock
import pytest
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_connections, clear_db_runs
+from tests_common.test_utils.gcp_system_helpers import (
+ GoogleSystemTest,
+ provide_gcp_context,
+ resolve_full_gcp_key_path,
+)
from airflow import settings
from airflow.example_dags import example_complex
@@ -30,13 +37,6 @@ from airflow.models import DagBag, TaskInstance
from airflow.utils.log.log_reader import TaskLogReader
from airflow.utils.session import provide_session
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_connections, clear_db_runs
-from dev.tests_common.test_utils.gcp_system_helpers import (
- GoogleSystemTest,
- provide_gcp_context,
- resolve_full_gcp_key_path,
-)
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_GCS_KEY
diff --git a/providers/tests/google/cloud/log/test_stackdriver_task_handler.py b/providers/tests/google/cloud/log/test_stackdriver_task_handler.py
index 783f1a34b7..60752f3b0d 100644
--- a/providers/tests/google/cloud/log/test_stackdriver_task_handler.py
+++ b/providers/tests/google/cloud/log/test_stackdriver_task_handler.py
@@ -24,16 +24,15 @@ from urllib.parse import parse_qs, urlsplit
import pytest
from google.cloud.logging import Resource
from google.cloud.logging_v2.types import ListLogEntriesRequest, ListLogEntriesResponse, LogEntry
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs
from airflow.exceptions import RemovedInAirflow3Warning
from airflow.providers.google.cloud.log.stackdriver_task_handler import StackdriverTaskHandler
from airflow.utils import timezone
from airflow.utils.state import TaskInstanceState
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs
-
def _create_list_log_entries_response_mock(messages, token):
return ListLogEntriesResponse(
diff --git a/providers/tests/google/cloud/log/test_stackdriver_task_handler_system.py b/providers/tests/google/cloud/log/test_stackdriver_task_handler_system.py
index a53dd43d08..da01e9c770 100644
--- a/providers/tests/google/cloud/log/test_stackdriver_task_handler_system.py
+++ b/providers/tests/google/cloud/log/test_stackdriver_task_handler_system.py
@@ -23,6 +23,13 @@ import subprocess
from unittest import mock
import pytest
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_runs
+from tests_common.test_utils.gcp_system_helpers import (
+ GoogleSystemTest,
+ provide_gcp_context,
+ resolve_full_gcp_key_path,
+)
from airflow import settings
from airflow.example_dags import example_complex
@@ -30,13 +37,6 @@ from airflow.models import TaskInstance
from airflow.utils.log.log_reader import TaskLogReader
from airflow.utils.session import provide_session
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_runs
-from dev.tests_common.test_utils.gcp_system_helpers import (
- GoogleSystemTest,
- provide_gcp_context,
- resolve_full_gcp_key_path,
-)
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_STACKDRIVER
diff --git a/providers/tests/google/cloud/operators/test_bigquery.py b/providers/tests/google/cloud/operators/test_bigquery.py
index 3836d73766..1edc67c45a 100644
--- a/providers/tests/google/cloud/operators/test_bigquery.py
+++ b/providers/tests/google/cloud/operators/test_bigquery.py
@@ -27,6 +27,12 @@ import pandas as pd
import pytest
from google.cloud.bigquery import DEFAULT_RETRY, ScalarQueryParameter
from google.cloud.exceptions import Conflict
+from tests_common.test_utils.db import (
+ clear_db_dags,
+ clear_db_runs,
+ clear_db_serialized_dags,
+ clear_db_xcom,
+)
from airflow.exceptions import (
AirflowException,
@@ -75,13 +81,6 @@ from airflow.providers.google.cloud.triggers.bigquery import (
from airflow.serialization.serialized_objects import SerializedDAG
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.db import (
- clear_db_dags,
- clear_db_runs,
- clear_db_serialized_dags,
- clear_db_xcom,
-)
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/google/cloud/operators/test_bigquery_dts.py b/providers/tests/google/cloud/operators/test_bigquery_dts.py
index f50c42805b..b196237b5b 100644
--- a/providers/tests/google/cloud/operators/test_bigquery_dts.py
+++ b/providers/tests/google/cloud/operators/test_bigquery_dts.py
@@ -21,6 +21,7 @@ from unittest import mock
from google.api_core.gapic_v1.method import DEFAULT
from google.cloud.bigquery_datatransfer_v1 import StartManualTransferRunsResponse, TransferConfig, TransferRun
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
from airflow.providers.google.cloud.operators.bigquery_dts import (
BigQueryCreateDataTransferOperator,
@@ -28,8 +29,6 @@ from airflow.providers.google.cloud.operators.bigquery_dts import (
BigQueryDeleteDataTransferConfigOperator,
)
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-
PROJECT_ID = "id"
diff --git a/providers/tests/google/cloud/operators/test_dataprep_system.py b/providers/tests/google/cloud/operators/test_dataprep_system.py
index dad77ac4ff..9706859f02 100644
--- a/providers/tests/google/cloud/operators/test_dataprep_system.py
+++ b/providers/tests/google/cloud/operators/test_dataprep_system.py
@@ -21,14 +21,13 @@ import json
import os
import pytest
+from tests_common.test_utils.db import clear_db_connections
+from tests_common.test_utils.gcp_system_helpers import GoogleSystemTest
+from tests_common.test_utils.system_tests import get_test_run
from airflow.models import Connection
from airflow.utils.session import create_session
-from dev.tests_common.test_utils.db import clear_db_connections
-from dev.tests_common.test_utils.gcp_system_helpers import GoogleSystemTest
-from dev.tests_common.test_utils.system_tests import get_test_run
-
TOKEN = os.environ.get("DATAPREP_TOKEN")
EXTRA = {"token": TOKEN}
diff --git a/providers/tests/google/cloud/operators/test_dataproc.py b/providers/tests/google/cloud/operators/test_dataproc.py
index cf4bffa3a0..a8b747d564 100644
--- a/providers/tests/google/cloud/operators/test_dataproc.py
+++ b/providers/tests/google/cloud/operators/test_dataproc.py
@@ -27,6 +27,8 @@ from google.api_core.retry import Retry
from google.api_core.retry_async import AsyncRetry
from google.cloud import dataproc
from google.cloud.dataproc_v1 import Batch, Cluster, JobStatus
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS, AIRFLOW_VERSION
+from tests_common.test_utils.db import clear_db_runs, clear_db_xcom
from airflow.exceptions import (
AirflowException,
@@ -80,9 +82,6 @@ from airflow.providers.google.common.consts import GOOGLE_DEFAULT_DEFERRABLE_MET
from airflow.serialization.serialized_objects import SerializedDAG
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS, AIRFLOW_VERSION
-from dev.tests_common.test_utils.db import clear_db_runs, clear_db_xcom
-
AIRFLOW_VERSION_LABEL = "v" + str(AIRFLOW_VERSION).replace(".", "-").replace("+", "-")
cluster_params = inspect.signature(ClusterGenerator.__init__).parameters
diff --git a/providers/tests/google/cloud/operators/test_datastore_system.py b/providers/tests/google/cloud/operators/test_datastore_system.py
index 8807288358..3a9e822411 100644
--- a/providers/tests/google/cloud/operators/test_datastore_system.py
+++ b/providers/tests/google/cloud/operators/test_datastore_system.py
@@ -20,12 +20,12 @@ from __future__ import annotations
import os
import pytest
-
-from dev.tests_common.test_utils.gcp_system_helpers import (
+from tests_common.test_utils.gcp_system_helpers import (
CLOUD_DAG_FOLDER,
GoogleSystemTest,
provide_gcp_context,
)
+
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_DATASTORE_KEY
BUCKET = os.environ.get("GCP_DATASTORE_BUCKET", "datastore-system-test")
diff --git a/providers/tests/google/cloud/operators/test_looker.py b/providers/tests/google/cloud/operators/test_looker.py
index b368259c50..faefc0eca5 100644
--- a/providers/tests/google/cloud/operators/test_looker.py
+++ b/providers/tests/google/cloud/operators/test_looker.py
@@ -20,14 +20,13 @@ from unittest import mock
from unittest.mock import MagicMock
import pytest
+from tests_common.test_utils.db import clear_db_runs, clear_db_xcom
from airflow.exceptions import AirflowException
from airflow.models import DAG, DagBag
from airflow.providers.google.cloud.operators.looker import LookerStartPdtBuildOperator
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.db import clear_db_runs, clear_db_xcom
-
OPERATOR_PATH = "airflow.providers.google.cloud.operators.looker.{}"
TASK_ID = "task-id"
diff --git a/providers/tests/google/cloud/secrets/test_secret_manager_system.py b/providers/tests/google/cloud/secrets/test_secret_manager_system.py
index b9b8b6d0ca..04b7a9fe1f 100644
--- a/providers/tests/google/cloud/secrets/test_secret_manager_system.py
+++ b/providers/tests/google/cloud/secrets/test_secret_manager_system.py
@@ -22,8 +22,8 @@ import subprocess
from unittest import mock
import pytest
+from tests_common.test_utils.gcp_system_helpers import GoogleSystemTest, provide_gcp_context
-from dev.tests_common.test_utils.gcp_system_helpers import GoogleSystemTest, provide_gcp_context
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_SECRET_MANAGER_KEY
BACKEND_IMPORT_PATH = "airflow.providers.google.cloud.secrets.secret_manager.CloudSecretManagerBackend"
diff --git a/providers/tests/google/cloud/transfers/test_facebook_ads_to_gcs_system.py b/providers/tests/google/cloud/transfers/test_facebook_ads_to_gcs_system.py
index 9cb0a9be52..4942a36648 100644
--- a/providers/tests/google/cloud/transfers/test_facebook_ads_to_gcs_system.py
+++ b/providers/tests/google/cloud/transfers/test_facebook_ads_to_gcs_system.py
@@ -22,17 +22,17 @@ import os
from contextlib import contextmanager
import pytest
+from tests_common.test_utils.gcp_system_helpers import (
+ GoogleSystemTest,
+ provide_gcp_context,
+)
+from tests_common.test_utils.system_tests import get_test_run
from airflow.exceptions import AirflowException
from airflow.models import Connection
from airflow.providers.google.cloud.example_dags import example_facebook_ads_to_gcs
from airflow.utils.process_utils import patch_environ
-from dev.tests_common.test_utils.gcp_system_helpers import (
- GoogleSystemTest,
- provide_gcp_context,
-)
-from dev.tests_common.test_utils.system_tests import get_test_run
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_BIGQUERY_KEY
CREDENTIALS_DIR = os.environ.get("CREDENTIALS_DIR", "/files/airflow-breeze-config/keys")
diff --git a/providers/tests/google/cloud/transfers/test_salesforce_to_gcs_system.py b/providers/tests/google/cloud/transfers/test_salesforce_to_gcs_system.py
index d556f2d86e..e525ca1add 100644
--- a/providers/tests/google/cloud/transfers/test_salesforce_to_gcs_system.py
+++ b/providers/tests/google/cloud/transfers/test_salesforce_to_gcs_system.py
@@ -19,15 +19,15 @@ from __future__ import annotations
import os
import pytest
-
-from airflow.providers.google.cloud.example_dags import example_salesforce_to_gcs
-
-from dev.tests_common.test_utils.gcp_system_helpers import (
+from tests_common.test_utils.gcp_system_helpers import (
GoogleSystemTest,
provide_gcp_context,
)
-from dev.tests_common.test_utils.salesforce_system_helpers import provide_salesforce_connection
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.salesforce_system_helpers import provide_salesforce_connection
+from tests_common.test_utils.system_tests import get_test_run
+
+from airflow.providers.google.cloud.example_dags import example_salesforce_to_gcs
+
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_BIGQUERY_KEY
CREDENTIALS_DIR = os.environ.get("CREDENTIALS_DIR", "/files/airflow-breeze-config/keys")
diff --git a/providers/tests/google/cloud/utils/gcp_authenticator.py b/providers/tests/google/cloud/utils/gcp_authenticator.py
index 6bb11b2606..36bca98e9a 100644
--- a/providers/tests/google/cloud/utils/gcp_authenticator.py
+++ b/providers/tests/google/cloud/utils/gcp_authenticator.py
@@ -21,13 +21,13 @@ import json
import os
import subprocess
+from tests_common.test_utils import AIRFLOW_MAIN_FOLDER
+from tests_common.test_utils.logging_command_executor import CommandExecutor
+
from airflow import settings
from airflow.exceptions import AirflowException
from airflow.models import Connection
-from dev.tests_common.test_utils import AIRFLOW_MAIN_FOLDER
-from dev.tests_common.test_utils.logging_command_executor import CommandExecutor
-
# Please keep these variables in alphabetical order.
GCP_AI_KEY = "gcp_ai.json"
diff --git a/providers/tests/google/common/auth_backend/test_google_openid.py b/providers/tests/google/common/auth_backend/test_google_openid.py
index 67b0ff2003..02f08e6107 100644
--- a/providers/tests/google/common/auth_backend/test_google_openid.py
+++ b/providers/tests/google/common/auth_backend/test_google_openid.py
@@ -20,14 +20,13 @@ from unittest import mock
import pytest
from google.auth.exceptions import GoogleAuthError
+from tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_pools
+from tests_common.test_utils.decorators import dont_initialize_flask_app_submodules
from airflow.www.app import create_app
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_9_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_pools
-from dev.tests_common.test_utils.decorators import dont_initialize_flask_app_submodules
-
@pytest.fixture(scope="module")
def google_openid_app():
diff --git a/providers/tests/google/marketing_platform/operators/test_display_video_system.py b/providers/tests/google/marketing_platform/operators/test_display_video_system.py
index 49f44948ab..57e93a0544 100644
--- a/providers/tests/google/marketing_platform/operators/test_display_video_system.py
+++ b/providers/tests/google/marketing_platform/operators/test_display_video_system.py
@@ -17,6 +17,12 @@
from __future__ import annotations
import pytest
+from tests_common.test_utils.gcp_system_helpers import (
+ MARKETING_DAG_FOLDER,
+ GoogleSystemTest,
+ provide_gcp_context,
+)
+from tests_common.test_utils.system_tests import get_test_run
from airflow.providers.google.cloud.hooks.bigquery import BigQueryHook
from airflow.providers.google.marketing_platform.example_dags.example_display_video import (
@@ -25,12 +31,6 @@ from airflow.providers.google.marketing_platform.example_dags.example_display_vi
dag_example_display_video_sdf,
)
-from dev.tests_common.test_utils.gcp_system_helpers import (
- MARKETING_DAG_FOLDER,
- GoogleSystemTest,
- provide_gcp_context,
-)
-from dev.tests_common.test_utils.system_tests import get_test_run
from providers.tests.google.cloud.utils.gcp_authenticator import GCP_BIGQUERY_KEY, GMP_KEY
# Requires the following scope:
diff --git a/providers/tests/hashicorp/hooks/test_vault.py b/providers/tests/hashicorp/hooks/test_vault.py
index 442a656d20..8aa555567b 100644
--- a/providers/tests/hashicorp/hooks/test_vault.py
+++ b/providers/tests/hashicorp/hooks/test_vault.py
@@ -22,13 +22,12 @@ from unittest.mock import PropertyMock, mock_open, patch
import pytest
from hvac.exceptions import VaultError
+from tests_common.test_utils.config import conf_vars
from airflow.configuration import AirflowConfigParser
from airflow.exceptions import AirflowConfigException
from airflow.providers.hashicorp.hooks.vault import VaultHook
-from dev.tests_common.test_utils.config import conf_vars
-
class TestVaultHook:
@staticmethod
diff --git a/providers/tests/imap/hooks/test_imap.py b/providers/tests/imap/hooks/test_imap.py
index 2971cfd192..9d2fcad5c7 100644
--- a/providers/tests/imap/hooks/test_imap.py
+++ b/providers/tests/imap/hooks/test_imap.py
@@ -22,14 +22,13 @@ import json
from unittest.mock import Mock, mock_open, patch
import pytest
+from tests_common.test_utils.config import conf_vars
from airflow.exceptions import AirflowException
from airflow.models import Connection
from airflow.providers.imap.hooks.imap import ImapHook
from airflow.utils import db
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/microsoft/azure/log/test_wasb_task_handler.py b/providers/tests/microsoft/azure/log/test_wasb_task_handler.py
index 224961efe4..2200a65f4d 100644
--- a/providers/tests/microsoft/azure/log/test_wasb_task_handler.py
+++ b/providers/tests/microsoft/azure/log/test_wasb_task_handler.py
@@ -24,15 +24,14 @@ from unittest import mock
import pytest
from azure.common import AzureHttpError
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs
from airflow.providers.microsoft.azure.hooks.wasb import WasbHook
from airflow.providers.microsoft.azure.log.wasb_task_handler import WasbTaskHandler
from airflow.utils.state import TaskInstanceState
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/mongo/hooks/test_mongo.py b/providers/tests/mongo/hooks/test_mongo.py
index 78756b0cc4..9cd5b4f067 100644
--- a/providers/tests/mongo/hooks/test_mongo.py
+++ b/providers/tests/mongo/hooks/test_mongo.py
@@ -23,13 +23,12 @@ from typing import TYPE_CHECKING
import pymongo
import pytest
+from tests_common.test_utils.compat import connection_as_json
from airflow.exceptions import AirflowConfigException, AirflowProviderDeprecationWarning
from airflow.models import Connection
from airflow.providers.mongo.hooks.mongo import MongoHook
-from dev.tests_common.test_utils.compat import connection_as_json
-
pytestmark = pytest.mark.db_test
if TYPE_CHECKING:
diff --git a/providers/tests/mysql/hooks/test_mysql.py b/providers/tests/mysql/hooks/test_mysql.py
index 23748ae9ca..49358ac30d 100644
--- a/providers/tests/mysql/hooks/test_mysql.py
+++ b/providers/tests/mysql/hooks/test_mysql.py
@@ -36,9 +36,9 @@ except ImportError:
pytest.skip("MySQL not available", allow_module_level=True)
-from airflow.utils import timezone
+from tests_common.test_utils.asserts import assert_equal_ignore_multiple_spaces
-from dev.tests_common.test_utils.asserts import assert_equal_ignore_multiple_spaces
+from airflow.utils import timezone
SSL_DICT = {"cert": "/tmp/client-cert.pem", "ca": "/tmp/server-ca.pem", "key": "/tmp/client-key.pem"}
diff --git a/providers/tests/openlineage/extractors/test_bash.py b/providers/tests/openlineage/extractors/test_bash.py
index d4fcdf7af2..8782ce161c 100644
--- a/providers/tests/openlineage/extractors/test_bash.py
+++ b/providers/tests/openlineage/extractors/test_bash.py
@@ -23,13 +23,12 @@ from unittest.mock import patch
import pytest
from openlineage.client.facet_v2 import source_code_job
+from tests_common.test_utils.compat import BashOperator
from airflow import DAG
from airflow.exceptions import AirflowProviderDeprecationWarning
from airflow.providers.openlineage.extractors.bash import BashExtractor
-from dev.tests_common.test_utils.compat import BashOperator
-
pytestmark = pytest.mark.db_test
with DAG(
diff --git a/providers/tests/openlineage/extractors/test_manager.py b/providers/tests/openlineage/extractors/test_manager.py
index 6bbf303e32..dbd4a8e403 100644
--- a/providers/tests/openlineage/extractors/test_manager.py
+++ b/providers/tests/openlineage/extractors/test_manager.py
@@ -24,6 +24,7 @@ from unittest.mock import MagicMock
import pytest
from openlineage.client.event_v2 import Dataset as OpenLineageDataset
from openlineage.client.facet_v2 import documentation_dataset, ownership_dataset, schema_dataset
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
from airflow.io.path import ObjectStoragePath
from airflow.lineage.entities import Column, File, Table, User
@@ -35,8 +36,6 @@ from airflow.providers.openlineage.extractors.manager import ExtractorManager
from airflow.providers.openlineage.utils.utils import Asset
from airflow.utils.state import State
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
-
if TYPE_CHECKING:
from airflow.utils.context import Context
diff --git a/providers/tests/openlineage/extractors/test_python.py b/providers/tests/openlineage/extractors/test_python.py
index ef4fc0b861..92a9d073a1 100644
--- a/providers/tests/openlineage/extractors/test_python.py
+++ b/providers/tests/openlineage/extractors/test_python.py
@@ -25,14 +25,13 @@ from unittest.mock import patch
import pytest
from openlineage.client.facet_v2 import source_code_job
+from tests_common.test_utils.compat import BashOperator
from airflow import DAG
from airflow.exceptions import AirflowProviderDeprecationWarning
from airflow.operators.python import PythonOperator
from airflow.providers.openlineage.extractors.python import PythonExtractor
-from dev.tests_common.test_utils.compat import BashOperator
-
pytestmark = pytest.mark.db_test
dag = DAG(
diff --git a/providers/tests/openlineage/plugins/test_adapter.py b/providers/tests/openlineage/plugins/test_adapter.py
index 88f2250c63..8d57fad296 100644
--- a/providers/tests/openlineage/plugins/test_adapter.py
+++ b/providers/tests/openlineage/plugins/test_adapter.py
@@ -36,6 +36,8 @@ from openlineage.client.facet_v2 import (
processing_engine_run,
sql_job,
)
+from tests_common.test_utils.compat import BashOperator
+from tests_common.test_utils.config import conf_vars
from airflow import DAG
from airflow.models.dagrun import DagRun, DagRunState
@@ -52,9 +54,6 @@ from airflow.providers.openlineage.plugins.facets import (
from airflow.providers.openlineage.utils.utils import get_airflow_job_facet
from airflow.utils.task_group import TaskGroup
-from dev.tests_common.test_utils.compat import BashOperator
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/openlineage/plugins/test_execution.py b/providers/tests/openlineage/plugins/test_execution.py
index c308047b1b..09262adc4a 100644
--- a/providers/tests/openlineage/plugins/test_execution.py
+++ b/providers/tests/openlineage/plugins/test_execution.py
@@ -26,6 +26,8 @@ from pathlib import Path
from unittest import mock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS, AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.config import conf_vars
from airflow.jobs.job import Job
from airflow.jobs.local_task_job_runner import LocalTaskJobRunner
@@ -37,9 +39,6 @@ from airflow.task.task_runner.standard_task_runner import StandardTaskRunner
from airflow.utils import timezone
from airflow.utils.state import State
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS, AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
if AIRFLOW_V_3_0_PLUS:
from airflow.utils.types import DagRunTriggeredByType
diff --git a/providers/tests/openlineage/plugins/test_listener.py b/providers/tests/openlineage/plugins/test_listener.py
index 0c1651ccf0..b859160cf8 100644
--- a/providers/tests/openlineage/plugins/test_listener.py
+++ b/providers/tests/openlineage/plugins/test_listener.py
@@ -29,6 +29,8 @@ import pytest
from openlineage.client import OpenLineageClient
from openlineage.client.transport import ConsoleTransport
from openlineage.client.transport.console import ConsoleConfig
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS, AIRFLOW_V_3_0_PLUS
+from tests_common.test_utils.config import conf_vars
from airflow.models import DAG, DagRun, TaskInstance
from airflow.models.baseoperator import BaseOperator
@@ -39,9 +41,6 @@ from airflow.providers.openlineage.plugins.listener import OpenLineageListener
from airflow.providers.openlineage.utils.selective_enable import disable_lineage, enable_lineage
from airflow.utils.state import DagRunState, State
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS, AIRFLOW_V_3_0_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
if AIRFLOW_V_3_0_PLUS:
from airflow.utils.types import DagRunTriggeredByType
diff --git a/providers/tests/openlineage/plugins/test_openlineage.py b/providers/tests/openlineage/plugins/test_openlineage.py
index 8685326a29..e1c36001eb 100644
--- a/providers/tests/openlineage/plugins/test_openlineage.py
+++ b/providers/tests/openlineage/plugins/test_openlineage.py
@@ -22,9 +22,8 @@ import sys
from unittest.mock import patch
import pytest
-
-from dev.tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
-from dev.tests_common.test_utils.config import conf_vars
+from tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
+from tests_common.test_utils.config import conf_vars
@pytest.mark.skipif(
diff --git a/providers/tests/openlineage/plugins/test_utils.py b/providers/tests/openlineage/plugins/test_utils.py
index 444839ccae..f0685c1351 100644
--- a/providers/tests/openlineage/plugins/test_utils.py
+++ b/providers/tests/openlineage/plugins/test_utils.py
@@ -27,6 +27,7 @@ import pytest
from attrs import define
from openlineage.client.utils import RedactMixin
from pkg_resources import parse_version
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS, AIRFLOW_V_3_0_PLUS, BashOperator
from airflow.models import DAG as AIRFLOW_DAG, DagModel
from airflow.providers.openlineage.plugins.facets import AirflowDebugRunFacet
@@ -44,8 +45,6 @@ from airflow.utils import timezone
from airflow.utils.log.secrets_masker import _secrets_masker
from airflow.utils.state import State
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS, AIRFLOW_V_3_0_PLUS, BashOperator
-
if AIRFLOW_V_3_0_PLUS:
from airflow.utils.types import DagRunTriggeredByType
diff --git a/providers/tests/openlineage/test_conf.py b/providers/tests/openlineage/test_conf.py
index 7f78a6a4c2..b1dd57d4d8 100644
--- a/providers/tests/openlineage/test_conf.py
+++ b/providers/tests/openlineage/test_conf.py
@@ -20,6 +20,7 @@ import os
from unittest import mock
import pytest
+from tests_common.test_utils.config import conf_vars, env_vars
from airflow.exceptions import AirflowConfigException
from airflow.providers.openlineage.conf import (
@@ -39,8 +40,6 @@ from airflow.providers.openlineage.conf import (
transport,
)
-from dev.tests_common.test_utils.config import conf_vars, env_vars
-
_CONFIG_SECTION = "openlineage"
_VAR_CONFIG_PATH = "OPENLINEAGE_CONFIG"
_CONFIG_OPTION_CONFIG_PATH = "config_path"
diff --git a/providers/tests/openlineage/utils/test_utils.py b/providers/tests/openlineage/utils/test_utils.py
index 6cf7904546..5adeceb440 100644
--- a/providers/tests/openlineage/utils/test_utils.py
+++ b/providers/tests/openlineage/utils/test_utils.py
@@ -21,6 +21,9 @@ import datetime
import pathlib
from unittest.mock import MagicMock, patch
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS, BashOperator
+from tests_common.test_utils.mock_operators import MockOperator
+
from airflow import DAG
from airflow.decorators import task
from airflow.models.baseoperator import BaseOperator
@@ -44,9 +47,6 @@ from airflow.serialization.serialized_objects import SerializedBaseOperator
from airflow.utils.task_group import TaskGroup
from airflow.utils.types import DagRunType
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS, BashOperator
-from dev.tests_common.test_utils.mock_operators import MockOperator
-
BASH_OPERATOR_PATH = "airflow.providers.standard.operators.bash"
if not AIRFLOW_V_2_10_PLUS:
BASH_OPERATOR_PATH = "airflow.operators.bash"
@@ -191,7 +191,7 @@ def test_get_fully_qualified_class_name_mapped_operator():
mapped = MockOperator.partial(task_id="task_2").expand(arg2=["a", "b", "c"])
assert isinstance(mapped, MappedOperator)
mapped_op_path = get_fully_qualified_class_name(mapped)
- assert mapped_op_path == "dev.tests_common.test_utils.mock_operators.MockOperator"
+ assert mapped_op_path == "tests_common.test_utils.mock_operators.MockOperator"
def test_get_fully_qualified_class_name_bash_operator():
diff --git a/providers/tests/opensearch/log/test_os_task_handler.py b/providers/tests/opensearch/log/test_os_task_handler.py
index 1ffa0f5251..81ac75c855 100644
--- a/providers/tests/opensearch/log/test_os_task_handler.py
+++ b/providers/tests/opensearch/log/test_os_task_handler.py
@@ -32,6 +32,8 @@ import pytest
opensearchpy = pytest.importorskip("opensearchpy")
from opensearchpy.exceptions import NotFoundError
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.db import clear_db_dags, clear_db_runs
from airflow.configuration import conf
from airflow.providers.opensearch.log.os_response import OpensearchResponse
@@ -44,8 +46,6 @@ from airflow.utils import timezone
from airflow.utils.state import DagRunState, TaskInstanceState
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.db import clear_db_dags, clear_db_runs
from providers.tests.opensearch.conftest import MockClient
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/redis/log/test_redis_task_handler.py b/providers/tests/redis/log/test_redis_task_handler.py
index bc7345df9a..0d83f3a303 100644
--- a/providers/tests/redis/log/test_redis_task_handler.py
+++ b/providers/tests/redis/log/test_redis_task_handler.py
@@ -21,6 +21,7 @@ import logging
from unittest.mock import patch
import pytest
+from tests_common.test_utils.config import conf_vars
from airflow.models import DAG, DagRun, TaskInstance
from airflow.operators.empty import EmptyOperator
@@ -29,8 +30,6 @@ from airflow.utils.session import create_session
from airflow.utils.state import State
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/sftp/operators/test_sftp.py b/providers/tests/sftp/operators/test_sftp.py
index e6c90c2c74..9d5689c881 100644
--- a/providers/tests/sftp/operators/test_sftp.py
+++ b/providers/tests/sftp/operators/test_sftp.py
@@ -25,6 +25,7 @@ from unittest import mock
import paramiko
import pytest
+from tests_common.test_utils.config import conf_vars
from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
from airflow.models import DAG, Connection
@@ -36,8 +37,6 @@ from airflow.providers.ssh.operators.ssh import SSHOperator
from airflow.utils import timezone
from airflow.utils.timezone import datetime
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/smtp/hooks/test_smtp.py b/providers/tests/smtp/hooks/test_smtp.py
index 04a20e0ca6..fe9471c424 100644
--- a/providers/tests/smtp/hooks/test_smtp.py
+++ b/providers/tests/smtp/hooks/test_smtp.py
@@ -25,14 +25,13 @@ from email.mime.application import MIMEApplication
from unittest.mock import Mock, patch
import pytest
+from tests_common.test_utils.config import conf_vars
from airflow.models import Connection
from airflow.providers.smtp.hooks.smtp import SmtpHook
from airflow.utils import db
from airflow.utils.session import create_session
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/smtp/notifications/test_smtp.py b/providers/tests/smtp/notifications/test_smtp.py
index 75e1ebf530..594fa19dfe 100644
--- a/providers/tests/smtp/notifications/test_smtp.py
+++ b/providers/tests/smtp/notifications/test_smtp.py
@@ -21,6 +21,8 @@ import tempfile
from unittest import mock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
+from tests_common.test_utils.config import conf_vars
from airflow.configuration import conf
from airflow.models import SlaMiss
@@ -32,9 +34,6 @@ from airflow.providers.smtp.notifications.smtp import (
)
from airflow.utils import timezone
-from dev.tests_common.test_utils.compat import AIRFLOW_V_2_10_PLUS
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
SMTP_API_DEFAULT_CONN_ID = SmtpHook.default_conn_name
diff --git a/providers/tests/ssh/operators/test_ssh.py b/providers/tests/ssh/operators/test_ssh.py
index b304a24882..55431bd3b6 100644
--- a/providers/tests/ssh/operators/test_ssh.py
+++ b/providers/tests/ssh/operators/test_ssh.py
@@ -24,6 +24,7 @@ from unittest import mock
import pytest
from paramiko.client import SSHClient
+from tests_common.test_utils.config import conf_vars
from airflow.exceptions import AirflowException, AirflowSkipException, AirflowTaskTimeout
from airflow.models import TaskInstance
@@ -32,8 +33,6 @@ from airflow.providers.ssh.operators.ssh import SSHOperator
from airflow.utils.timezone import datetime
from airflow.utils.types import NOTSET
-from dev.tests_common.test_utils.config import conf_vars
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/standard/operators/test_bash.py b/providers/tests/standard/operators/test_bash.py
index 305651ee59..f040442ecc 100644
--- a/providers/tests/standard/operators/test_bash.py
+++ b/providers/tests/standard/operators/test_bash.py
@@ -26,6 +26,7 @@ from time import sleep
from unittest import mock
import pytest
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
from airflow.exceptions import AirflowException, AirflowSkipException, AirflowTaskTimeout
from airflow.providers.standard.operators.bash import BashOperator
@@ -33,8 +34,6 @@ from airflow.utils import timezone
from airflow.utils.state import State
from airflow.utils.types import DagRunType
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-
if AIRFLOW_V_3_0_PLUS:
from airflow.utils.types import DagRunTriggeredByType
diff --git a/providers/tests/standard/operators/test_datetime.py b/providers/tests/standard/operators/test_datetime.py
index 4fbb0863d3..da7f98f0e3 100644
--- a/providers/tests/standard/operators/test_datetime.py
+++ b/providers/tests/standard/operators/test_datetime.py
@@ -21,6 +21,7 @@ import datetime
import pytest
import time_machine
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
from airflow.exceptions import AirflowException
from airflow.models.dagrun import DagRun
@@ -31,8 +32,6 @@ from airflow.utils import timezone
from airflow.utils.session import create_session
from airflow.utils.state import State
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-
if AIRFLOW_V_3_0_PLUS:
from airflow.utils.types import DagRunTriggeredByType
diff --git a/providers/tests/standard/operators/test_weekday.py b/providers/tests/standard/operators/test_weekday.py
index 6c20299d15..576fd10b54 100644
--- a/providers/tests/standard/operators/test_weekday.py
+++ b/providers/tests/standard/operators/test_weekday.py
@@ -21,6 +21,7 @@ import datetime
import pytest
import time_machine
+from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
from airflow.exceptions import AirflowException
from airflow.models.dagrun import DagRun
@@ -33,8 +34,6 @@ from airflow.utils.session import create_session
from airflow.utils.state import State
from airflow.utils.weekday import WeekDay
-from dev.tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
-
if AIRFLOW_V_3_0_PLUS:
from airflow.utils.types import DagRunTriggeredByType
diff --git a/providers/tests/standard/sensors/test_weekday.py b/providers/tests/standard/sensors/test_weekday.py
index ef0091a9fd..5bdbf4bb79 100644
--- a/providers/tests/standard/sensors/test_weekday.py
+++ b/providers/tests/standard/sensors/test_weekday.py
@@ -20,6 +20,7 @@ from __future__ import annotations
from datetime import timedelta
import pytest
+from tests_common.test_utils import db
from airflow.exceptions import AirflowSensorTimeout
from airflow.models import DagBag
@@ -28,8 +29,6 @@ from airflow.providers.standard.sensors.weekday import DayOfWeekSensor
from airflow.utils.timezone import datetime
from airflow.utils.weekday import WeekDay
-from dev.tests_common.test_utils import db
-
pytestmark = pytest.mark.db_test
diff --git a/providers/tests/system/airbyte/example_airbyte_trigger_job.py b/providers/tests/system/airbyte/example_airbyte_trigger_job.py
index 1dd3f01d72..240a293bbd 100644
--- a/providers/tests/system/airbyte/example_airbyte_trigger_job.py
+++ b/providers/tests/system/airbyte/example_airbyte_trigger_job.py
@@ -61,7 +61,7 @@ with DAG(
# Task dependency created via `XComArgs`:
# async_source_destination >> airbyte_sensor
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/alibaba/example_adb_spark_batch.py b/providers/tests/system/alibaba/example_adb_spark_batch.py
index 35e03a0888..e8447cefc4 100644
--- a/providers/tests/system/alibaba/example_adb_spark_batch.py
+++ b/providers/tests/system/alibaba/example_adb_spark_batch.py
@@ -50,14 +50,14 @@ with DAG(
spark_pi >> spark_lr
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
# [END howto_operator_adb_spark_batch]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/alibaba/example_adb_spark_sql.py b/providers/tests/system/alibaba/example_adb_spark_sql.py
index ac29330d45..cc587e424b 100644
--- a/providers/tests/system/alibaba/example_adb_spark_sql.py
+++ b/providers/tests/system/alibaba/example_adb_spark_sql.py
@@ -42,14 +42,14 @@ with DAG(
show_databases >> show_tables
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
# [END howto_operator_adb_spark_sql]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/alibaba/example_oss_bucket.py b/providers/tests/system/alibaba/example_oss_bucket.py
index 4870a3b577..cfe7a42efc 100644
--- a/providers/tests/system/alibaba/example_oss_bucket.py
+++ b/providers/tests/system/alibaba/example_oss_bucket.py
@@ -41,14 +41,14 @@ with DAG(
create_bucket >> delete_bucket
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
# [END howto_operator_oss_bucket]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/alibaba/example_oss_object.py b/providers/tests/system/alibaba/example_oss_object.py
index 7305c05bf8..dfa59917fd 100644
--- a/providers/tests/system/alibaba/example_oss_object.py
+++ b/providers/tests/system/alibaba/example_oss_object.py
@@ -65,14 +65,14 @@ with DAG(
create_object >> download_object >> delete_object >> delete_batch_object
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/CONTRIBUTING.md b/providers/tests/system/amazon/CONTRIBUTING.md
index f12062aa7f..4abdf410dd 100644
--- a/providers/tests/system/amazon/CONTRIBUTING.md
+++ b/providers/tests/system/amazon/CONTRIBUTING.md
@@ -212,7 +212,7 @@ chain(
task2, # task2 has trigger rule "all done" defined
)
-from dev.tests_common.test_utils.watcher import watcher
+from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
diff --git a/providers/tests/system/amazon/aws/example_appflow.py b/providers/tests/system/amazon/aws/example_appflow.py
index db9dd9a428..262b6cb803 100644
--- a/providers/tests/system/amazon/aws/example_appflow.py
+++ b/providers/tests/system/amazon/aws/example_appflow.py
@@ -111,13 +111,13 @@ with DAG(
should_be_skipped,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_appflow_run.py b/providers/tests/system/amazon/aws/example_appflow_run.py
index d1151d1dd5..106fda1bb7 100644
--- a/providers/tests/system/amazon/aws/example_appflow_run.py
+++ b/providers/tests/system/amazon/aws/example_appflow_run.py
@@ -193,13 +193,13 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_athena.py b/providers/tests/system/amazon/aws/example_athena.py
index 2ee1c11ab7..201e017d12 100644
--- a/providers/tests/system/amazon/aws/example_athena.py
+++ b/providers/tests/system/amazon/aws/example_athena.py
@@ -174,14 +174,14 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_azure_blob_to_s3.py b/providers/tests/system/amazon/aws/example_azure_blob_to_s3.py
index 78a55a6a7f..3834365622 100644
--- a/providers/tests/system/amazon/aws/example_azure_blob_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_azure_blob_to_s3.py
@@ -72,13 +72,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_batch.py b/providers/tests/system/amazon/aws/example_batch.py
index da035050a7..828815a2e1 100644
--- a/providers/tests/system/amazon/aws/example_batch.py
+++ b/providers/tests/system/amazon/aws/example_batch.py
@@ -283,13 +283,13 @@ with DAG(
log_cleanup,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_bedrock.py b/providers/tests/system/amazon/aws/example_bedrock.py
index 8e67ee4d38..3caa9f3a35 100644
--- a/providers/tests/system/amazon/aws/example_bedrock.py
+++ b/providers/tests/system/amazon/aws/example_bedrock.py
@@ -218,14 +218,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py b/providers/tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py
index 0d9eb27846..094cb854c3 100644
--- a/providers/tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py
+++ b/providers/tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py
@@ -594,14 +594,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_cloudformation.py b/providers/tests/system/amazon/aws/example_cloudformation.py
index 38827eebbd..7553f5d9bd 100644
--- a/providers/tests/system/amazon/aws/example_cloudformation.py
+++ b/providers/tests/system/amazon/aws/example_cloudformation.py
@@ -108,14 +108,14 @@ with DAG(
wait_for_stack_delete,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_comprehend.py b/providers/tests/system/amazon/aws/example_comprehend.py
index d8a1f9af88..a4e3bdd485 100644
--- a/providers/tests/system/amazon/aws/example_comprehend.py
+++ b/providers/tests/system/amazon/aws/example_comprehend.py
@@ -126,13 +126,13 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_comprehend_document_classifier.py b/providers/tests/system/amazon/aws/example_comprehend_document_classifier.py
index f5308265a8..160e3f3cad 100644
--- a/providers/tests/system/amazon/aws/example_comprehend_document_classifier.py
+++ b/providers/tests/system/amazon/aws/example_comprehend_document_classifier.py
@@ -231,13 +231,13 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_datasync.py b/providers/tests/system/amazon/aws/example_datasync.py
index a8363fe81b..8d25fcf9c3 100644
--- a/providers/tests/system/amazon/aws/example_datasync.py
+++ b/providers/tests/system/amazon/aws/example_datasync.py
@@ -235,13 +235,13 @@ with DAG(
delete_s3_bucket_destination,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_dms.py b/providers/tests/system/amazon/aws/example_dms.py
index b2e99daf62..5d15aa0c4e 100644
--- a/providers/tests/system/amazon/aws/example_dms.py
+++ b/providers/tests/system/amazon/aws/example_dms.py
@@ -429,13 +429,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_dynamodb.py b/providers/tests/system/amazon/aws/example_dynamodb.py
index 3d709c36b0..3cfa15ecaa 100644
--- a/providers/tests/system/amazon/aws/example_dynamodb.py
+++ b/providers/tests/system/amazon/aws/example_dynamodb.py
@@ -117,14 +117,14 @@ with DAG(
delete_table,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_dynamodb_to_s3.py b/providers/tests/system/amazon/aws/example_dynamodb_to_s3.py
index 86a5d76c95..989fe79b2d 100644
--- a/providers/tests/system/amazon/aws/example_dynamodb_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_dynamodb_to_s3.py
@@ -256,13 +256,13 @@ with DAG(
delete_table,
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_ec2.py b/providers/tests/system/amazon/aws/example_ec2.py
index e55adab775..750ec19c27 100644
--- a/providers/tests/system/amazon/aws/example_ec2.py
+++ b/providers/tests/system/amazon/aws/example_ec2.py
@@ -202,13 +202,13 @@ with DAG(
delete_key_pair(key_name),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_ecs.py b/providers/tests/system/amazon/aws/example_ecs.py
index e1aad99b5e..0f9118ae34 100644
--- a/providers/tests/system/amazon/aws/example_ecs.py
+++ b/providers/tests/system/amazon/aws/example_ecs.py
@@ -212,14 +212,14 @@ with DAG(
clean_logs(log_group_name),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_ecs_fargate.py b/providers/tests/system/amazon/aws/example_ecs_fargate.py
index ca592234aa..df3afbce7d 100644
--- a/providers/tests/system/amazon/aws/example_ecs_fargate.py
+++ b/providers/tests/system/amazon/aws/example_ecs_fargate.py
@@ -156,13 +156,13 @@ with DAG(
delete_cluster(cluster_name),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_eks_templated.py b/providers/tests/system/amazon/aws/example_eks_templated.py
index 3b1ba50597..de4d5e621e 100644
--- a/providers/tests/system/amazon/aws/example_eks_templated.py
+++ b/providers/tests/system/amazon/aws/example_eks_templated.py
@@ -148,13 +148,13 @@ with DAG(
await_delete_cluster,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_eks_with_fargate_in_one_step.py b/providers/tests/system/amazon/aws/example_eks_with_fargate_in_one_step.py
index 27ee2be0d7..a267463c11 100644
--- a/providers/tests/system/amazon/aws/example_eks_with_fargate_in_one_step.py
+++ b/providers/tests/system/amazon/aws/example_eks_with_fargate_in_one_step.py
@@ -141,14 +141,14 @@ with DAG(
await_delete_cluster,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_eks_with_fargate_profile.py b/providers/tests/system/amazon/aws/example_eks_with_fargate_profile.py
index 34cbed1645..1348b352cc 100644
--- a/providers/tests/system/amazon/aws/example_eks_with_fargate_profile.py
+++ b/providers/tests/system/amazon/aws/example_eks_with_fargate_profile.py
@@ -175,13 +175,13 @@ with DAG(
await_delete_cluster,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_eks_with_nodegroup_in_one_step.py b/providers/tests/system/amazon/aws/example_eks_with_nodegroup_in_one_step.py
index 11f8436885..07538d94d1 100644
--- a/providers/tests/system/amazon/aws/example_eks_with_nodegroup_in_one_step.py
+++ b/providers/tests/system/amazon/aws/example_eks_with_nodegroup_in_one_step.py
@@ -155,14 +155,14 @@ with DAG(
delete_launch_template(launch_template_name),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_eks_with_nodegroups.py b/providers/tests/system/amazon/aws/example_eks_with_nodegroups.py
index 8356a8ee43..98448767d9 100644
--- a/providers/tests/system/amazon/aws/example_eks_with_nodegroups.py
+++ b/providers/tests/system/amazon/aws/example_eks_with_nodegroups.py
@@ -198,14 +198,14 @@ with DAG(
delete_launch_template(launch_template_name),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_emr.py b/providers/tests/system/amazon/aws/example_emr.py
index 44c28630eb..9a29667b5d 100644
--- a/providers/tests/system/amazon/aws/example_emr.py
+++ b/providers/tests/system/amazon/aws/example_emr.py
@@ -227,14 +227,14 @@ with DAG(
]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_emr_eks.py b/providers/tests/system/amazon/aws/example_emr_eks.py
index d95645fd30..cafaf09ed7 100644
--- a/providers/tests/system/amazon/aws/example_emr_eks.py
+++ b/providers/tests/system/amazon/aws/example_emr_eks.py
@@ -320,13 +320,13 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_emr_notebook_execution.py b/providers/tests/system/amazon/aws/example_emr_notebook_execution.py
index 48d2b14c96..681e97048a 100644
--- a/providers/tests/system/amazon/aws/example_emr_notebook_execution.py
+++ b/providers/tests/system/amazon/aws/example_emr_notebook_execution.py
@@ -112,13 +112,13 @@ with DAG(
wait_for_execution_finish,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_emr_serverless.py b/providers/tests/system/amazon/aws/example_emr_serverless.py
index 4901fa6736..9059f0e1b1 100644
--- a/providers/tests/system/amazon/aws/example_emr_serverless.py
+++ b/providers/tests/system/amazon/aws/example_emr_serverless.py
@@ -155,13 +155,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_eventbridge.py b/providers/tests/system/amazon/aws/example_eventbridge.py
index 5470bd0ca7..6f31fc9ce3 100644
--- a/providers/tests/system/amazon/aws/example_eventbridge.py
+++ b/providers/tests/system/amazon/aws/example_eventbridge.py
@@ -81,7 +81,7 @@ with DAG(
chain(test_context, put_events, put_rule, enable_rule, disable_rule)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_ftp_to_s3.py b/providers/tests/system/amazon/aws/example_ftp_to_s3.py
index 98a37e197f..b3e0d2ccf4 100644
--- a/providers/tests/system/amazon/aws/example_ftp_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_ftp_to_s3.py
@@ -72,13 +72,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_gcs_to_s3.py b/providers/tests/system/amazon/aws/example_gcs_to_s3.py
index 97ed8dfda3..5257b44080 100644
--- a/providers/tests/system/amazon/aws/example_gcs_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_gcs_to_s3.py
@@ -115,13 +115,13 @@ with DAG(
delete_gcs_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_glacier_to_gcs.py b/providers/tests/system/amazon/aws/example_glacier_to_gcs.py
index 28c061ae05..9281d3c92e 100644
--- a/providers/tests/system/amazon/aws/example_glacier_to_gcs.py
+++ b/providers/tests/system/amazon/aws/example_glacier_to_gcs.py
@@ -108,13 +108,13 @@ with DAG(
delete_vault(vault_name),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_glue.py b/providers/tests/system/amazon/aws/example_glue.py
index 5314d26eff..3d042c79d4 100644
--- a/providers/tests/system/amazon/aws/example_glue.py
+++ b/providers/tests/system/amazon/aws/example_glue.py
@@ -217,13 +217,13 @@ with DAG(
log_cleanup,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_glue_data_quality.py b/providers/tests/system/amazon/aws/example_glue_data_quality.py
index 4d05d6e22b..3c1bdb378b 100644
--- a/providers/tests/system/amazon/aws/example_glue_data_quality.py
+++ b/providers/tests/system/amazon/aws/example_glue_data_quality.py
@@ -198,13 +198,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_glue_data_quality_with_recommendation.py b/providers/tests/system/amazon/aws/example_glue_data_quality_with_recommendation.py
index fdb3cb8e63..a0dd4ebba7 100644
--- a/providers/tests/system/amazon/aws/example_glue_data_quality_with_recommendation.py
+++ b/providers/tests/system/amazon/aws/example_glue_data_quality_with_recommendation.py
@@ -197,13 +197,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_glue_databrew.py b/providers/tests/system/amazon/aws/example_glue_databrew.py
index 3218b465c2..95799666f4 100644
--- a/providers/tests/system/amazon/aws/example_glue_databrew.py
+++ b/providers/tests/system/amazon/aws/example_glue_databrew.py
@@ -155,14 +155,14 @@ with DAG(DAG_ID, schedule="@once", start_date=pendulum.datetime(2023, 1, 1, tz="
delete_output_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_google_api_sheets_to_s3.py b/providers/tests/system/amazon/aws/example_google_api_sheets_to_s3.py
index 7ea7ae9d07..24a4eba54d 100644
--- a/providers/tests/system/amazon/aws/example_google_api_sheets_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_google_api_sheets_to_s3.py
@@ -83,13 +83,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_google_api_youtube_to_s3.py b/providers/tests/system/amazon/aws/example_google_api_youtube_to_s3.py
index ae4f721567..3d5ff917bf 100644
--- a/providers/tests/system/amazon/aws/example_google_api_youtube_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_google_api_youtube_to_s3.py
@@ -197,13 +197,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_hive_to_dynamodb.py b/providers/tests/system/amazon/aws/example_hive_to_dynamodb.py
index 5582c13b85..0c0978361b 100644
--- a/providers/tests/system/amazon/aws/example_hive_to_dynamodb.py
+++ b/providers/tests/system/amazon/aws/example_hive_to_dynamodb.py
@@ -150,13 +150,13 @@ with DAG(
delete_dynamodb_table(dynamodb_table_name),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_http_to_s3.py b/providers/tests/system/amazon/aws/example_http_to_s3.py
index cfbc131ee5..612d68ef36 100644
--- a/providers/tests/system/amazon/aws/example_http_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_http_to_s3.py
@@ -119,13 +119,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_imap_attachment_to_s3.py b/providers/tests/system/amazon/aws/example_imap_attachment_to_s3.py
index b43023115c..c376778b0d 100644
--- a/providers/tests/system/amazon/aws/example_imap_attachment_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_imap_attachment_to_s3.py
@@ -90,13 +90,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_kinesis_analytics.py b/providers/tests/system/amazon/aws/example_kinesis_analytics.py
index 600710cc43..b4d4566fc7 100644
--- a/providers/tests/system/amazon/aws/example_kinesis_analytics.py
+++ b/providers/tests/system/amazon/aws/example_kinesis_analytics.py
@@ -263,13 +263,13 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_lambda.py b/providers/tests/system/amazon/aws/example_lambda.py
index fd346a3458..767fae8638 100644
--- a/providers/tests/system/amazon/aws/example_lambda.py
+++ b/providers/tests/system/amazon/aws/example_lambda.py
@@ -129,13 +129,13 @@ with DAG(
log_cleanup,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_local_to_s3.py b/providers/tests/system/amazon/aws/example_local_to_s3.py
index 47344c429b..cdd9748974 100644
--- a/providers/tests/system/amazon/aws/example_local_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_local_to_s3.py
@@ -91,14 +91,14 @@ with DAG(
delete_temp_file(),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_mongo_to_s3.py b/providers/tests/system/amazon/aws/example_mongo_to_s3.py
index 3e5bc318e5..6fa5f612b0 100644
--- a/providers/tests/system/amazon/aws/example_mongo_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_mongo_to_s3.py
@@ -83,13 +83,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_neptune.py b/providers/tests/system/amazon/aws/example_neptune.py
index 4685840b42..7682a65b65 100644
--- a/providers/tests/system/amazon/aws/example_neptune.py
+++ b/providers/tests/system/amazon/aws/example_neptune.py
@@ -77,13 +77,13 @@ with DAG(
# TEST TEARDOWN
delete_cluster(cluster_id),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_quicksight.py b/providers/tests/system/amazon/aws/example_quicksight.py
index ec5cd38003..eb53ffe60e 100644
--- a/providers/tests/system/amazon/aws/example_quicksight.py
+++ b/providers/tests/system/amazon/aws/example_quicksight.py
@@ -218,13 +218,13 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_rds_event.py b/providers/tests/system/amazon/aws/example_rds_event.py
index 79d70c9d39..58ab603326 100644
--- a/providers/tests/system/amazon/aws/example_rds_event.py
+++ b/providers/tests/system/amazon/aws/example_rds_event.py
@@ -119,14 +119,14 @@ with DAG(
delete_sns_topic(sns_topic),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_rds_export.py b/providers/tests/system/amazon/aws/example_rds_export.py
index 34638f030f..756105d802 100644
--- a/providers/tests/system/amazon/aws/example_rds_export.py
+++ b/providers/tests/system/amazon/aws/example_rds_export.py
@@ -178,13 +178,13 @@ with DAG(
delete_db_instance,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_rds_instance.py b/providers/tests/system/amazon/aws/example_rds_instance.py
index ce49f73c1f..b2a9e7482e 100644
--- a/providers/tests/system/amazon/aws/example_rds_instance.py
+++ b/providers/tests/system/amazon/aws/example_rds_instance.py
@@ -111,14 +111,14 @@ with DAG(
delete_db_instance,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_rds_snapshot.py b/providers/tests/system/amazon/aws/example_rds_snapshot.py
index c58c1db54d..b6b0c468e5 100644
--- a/providers/tests/system/amazon/aws/example_rds_snapshot.py
+++ b/providers/tests/system/amazon/aws/example_rds_snapshot.py
@@ -138,14 +138,14 @@ with DAG(
delete_db_instance,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_redshift.py b/providers/tests/system/amazon/aws/example_redshift.py
index 986bce5a07..146a20da45 100644
--- a/providers/tests/system/amazon/aws/example_redshift.py
+++ b/providers/tests/system/amazon/aws/example_redshift.py
@@ -248,13 +248,13 @@ with DAG(
delete_cluster_snapshot,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_redshift_s3_transfers.py b/providers/tests/system/amazon/aws/example_redshift_s3_transfers.py
index 2a553eae8c..536f9b9715 100644
--- a/providers/tests/system/amazon/aws/example_redshift_s3_transfers.py
+++ b/providers/tests/system/amazon/aws/example_redshift_s3_transfers.py
@@ -318,13 +318,13 @@ with DAG(
delete_cluster,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_s3.py b/providers/tests/system/amazon/aws/example_s3.py
index d016076022..a9c93fdd5b 100644
--- a/providers/tests/system/amazon/aws/example_s3.py
+++ b/providers/tests/system/amazon/aws/example_s3.py
@@ -323,13 +323,13 @@ with DAG(
delete_bucket,
delete_bucket_2,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_s3_to_dynamodb.py b/providers/tests/system/amazon/aws/example_s3_to_dynamodb.py
index efd762cfd9..4bc4fa6b93 100644
--- a/providers/tests/system/amazon/aws/example_s3_to_dynamodb.py
+++ b/providers/tests/system/amazon/aws/example_s3_to_dynamodb.py
@@ -181,13 +181,13 @@ with DAG(
delete_new_table,
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_s3_to_ftp.py b/providers/tests/system/amazon/aws/example_s3_to_ftp.py
index 0ffb4722ba..4dd134f95c 100644
--- a/providers/tests/system/amazon/aws/example_s3_to_ftp.py
+++ b/providers/tests/system/amazon/aws/example_s3_to_ftp.py
@@ -71,13 +71,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_s3_to_sftp.py b/providers/tests/system/amazon/aws/example_s3_to_sftp.py
index 5270d3ec7d..dc7d010628 100644
--- a/providers/tests/system/amazon/aws/example_s3_to_sftp.py
+++ b/providers/tests/system/amazon/aws/example_s3_to_sftp.py
@@ -71,13 +71,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_s3_to_sql.py b/providers/tests/system/amazon/aws/example_s3_to_sql.py
index e7e8a5e954..d7f452bfeb 100644
--- a/providers/tests/system/amazon/aws/example_s3_to_sql.py
+++ b/providers/tests/system/amazon/aws/example_s3_to_sql.py
@@ -18,6 +18,8 @@ from __future__ import annotations
from datetime import datetime
+from tests_common.test_utils.watcher import watcher
+
from airflow import settings
from airflow.decorators import task
from airflow.models import Connection
@@ -40,7 +42,6 @@ from airflow.providers.amazon.aws.transfers.s3_to_sql import S3ToSqlOperator
from airflow.providers.common.sql.operators.sql import SQLTableCheckOperator
from airflow.utils.trigger_rule import TriggerRule
-from dev.tests_common.test_utils.watcher import watcher
from providers.tests.system.amazon.aws.utils import ENV_ID_KEY, SystemTestContextBuilder
# Externally fetched variables:
@@ -254,7 +255,7 @@ with DAG(
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_sagemaker.py b/providers/tests/system/amazon/aws/example_sagemaker.py
index acb05400b9..96e9756659 100644
--- a/providers/tests/system/amazon/aws/example_sagemaker.py
+++ b/providers/tests/system/amazon/aws/example_sagemaker.py
@@ -646,13 +646,13 @@ with DAG(
log_cleanup,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_sagemaker_endpoint.py b/providers/tests/system/amazon/aws/example_sagemaker_endpoint.py
index fecaa92264..165e5d5edd 100644
--- a/providers/tests/system/amazon/aws/example_sagemaker_endpoint.py
+++ b/providers/tests/system/amazon/aws/example_sagemaker_endpoint.py
@@ -291,13 +291,13 @@ with DAG(
archive_logs(f"/aws/sagemaker/Endpoints/{test_setup['endpoint_name']}"),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_sagemaker_notebook.py b/providers/tests/system/amazon/aws/example_sagemaker_notebook.py
index 76d35dc2b1..e989be7a92 100644
--- a/providers/tests/system/amazon/aws/example_sagemaker_notebook.py
+++ b/providers/tests/system/amazon/aws/example_sagemaker_notebook.py
@@ -97,13 +97,13 @@ with DAG(
delete_instance,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_sagemaker_pipeline.py b/providers/tests/system/amazon/aws/example_sagemaker_pipeline.py
index 614fb5c13c..9c0c9d7284 100644
--- a/providers/tests/system/amazon/aws/example_sagemaker_pipeline.py
+++ b/providers/tests/system/amazon/aws/example_sagemaker_pipeline.py
@@ -117,13 +117,13 @@ with DAG(
delete_experiments([pipeline_name]),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_salesforce_to_s3.py b/providers/tests/system/amazon/aws/example_salesforce_to_s3.py
index 7ee15f7030..a533c8c3fb 100644
--- a/providers/tests/system/amazon/aws/example_salesforce_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_salesforce_to_s3.py
@@ -78,13 +78,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_sftp_to_s3.py b/providers/tests/system/amazon/aws/example_sftp_to_s3.py
index 85dac988cb..765a68c994 100644
--- a/providers/tests/system/amazon/aws/example_sftp_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_sftp_to_s3.py
@@ -71,13 +71,13 @@ with DAG(
delete_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_sns.py b/providers/tests/system/amazon/aws/example_sns.py
index d00227536d..06756a244a 100644
--- a/providers/tests/system/amazon/aws/example_sns.py
+++ b/providers/tests/system/amazon/aws/example_sns.py
@@ -75,14 +75,14 @@ with DAG(
delete_topic(create_sns_topic),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_sql_to_s3.py b/providers/tests/system/amazon/aws/example_sql_to_s3.py
index dd333073a0..4cb8a2e7b9 100644
--- a/providers/tests/system/amazon/aws/example_sql_to_s3.py
+++ b/providers/tests/system/amazon/aws/example_sql_to_s3.py
@@ -199,14 +199,14 @@ with DAG(
delete_cluster,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_sqs.py b/providers/tests/system/amazon/aws/example_sqs.py
index 2aba6a580c..f14cf3b7a7 100644
--- a/providers/tests/system/amazon/aws/example_sqs.py
+++ b/providers/tests/system/amazon/aws/example_sqs.py
@@ -100,14 +100,14 @@ with DAG(
delete_queue(sqs_queue),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/example_step_functions.py b/providers/tests/system/amazon/aws/example_step_functions.py
index beeb12528c..eb3a399c69 100644
--- a/providers/tests/system/amazon/aws/example_step_functions.py
+++ b/providers/tests/system/amazon/aws/example_step_functions.py
@@ -112,13 +112,13 @@ with DAG(
delete_state_machine(state_machine_arn),
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/amazon/aws/tests/test_aws_auth_manager.py b/providers/tests/system/amazon/aws/tests/test_aws_auth_manager.py
index 774aec21b2..2656474cbb 100644
--- a/providers/tests/system/amazon/aws/tests/test_aws_auth_manager.py
+++ b/providers/tests/system/amazon/aws/tests/test_aws_auth_manager.py
@@ -21,11 +21,11 @@ from unittest.mock import Mock, patch
import boto3
import pytest
+from tests_common.test_utils.config import conf_vars
+from tests_common.test_utils.www import check_content_in_response
from airflow.www import app as application
-from dev.tests_common.test_utils.config import conf_vars
-from dev.tests_common.test_utils.www import check_content_in_response
from providers.tests.system.amazon.aws.utils import set_env_id
SAML_METADATA_URL = "/saml/metadata"
diff --git a/providers/tests/system/apache/beam/example_beam.py b/providers/tests/system/apache/beam/example_beam.py
index 4166caa347..b90971149b 100644
--- a/providers/tests/system/apache/beam/example_beam.py
+++ b/providers/tests/system/apache/beam/example_beam.py
@@ -61,7 +61,7 @@ with models.DAG(
# [END howto_operator_start_java_direct_runner_pipeline]
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/beam/example_beam_java_flink.py b/providers/tests/system/apache/beam/example_beam_java_flink.py
index a68637d8ef..8f639b0829 100644
--- a/providers/tests/system/apache/beam/example_beam_java_flink.py
+++ b/providers/tests/system/apache/beam/example_beam_java_flink.py
@@ -60,7 +60,7 @@ with models.DAG(
jar_to_local_flink_runner >> start_java_pipeline_flink_runner
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/beam/example_beam_java_spark.py b/providers/tests/system/apache/beam/example_beam_java_spark.py
index eb3fea6d16..b45ce5bbba 100644
--- a/providers/tests/system/apache/beam/example_beam_java_spark.py
+++ b/providers/tests/system/apache/beam/example_beam_java_spark.py
@@ -60,7 +60,7 @@ with models.DAG(
jar_to_local_spark_runner >> start_java_pipeline_spark_runner
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/beam/example_go.py b/providers/tests/system/apache/beam/example_go.py
index 792eee088b..ecdd8a20ed 100644
--- a/providers/tests/system/apache/beam/example_go.py
+++ b/providers/tests/system/apache/beam/example_go.py
@@ -103,7 +103,7 @@ with models.DAG(
)
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/beam/example_go_dataflow.py b/providers/tests/system/apache/beam/example_go_dataflow.py
index 35a63cb56e..e1904ea130 100644
--- a/providers/tests/system/apache/beam/example_go_dataflow.py
+++ b/providers/tests/system/apache/beam/example_go_dataflow.py
@@ -76,7 +76,7 @@ with models.DAG(
# [END howto_operator_start_go_dataflow_runner_pipeline_async_gcs_file]
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/beam/example_java_dataflow.py b/providers/tests/system/apache/beam/example_java_dataflow.py
index 334142dfd3..d01d52fe65 100644
--- a/providers/tests/system/apache/beam/example_java_dataflow.py
+++ b/providers/tests/system/apache/beam/example_java_dataflow.py
@@ -66,7 +66,7 @@ with models.DAG(
# [END howto_operator_start_java_dataflow_runner_pipeline]
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/beam/example_python.py b/providers/tests/system/apache/beam/example_python.py
index d685cb33d2..4b4b9c2c05 100644
--- a/providers/tests/system/apache/beam/example_python.py
+++ b/providers/tests/system/apache/beam/example_python.py
@@ -120,7 +120,7 @@ with models.DAG(
)
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/beam/example_python_async.py b/providers/tests/system/apache/beam/example_python_async.py
index 8465278b6b..0419a13cce 100644
--- a/providers/tests/system/apache/beam/example_python_async.py
+++ b/providers/tests/system/apache/beam/example_python_async.py
@@ -129,7 +129,7 @@ with models.DAG(
)
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/beam/example_python_dataflow.py b/providers/tests/system/apache/beam/example_python_dataflow.py
index 36d6b3b856..8d1295ed29 100644
--- a/providers/tests/system/apache/beam/example_python_dataflow.py
+++ b/providers/tests/system/apache/beam/example_python_dataflow.py
@@ -79,7 +79,7 @@ with models.DAG(
# [END howto_operator_start_python_dataflow_runner_pipeline_async_gcs_file]
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/cassandra/example_cassandra_dag.py b/providers/tests/system/apache/cassandra/example_cassandra_dag.py
index 0b25fa2975..5c1fe9b527 100644
--- a/providers/tests/system/apache/cassandra/example_cassandra_dag.py
+++ b/providers/tests/system/apache/cassandra/example_cassandra_dag.py
@@ -50,7 +50,7 @@ with DAG(
record_sensor = CassandraRecordSensor(task_id="cassandra_record_sensor", keys={"p1": "v1", "p2": "v2"})
# [END howto_operator_cassandra_sensors]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/drill/example_drill_dag.py b/providers/tests/system/apache/drill/example_drill_dag.py
index fdaf376d5a..7386f3b59c 100644
--- a/providers/tests/system/apache/drill/example_drill_dag.py
+++ b/providers/tests/system/apache/drill/example_drill_dag.py
@@ -47,7 +47,7 @@ with DAG(
)
# [END howto_operator_drill]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/druid/example_druid_dag.py b/providers/tests/system/apache/druid/example_druid_dag.py
index 5e1bfc446b..eac9af6d19 100644
--- a/providers/tests/system/apache/druid/example_druid_dag.py
+++ b/providers/tests/system/apache/druid/example_druid_dag.py
@@ -55,7 +55,7 @@ with DAG(
"""
# [END howto_operator_druid_submit]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/hive/example_twitter_dag.py b/providers/tests/system/apache/hive/example_twitter_dag.py
index 129a8ebf9f..c247bf9033 100644
--- a/providers/tests/system/apache/hive/example_twitter_dag.py
+++ b/providers/tests/system/apache/hive/example_twitter_dag.py
@@ -154,13 +154,13 @@ with DAG(
analyze >> load_to_hdfs >> load_to_hive >> hive_to_mysql
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/iceberg/example_iceberg.py b/providers/tests/system/apache/iceberg/example_iceberg.py
index a4708805b6..3c9f59f1ea 100644
--- a/providers/tests/system/apache/iceberg/example_iceberg.py
+++ b/providers/tests/system/apache/iceberg/example_iceberg.py
@@ -49,7 +49,7 @@ with DAG(
)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/kafka/example_dag_event_listener.py b/providers/tests/system/apache/kafka/example_dag_event_listener.py
index eb6866735e..734cb9c495 100644
--- a/providers/tests/system/apache/kafka/example_dag_event_listener.py
+++ b/providers/tests/system/apache/kafka/example_dag_event_listener.py
@@ -120,7 +120,7 @@ with DAG(
t0 >> t1
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/kafka/example_dag_hello_kafka.py b/providers/tests/system/apache/kafka/example_dag_hello_kafka.py
index 5e70d7324a..88fef458ea 100644
--- a/providers/tests/system/apache/kafka/example_dag_hello_kafka.py
+++ b/providers/tests/system/apache/kafka/example_dag_hello_kafka.py
@@ -240,7 +240,7 @@ with DAG(
t0 >> t3 >> [t4, t4b] >> t5 >> t6
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/kylin/example_kylin_dag.py b/providers/tests/system/apache/kylin/example_kylin_dag.py
index 1bfec2cd1a..e4de4105e0 100644
--- a/providers/tests/system/apache/kylin/example_kylin_dag.py
+++ b/providers/tests/system/apache/kylin/example_kylin_dag.py
@@ -109,13 +109,13 @@ with DAG(
# gen_build_time >> refresh_task1
# gen_build_time >> merge_task
# gen_build_time >> build_task3
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/livy/example_livy.py b/providers/tests/system/apache/livy/example_livy.py
index 09f8f7e244..8f3ea04a00 100644
--- a/providers/tests/system/apache/livy/example_livy.py
+++ b/providers/tests/system/apache/livy/example_livy.py
@@ -73,13 +73,13 @@ with DAG(
livy_java_task_deferrable >> livy_python_task_deferrable
# [END create_livy_deferrable]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/pig/example_pig.py b/providers/tests/system/apache/pig/example_pig.py
index bb556a8511..4d47bf5e65 100644
--- a/providers/tests/system/apache/pig/example_pig.py
+++ b/providers/tests/system/apache/pig/example_pig.py
@@ -44,7 +44,7 @@ with DAG(
# [END create_pig]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/pinot/example_pinot_dag.py b/providers/tests/system/apache/pinot/example_pinot_dag.py
index d95022f366..825c5070e2 100644
--- a/providers/tests/system/apache/pinot/example_pinot_dag.py
+++ b/providers/tests/system/apache/pinot/example_pinot_dag.py
@@ -52,7 +52,7 @@ with DAG(
pinot_admin()
pinot_dbi_api()
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/spark/example_pyspark.py b/providers/tests/system/apache/spark/example_pyspark.py
index 0ca14a76f4..d2d8ef6983 100644
--- a/providers/tests/system/apache/spark/example_pyspark.py
+++ b/providers/tests/system/apache/spark/example_pyspark.py
@@ -70,7 +70,7 @@ def example_pyspark():
dag = example_pyspark() # type: ignore
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/apache/spark/example_spark_dag.py b/providers/tests/system/apache/spark/example_spark_dag.py
index b68fc2cf76..6680450ae9 100644
--- a/providers/tests/system/apache/spark/example_spark_dag.py
+++ b/providers/tests/system/apache/spark/example_spark_dag.py
@@ -75,7 +75,7 @@ with DAG(
)
# [END howto_operator_spark_sql]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/asana/example_asana.py b/providers/tests/system/asana/example_asana.py
index ca68b6cc3d..a7f4943bed 100644
--- a/providers/tests/system/asana/example_asana.py
+++ b/providers/tests/system/asana/example_asana.py
@@ -100,13 +100,13 @@ with DAG(
create >> find >> update >> delete
# [END asana_example_dag]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/cncf/kubernetes/example_kubernetes.py b/providers/tests/system/cncf/kubernetes/example_kubernetes.py
index bed3b9da74..dc74895a3e 100644
--- a/providers/tests/system/cncf/kubernetes/example_kubernetes.py
+++ b/providers/tests/system/cncf/kubernetes/example_kubernetes.py
@@ -167,13 +167,13 @@ with DAG(
write_xcom >> pod_task_xcom_result
# [END howto_operator_k8s_write_xcom]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/cncf/kubernetes/example_kubernetes_async.py b/providers/tests/system/cncf/kubernetes/example_kubernetes_async.py
index 43eb0a1f91..7eb08442be 100644
--- a/providers/tests/system/cncf/kubernetes/example_kubernetes_async.py
+++ b/providers/tests/system/cncf/kubernetes/example_kubernetes_async.py
@@ -198,13 +198,13 @@ with DAG(
write_xcom_async >> pod_task_xcom_result_async
# [END howto_operator_k8s_write_xcom_async]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/cncf/kubernetes/example_kubernetes_decorator.py b/providers/tests/system/cncf/kubernetes/example_kubernetes_decorator.py
index 088a3fdae2..6784b22f3a 100644
--- a/providers/tests/system/cncf/kubernetes/example_kubernetes_decorator.py
+++ b/providers/tests/system/cncf/kubernetes/example_kubernetes_decorator.py
@@ -63,7 +63,7 @@ with DAG(
# [END howto_operator_kubernetes]
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/cncf/kubernetes/example_kubernetes_job.py b/providers/tests/system/cncf/kubernetes/example_kubernetes_job.py
index 1a2432c135..c81ff48e79 100644
--- a/providers/tests/system/cncf/kubernetes/example_kubernetes_job.py
+++ b/providers/tests/system/cncf/kubernetes/example_kubernetes_job.py
@@ -94,13 +94,13 @@ with DAG(
k8s_job >> update_job >> delete_job_task
k8s_job_def >> delete_job_task_def
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/cncf/kubernetes/example_kubernetes_resource.py b/providers/tests/system/cncf/kubernetes/example_kubernetes_resource.py
index 1c24a533a8..e7f6a4d42d 100644
--- a/providers/tests/system/cncf/kubernetes/example_kubernetes_resource.py
+++ b/providers/tests/system/cncf/kubernetes/example_kubernetes_resource.py
@@ -72,13 +72,13 @@ with DAG(
t1 >> t2
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/cncf/kubernetes/example_spark_kubernetes.py b/providers/tests/system/cncf/kubernetes/example_spark_kubernetes.py
index 072cffeb90..b244a9a800 100644
--- a/providers/tests/system/cncf/kubernetes/example_spark_kubernetes.py
+++ b/providers/tests/system/cncf/kubernetes/example_spark_kubernetes.py
@@ -76,13 +76,13 @@ with DAG(
t1 >> t2
# [END SparkKubernetesOperator_DAG]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/cohere/example_cohere_embedding_operator.py b/providers/tests/system/cohere/example_cohere_embedding_operator.py
index 9686724f2c..d8907260d3 100644
--- a/providers/tests/system/cohere/example_cohere_embedding_operator.py
+++ b/providers/tests/system/cohere/example_cohere_embedding_operator.py
@@ -34,7 +34,7 @@ with DAG("example_cohere_embedding", schedule=None, start_date=datetime(2023, 1,
# [END howto_operator_cohere_embedding]
-from dev.tests_common.test_utils.system_tests import get_test_run
+from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/common/io/example_file_transfer_local_to_s3.py b/providers/tests/system/common/io/example_file_transfer_local_to_s3.py
index 0b90a75f2e..f1522d5f26 100644
--- a/providers/tests/system/common/io/example_file_transfer_local_to_s3.py
+++ b/providers/tests/system/common/io/example_file_transfer_local_to_s3.py
@@ -76,12 +76,12 @@ with DAG(
temp_file >> transfer >> remove_bucket() >> delete_temp_file(temp_file_path)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/common/sql/example_sql_column_table_check.py b/providers/tests/system/common/sql/example_sql_column_table_check.py
index c48dc60bfc..4033c12faf 100644
--- a/providers/tests/system/common/sql/example_sql_column_table_check.py
+++ b/providers/tests/system/common/sql/example_sql_column_table_check.py
@@ -79,7 +79,7 @@ with DAG(
column_check >> row_count_check
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/common/sql/example_sql_execute_query.py b/providers/tests/system/common/sql/example_sql_execute_query.py
index 8dced19cef..5357980463 100644
--- a/providers/tests/system/common/sql/example_sql_execute_query.py
+++ b/providers/tests/system/common/sql/example_sql_execute_query.py
@@ -56,7 +56,7 @@ with DAG(
# [END howto_operator_sql_execute_query]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/databricks/example_databricks.py b/providers/tests/system/databricks/example_databricks.py
index 00d9969bd9..999cebb674 100644
--- a/providers/tests/system/databricks/example_databricks.py
+++ b/providers/tests/system/databricks/example_databricks.py
@@ -238,13 +238,13 @@ with DAG(
)
# [END howto_operator_databricks_task_sql]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/databricks/example_databricks_repos.py b/providers/tests/system/databricks/example_databricks_repos.py
index d07226e76e..33b0d1266f 100644
--- a/providers/tests/system/databricks/example_databricks_repos.py
+++ b/providers/tests/system/databricks/example_databricks_repos.py
@@ -78,13 +78,13 @@ with DAG(
(create_repo >> update_repo >> notebook_task >> delete_repo)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/databricks/example_databricks_sensors.py b/providers/tests/system/databricks/example_databricks_sensors.py
index ea233a7939..cf676183e6 100644
--- a/providers/tests/system/databricks/example_databricks_sensors.py
+++ b/providers/tests/system/databricks/example_databricks_sensors.py
@@ -88,7 +88,7 @@ with DAG(
# runs, else all the subsequent DAG tasks and the DAG are marked as failed.
(sql_sensor >> partition_sensor)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This example does not need a watcher in order to properly mark success/failure
# since it is a single task, but it is given here as an example for users to
@@ -96,7 +96,7 @@ with DAG(
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/databricks/example_databricks_sql.py b/providers/tests/system/databricks/example_databricks_sql.py
index f08ce3cd56..79c2bfa2fc 100644
--- a/providers/tests/system/databricks/example_databricks_sql.py
+++ b/providers/tests/system/databricks/example_databricks_sql.py
@@ -113,13 +113,13 @@ with DAG(
(create >> create_file >> import_csv >> select >> select_into_file)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/databricks/example_databricks_workflow.py b/providers/tests/system/databricks/example_databricks_workflow.py
index 3a6c829520..5921f109d0 100644
--- a/providers/tests/system/databricks/example_databricks_workflow.py
+++ b/providers/tests/system/databricks/example_databricks_workflow.py
@@ -143,13 +143,13 @@ with dag:
notebook_1 >> notebook_2 >> task_operator_nb_1 >> sql_query
# [END howto_databricks_workflow_notebook]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/dbt/cloud/example_dbt_cloud.py b/providers/tests/system/dbt/cloud/example_dbt_cloud.py
index 17f8a59efe..7a9a58a87a 100644
--- a/providers/tests/system/dbt/cloud/example_dbt_cloud.py
+++ b/providers/tests/system/dbt/cloud/example_dbt_cloud.py
@@ -18,6 +18,8 @@ from __future__ import annotations
from datetime import datetime
+from tests_common.test_utils.system_tests import get_test_env_id
+
from airflow.models import DAG
from airflow.operators.empty import EmptyOperator
from airflow.providers.dbt.cloud.operators.dbt import (
@@ -28,8 +30,6 @@ from airflow.providers.dbt.cloud.operators.dbt import (
from airflow.providers.dbt.cloud.sensors.dbt import DbtCloudJobRunSensor
from airflow.utils.edgemodifier import Label
-from dev.tests_common.test_utils.system_tests import get_test_env_id
-
ENV_ID = get_test_env_id()
DAG_ID = "example_dbt_cloud"
@@ -92,13 +92,13 @@ with DAG(
# trigger_job_run2 >> job_run_sensor
# trigger_job_run2 >> job_run_sensor_deferred
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/dingding/example_dingding.py b/providers/tests/system/dingding/example_dingding.py
index 4ecb41c68c..536f28fcc0 100644
--- a/providers/tests/system/dingding/example_dingding.py
+++ b/providers/tests/system/dingding/example_dingding.py
@@ -199,13 +199,13 @@ with DAG(
>> msg_failure_callback
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/docker/example_docker.py b/providers/tests/system/docker/example_docker.py
index 108813236a..14fa3fa320 100644
--- a/providers/tests/system/docker/example_docker.py
+++ b/providers/tests/system/docker/example_docker.py
@@ -57,7 +57,7 @@ with models.DAG(
t1 >> [t2, t3] >> t4
)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/docker/example_docker_copy_data.py b/providers/tests/system/docker/example_docker_copy_data.py
index c270408390..584b16b8e0 100644
--- a/providers/tests/system/docker/example_docker_copy_data.py
+++ b/providers/tests/system/docker/example_docker_copy_data.py
@@ -103,7 +103,7 @@ with models.DAG(
t_is_data_available >> t_move >> t_print
)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/docker/example_docker_swarm.py b/providers/tests/system/docker/example_docker_swarm.py
index 3a68dcea81..a45d1bba35 100644
--- a/providers/tests/system/docker/example_docker_swarm.py
+++ b/providers/tests/system/docker/example_docker_swarm.py
@@ -47,7 +47,7 @@ with models.DAG(
t1
)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/docker/example_taskflow_api_docker_virtualenv.py b/providers/tests/system/docker/example_taskflow_api_docker_virtualenv.py
index 3c8b528606..b026d9e950 100644
--- a/providers/tests/system/docker/example_taskflow_api_docker_virtualenv.py
+++ b/providers/tests/system/docker/example_taskflow_api_docker_virtualenv.py
@@ -116,9 +116,15 @@ with models.DAG(
tutorial_dag = tutorial_taskflow_api_docker_virtualenv()
# [END dag_invocation]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
# [END tutorial]
+
+
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
+
+# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
+test_run = get_test_run(dag)
diff --git a/providers/tests/system/elasticsearch/example_elasticsearch_query.py b/providers/tests/system/elasticsearch/example_elasticsearch_query.py
index 93471d874c..f4450b8e33 100644
--- a/providers/tests/system/elasticsearch/example_elasticsearch_query.py
+++ b/providers/tests/system/elasticsearch/example_elasticsearch_query.py
@@ -81,7 +81,7 @@ with models.DAG(
task_id="print_data_from_elasticsearch", python_callable=use_elasticsearch_hook
)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/ftp/example_ftp.py b/providers/tests/system/ftp/example_ftp.py
index 8fb2e38806..5b9bfe65db 100644
--- a/providers/tests/system/ftp/example_ftp.py
+++ b/providers/tests/system/ftp/example_ftp.py
@@ -87,13 +87,13 @@ with DAG(
ftp_put >> ftp_get
ftps_put >> ftps_get
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/github/example_github.py b/providers/tests/system/github/example_github.py
index 70eb8bf902..b076647177 100644
--- a/providers/tests/system/github/example_github.py
+++ b/providers/tests/system/github/example_github.py
@@ -100,7 +100,7 @@ with DAG(
# [END howto_operator_list_tags_github]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/ads/example_ads.py b/providers/tests/system/google/ads/example_ads.py
index 162fd45334..f55c7306f3 100644
--- a/providers/tests/system/google/ads/example_ads.py
+++ b/providers/tests/system/google/ads/example_ads.py
@@ -117,14 +117,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/automl/example_automl_dataset.py b/providers/tests/system/google/cloud/automl/example_automl_dataset.py
index 49de111ffb..d1305da4fb 100644
--- a/providers/tests/system/google/cloud/automl/example_automl_dataset.py
+++ b/providers/tests/system/google/cloud/automl/example_automl_dataset.py
@@ -162,14 +162,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/automl/example_automl_translation.py b/providers/tests/system/google/cloud/automl/example_automl_translation.py
index 60dbf782f2..e758bdf113 100644
--- a/providers/tests/system/google/cloud/automl/example_automl_translation.py
+++ b/providers/tests/system/google/cloud/automl/example_automl_translation.py
@@ -188,14 +188,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/automl/example_automl_video_classification.py b/providers/tests/system/google/cloud/automl/example_automl_video_classification.py
index 36853c3826..538831d307 100644
--- a/providers/tests/system/google/cloud/automl/example_automl_video_classification.py
+++ b/providers/tests/system/google/cloud/automl/example_automl_video_classification.py
@@ -158,13 +158,13 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/automl/example_automl_video_tracking.py b/providers/tests/system/google/cloud/automl/example_automl_video_tracking.py
index 65718f10b7..3876e9b0a3 100644
--- a/providers/tests/system/google/cloud/automl/example_automl_video_tracking.py
+++ b/providers/tests/system/google/cloud/automl/example_automl_video_tracking.py
@@ -158,13 +158,13 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/automl/example_automl_vision_classification.py b/providers/tests/system/google/cloud/automl/example_automl_vision_classification.py
index 40e68d3291..e329febf5d 100644
--- a/providers/tests/system/google/cloud/automl/example_automl_vision_classification.py
+++ b/providers/tests/system/google/cloud/automl/example_automl_vision_classification.py
@@ -132,13 +132,13 @@ with DAG(
>> delete_image_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/automl/example_automl_vision_object_detection.py b/providers/tests/system/google/cloud/automl/example_automl_vision_object_detection.py
index 334e154cea..b87b42008d 100644
--- a/providers/tests/system/google/cloud/automl/example_automl_vision_object_detection.py
+++ b/providers/tests/system/google/cloud/automl/example_automl_vision_object_detection.py
@@ -134,13 +134,13 @@ with DAG(
>> delete_image_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/azure/example_azure_blob_to_gcs.py b/providers/tests/system/google/cloud/azure/example_azure_blob_to_gcs.py
index 837cb99e5f..e915642b70 100644
--- a/providers/tests/system/google/cloud/azure/example_azure_blob_to_gcs.py
+++ b/providers/tests/system/google/cloud/azure/example_azure_blob_to_gcs.py
@@ -59,13 +59,13 @@ with DAG(
(wait_for_blob >> transfer_files_to_gcs)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/azure/example_azure_fileshare_to_gcs.py b/providers/tests/system/google/cloud/azure/example_azure_fileshare_to_gcs.py
index a897b6dc7d..45172fa207 100644
--- a/providers/tests/system/google/cloud/azure/example_azure_fileshare_to_gcs.py
+++ b/providers/tests/system/google/cloud/azure/example_azure_fileshare_to_gcs.py
@@ -81,14 +81,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_dataset.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_dataset.py
index 87e13a2c25..7be37b8631 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_dataset.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_dataset.py
@@ -85,14 +85,14 @@ with DAG(
>> delete_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_dts.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_dts.py
index b387563069..37677a1514 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_dts.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_dts.py
@@ -180,14 +180,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_operations.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_operations.py
index 5e487f5a93..308c433768 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_operations.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_operations.py
@@ -97,14 +97,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_operations_location.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_operations_location.py
index 2b026070a0..d7547e2b81 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_operations_location.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_operations_location.py
@@ -77,14 +77,14 @@ with DAG(
>> delete_dataset_with_location
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_queries.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_queries.py
index cd31ea05bc..0e4d930442 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_queries.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_queries.py
@@ -247,7 +247,7 @@ for index, location in enumerate(locations, 1):
execute_insert_query >> [check_count, check_value, check_interval] >> delete_dataset
execute_insert_query >> [column_check, table_check] >> delete_dataset
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
@@ -257,7 +257,7 @@ for index, location in enumerate(locations, 1):
globals()[DAG_ID] = dag
for dag in DAGS_LIST:
- from dev.tests_common.test_utils.system_tests import get_test_run
+ from tests_common.test_utils.system_tests import get_test_run
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_queries_async.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_queries_async.py
index 012eba080d..104f09ea33 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_queries_async.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_queries_async.py
@@ -259,14 +259,14 @@ with DAG(
insert_query_job >> execute_long_running_query >> check_value >> check_interval
[check_count, check_interval, bigquery_execute_multi_query, get_data_result] >> delete_dataset
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_sensors.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_sensors.py
index aa8d36f5ed..760275f904 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_sensors.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_sensors.py
@@ -165,14 +165,14 @@ with DAG(
>> delete_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_tables.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_tables.py
index 9c1432a0a8..588e33d368 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_tables.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_tables.py
@@ -224,14 +224,14 @@ with DAG(
>> delete_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_bigquery.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_bigquery.py
index cb1fa63498..879d8aa3fb 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_bigquery.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_bigquery.py
@@ -100,14 +100,14 @@ with DAG(
>> delete_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_gcs.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_gcs.py
index 21acb27f16..76684a1c05 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_gcs.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_gcs.py
@@ -98,14 +98,14 @@ with DAG(
>> [delete_bucket, delete_dataset]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_gcs_async.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_gcs_async.py
index fdf084601c..2bc4846fea 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_gcs_async.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_gcs_async.py
@@ -97,14 +97,14 @@ with DAG(
>> [delete_bucket, delete_dataset]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_mssql.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_mssql.py
index 51e713560b..3fd5e92595 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_mssql.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_mssql.py
@@ -326,13 +326,13 @@ with DAG(
>> delete_persistent_disk
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_mysql.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_mysql.py
index a3b7260987..623924c2c9 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_mysql.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_mysql.py
@@ -88,13 +88,13 @@ with DAG(
>> delete_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_postgres.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_postgres.py
index ca4e3148d0..54f5efc13d 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_to_postgres.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_to_postgres.py
@@ -362,13 +362,13 @@ with DAG(
>> delete_persistent_disk
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_transfer.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_transfer.py
index 73d55710c9..39f1eea2bd 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_transfer.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_transfer.py
@@ -117,13 +117,13 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigquery/example_bigquery_value_check.py b/providers/tests/system/google/cloud/bigquery/example_bigquery_value_check.py
index 3872ca0923..5a51ad3101 100644
--- a/providers/tests/system/google/cloud/bigquery/example_bigquery_value_check.py
+++ b/providers/tests/system/google/cloud/bigquery/example_bigquery_value_check.py
@@ -131,8 +131,8 @@ with DAG(
>> delete_dataset
)
- from dev.tests_common.test_utils.system_tests import get_test_run
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.system_tests import get_test_run
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
@@ -142,7 +142,7 @@ with DAG(
test_run = get_test_run(dag)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/bigtable/example_bigtable.py b/providers/tests/system/google/cloud/bigtable/example_bigtable.py
index 77abc49fae..b1f3469f4f 100644
--- a/providers/tests/system/google/cloud/bigtable/example_bigtable.py
+++ b/providers/tests/system/google/cloud/bigtable/example_bigtable.py
@@ -225,14 +225,14 @@ with DAG(
>> [delete_instance_task, delete_instance_task2]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_batch/example_cloud_batch.py b/providers/tests/system/google/cloud/cloud_batch/example_cloud_batch.py
index 84dc031a52..322c6cd7ec 100644
--- a/providers/tests/system/google/cloud/cloud_batch/example_cloud_batch.py
+++ b/providers/tests/system/google/cloud/cloud_batch/example_cloud_batch.py
@@ -187,13 +187,13 @@ with DAG(
([submit1, submit2] >> list_tasks >> assert_tasks >> list_jobs >> get_name >> [delete_job1, delete_job2])
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_build/example_cloud_build.py b/providers/tests/system/google/cloud/cloud_build/example_cloud_build.py
index bf2391413a..319047fbaa 100644
--- a/providers/tests/system/google/cloud/cloud_build/example_cloud_build.py
+++ b/providers/tests/system/google/cloud/cloud_build/example_cloud_build.py
@@ -267,14 +267,14 @@ with DAG(
]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_build/example_cloud_build_trigger.py b/providers/tests/system/google/cloud/cloud_build/example_cloud_build_trigger.py
index f6873da9c5..e799af6650 100644
--- a/providers/tests/system/google/cloud/cloud_build/example_cloud_build_trigger.py
+++ b/providers/tests/system/google/cloud/cloud_build/example_cloud_build_trigger.py
@@ -182,14 +182,14 @@ with DAG(
>> list_build_triggers
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_functions/example_functions.py b/providers/tests/system/google/cloud/cloud_functions/example_functions.py
index fd0b8ea5f5..6752249804 100644
--- a/providers/tests/system/google/cloud/cloud_functions/example_functions.py
+++ b/providers/tests/system/google/cloud/cloud_functions/example_functions.py
@@ -122,14 +122,14 @@ with DAG(
delete_function,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_memorystore/example_cloud_memorystore_memcached.py b/providers/tests/system/google/cloud/cloud_memorystore/example_cloud_memorystore_memcached.py
index 670a850b04..5d425b8b0e 100644
--- a/providers/tests/system/google/cloud/cloud_memorystore/example_cloud_memorystore_memcached.py
+++ b/providers/tests/system/google/cloud/cloud_memorystore/example_cloud_memorystore_memcached.py
@@ -198,14 +198,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_memorystore/example_cloud_memorystore_redis.py b/providers/tests/system/google/cloud/cloud_memorystore/example_cloud_memorystore_redis.py
index 3d8ebf6f28..cb5551f070 100644
--- a/providers/tests/system/google/cloud/cloud_memorystore/example_cloud_memorystore_redis.py
+++ b/providers/tests/system/google/cloud/cloud_memorystore/example_cloud_memorystore_redis.py
@@ -266,14 +266,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_run/example_cloud_run.py b/providers/tests/system/google/cloud/cloud_run/example_cloud_run.py
index 08145e0336..368d835ddb 100644
--- a/providers/tests/system/google/cloud/cloud_run/example_cloud_run.py
+++ b/providers/tests/system/google/cloud/cloud_run/example_cloud_run.py
@@ -367,13 +367,13 @@ with DAG(
>> (delete_job1, delete_job2, delete_job3)
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_run/example_cloud_run_service.py b/providers/tests/system/google/cloud/cloud_run/example_cloud_run_service.py
index 43318c4b14..8d87bcbc10 100644
--- a/providers/tests/system/google/cloud/cloud_run/example_cloud_run_service.py
+++ b/providers/tests/system/google/cloud/cloud_run/example_cloud_run_service.py
@@ -84,14 +84,14 @@ with DAG(
>> delete_cloud_run_service
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql.py b/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql.py
index 52414b2784..57da5f43c1 100644
--- a/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql.py
+++ b/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql.py
@@ -294,14 +294,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql_query.py b/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql_query.py
index 6cf7e0c087..3d97fad5aa 100644
--- a/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql_query.py
+++ b/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql_query.py
@@ -541,13 +541,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql_query_ssl.py b/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql_query_ssl.py
index db77f18339..69ec1784fa 100644
--- a/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql_query_ssl.py
+++ b/providers/tests/system/google/cloud/cloud_sql/example_cloud_sql_query_ssl.py
@@ -491,13 +491,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/composer/example_cloud_composer.py b/providers/tests/system/google/cloud/composer/example_cloud_composer.py
index 266a7e4a44..f04a2c8596 100644
--- a/providers/tests/system/google/cloud/composer/example_cloud_composer.py
+++ b/providers/tests/system/google/cloud/composer/example_cloud_composer.py
@@ -214,14 +214,14 @@ with DAG(
[delete_env, defer_delete_env],
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/compute/example_compute.py b/providers/tests/system/google/cloud/compute/example_compute.py
index 8343920890..82f2b008c1 100644
--- a/providers/tests/system/google/cloud/compute/example_compute.py
+++ b/providers/tests/system/google/cloud/compute/example_compute.py
@@ -267,14 +267,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/compute/example_compute_igm.py b/providers/tests/system/google/cloud/compute/example_compute_igm.py
index 11357452df..5f292958c2 100644
--- a/providers/tests/system/google/cloud/compute/example_compute_igm.py
+++ b/providers/tests/system/google/cloud/compute/example_compute_igm.py
@@ -236,14 +236,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/compute/example_compute_ssh.py b/providers/tests/system/google/cloud/compute/example_compute_ssh.py
index 5b7d152301..ffba97e91d 100644
--- a/providers/tests/system/google/cloud/compute/example_compute_ssh.py
+++ b/providers/tests/system/google/cloud/compute/example_compute_ssh.py
@@ -138,14 +138,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/compute/example_compute_ssh_os_login.py b/providers/tests/system/google/cloud/compute/example_compute_ssh_os_login.py
index 63ed278a08..89a9223e70 100644
--- a/providers/tests/system/google/cloud/compute/example_compute_ssh_os_login.py
+++ b/providers/tests/system/google/cloud/compute/example_compute_ssh_os_login.py
@@ -146,14 +146,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/compute/example_compute_ssh_parallel.py b/providers/tests/system/google/cloud/compute/example_compute_ssh_parallel.py
index 3ab61afc0c..e8b8767fff 100644
--- a/providers/tests/system/google/cloud/compute/example_compute_ssh_parallel.py
+++ b/providers/tests/system/google/cloud/compute/example_compute_ssh_parallel.py
@@ -139,14 +139,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_deidentify_content.py b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_deidentify_content.py
index 75b1ad93dd..78549f3735 100644
--- a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_deidentify_content.py
+++ b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_deidentify_content.py
@@ -157,14 +157,14 @@ with DAG(
>> delete_template
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_info_types.py b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_info_types.py
index fb2198b167..72e35d3234 100644
--- a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_info_types.py
+++ b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_info_types.py
@@ -154,14 +154,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_inspect_template.py b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_inspect_template.py
index 22b259582f..0d48b2fc19 100644
--- a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_inspect_template.py
+++ b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_inspect_template.py
@@ -115,14 +115,14 @@ with DAG(
>> delete_template
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_job.py b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_job.py
index b7dc2ac1a4..27db1730ad 100644
--- a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_job.py
+++ b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_job.py
@@ -90,14 +90,14 @@ with DAG(
(create_job >> list_jobs >> get_job >> cancel_job >> delete_job)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_job_trigger.py b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_job_trigger.py
index 902875f7b5..11fd48e1e4 100644
--- a/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_job_trigger.py
+++ b/providers/tests/system/google/cloud/data_loss_prevention/example_dlp_job_trigger.py
@@ -95,14 +95,14 @@ with DAG(
(create_trigger >> list_triggers >> get_trigger >> update_trigger >> delete_trigger)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_go.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_go.py
index 57e5941e3e..92e779f69d 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_go.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_go.py
@@ -149,7 +149,7 @@ with DAG(
)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_native_java.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_native_java.py
index 3629194e6e..66d0ce0995 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_native_java.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_native_java.py
@@ -148,14 +148,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_native_python.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_native_python.py
index 229373ed1b..70cd5db9e9 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_native_python.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_native_python.py
@@ -110,14 +110,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_native_python_async.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_native_python_async.py
index 31f1cd026f..9bca9b0287 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_native_python_async.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_native_python_async.py
@@ -176,14 +176,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_pipeline.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_pipeline.py
index cfd5e06b02..800540eb4d 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_pipeline.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_pipeline.py
@@ -137,14 +137,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_sensors_deferrable.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_sensors_deferrable.py
index 1a6d0aec39..c6303d7423 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_sensors_deferrable.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_sensors_deferrable.py
@@ -177,14 +177,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_sql.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_sql.py
index e3b0ee711a..2ba0bf0534 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_sql.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_sql.py
@@ -137,13 +137,13 @@ with DAG(
>> delete_bq_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_streaming_python.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_streaming_python.py
index a15dffb3a3..4e7a7ccebd 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_streaming_python.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_streaming_python.py
@@ -114,14 +114,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_template.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_template.py
index 8654551460..4b5bccacb0 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_template.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_template.py
@@ -158,14 +158,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataflow/example_dataflow_yaml.py b/providers/tests/system/google/cloud/dataflow/example_dataflow_yaml.py
index 2243ad695b..14db8b8936 100644
--- a/providers/tests/system/google/cloud/dataflow/example_dataflow_yaml.py
+++ b/providers/tests/system/google/cloud/dataflow/example_dataflow_yaml.py
@@ -162,13 +162,13 @@ with DAG(
>> delete_bq_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataform/example_dataform.py b/providers/tests/system/google/cloud/dataform/example_dataform.py
index b61247a877..e88a37caef 100644
--- a/providers/tests/system/google/cloud/dataform/example_dataform.py
+++ b/providers/tests/system/google/cloud/dataform/example_dataform.py
@@ -328,13 +328,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/datafusion/example_datafusion.py b/providers/tests/system/google/cloud/datafusion/example_datafusion.py
index a61fbbd01d..2181d2b2ba 100644
--- a/providers/tests/system/google/cloud/datafusion/example_datafusion.py
+++ b/providers/tests/system/google/cloud/datafusion/example_datafusion.py
@@ -340,13 +340,13 @@ with DAG(
>> [delete_bucket1, delete_bucket2]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/datapipelines/example_datapipeline.py b/providers/tests/system/google/cloud/datapipelines/example_datapipeline.py
index 91fb7eed26..0c0c430eae 100644
--- a/providers/tests/system/google/cloud/datapipelines/example_datapipeline.py
+++ b/providers/tests/system/google/cloud/datapipelines/example_datapipeline.py
@@ -138,14 +138,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataplex/example_dataplex.py b/providers/tests/system/google/cloud/dataplex/example_dataplex.py
index 34e9a0fd05..f5ce1bbd41 100644
--- a/providers/tests/system/google/cloud/dataplex/example_dataplex.py
+++ b/providers/tests/system/google/cloud/dataplex/example_dataplex.py
@@ -206,14 +206,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataplex/example_dataplex_dp.py b/providers/tests/system/google/cloud/dataplex/example_dataplex_dp.py
index 44f4af2435..da447a8ced 100644
--- a/providers/tests/system/google/cloud/dataplex/example_dataplex_dp.py
+++ b/providers/tests/system/google/cloud/dataplex/example_dataplex_dp.py
@@ -329,14 +329,14 @@ with DAG(
[delete_lake, delete_dataset],
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataplex/example_dataplex_dq.py b/providers/tests/system/google/cloud/dataplex/example_dataplex_dq.py
index d31d574755..1a1910757f 100644
--- a/providers/tests/system/google/cloud/dataplex/example_dataplex_dq.py
+++ b/providers/tests/system/google/cloud/dataplex/example_dataplex_dq.py
@@ -363,14 +363,14 @@ with DAG(
[delete_lake, delete_dataset],
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataprep/example_dataprep.py b/providers/tests/system/google/cloud/dataprep/example_dataprep.py
index cdc736a41c..edffb69e58 100644
--- a/providers/tests/system/google/cloud/dataprep/example_dataprep.py
+++ b/providers/tests/system/google/cloud/dataprep/example_dataprep.py
@@ -307,14 +307,14 @@ with models.DAG(
[delete_bucket_task, delete_connection_task],
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_batch.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_batch.py
index 54481f43ca..c88a1c3c68 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_batch.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_batch.py
@@ -176,13 +176,13 @@ with DAG(
>> delete_batch_4
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_batch_deferrable.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_batch_deferrable.py
index 79d3766995..17e5f99b52 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_batch_deferrable.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_batch_deferrable.py
@@ -91,14 +91,14 @@ with DAG(
>> delete_batch
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_batch_persistent.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_batch_persistent.py
index a8ed6b13de..5bf500aeef 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_batch_persistent.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_batch_persistent.py
@@ -141,14 +141,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_create_existing_stopped_cluster.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_create_existing_stopped_cluster.py
index 0eb27762c2..e64a091494 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_create_existing_stopped_cluster.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_create_existing_stopped_cluster.py
@@ -121,13 +121,13 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_deferrable.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_deferrable.py
index 8ae262dbf6..2f7153d440 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_deferrable.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_deferrable.py
@@ -137,14 +137,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_diagnose.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_diagnose.py
index 3eacc905bc..9e6c228a9d 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_diagnose.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_diagnose.py
@@ -115,14 +115,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_generator.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_generator.py
index f990363c48..0562be439b 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_generator.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_generator.py
@@ -135,14 +135,14 @@ with DAG(
>> [delete_cluster, delete_bucket]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_start_stop.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_start_stop.py
index 2e4b698573..c68bdd013b 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_start_stop.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_start_stop.py
@@ -111,13 +111,13 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_update.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_update.py
index 7ed0061d99..620019cd48 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_update.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_cluster_update.py
@@ -120,14 +120,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_flink.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_flink.py
index ce1a6fc445..34bcdb17e5 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_flink.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_flink.py
@@ -124,14 +124,14 @@ with DAG(
>> [delete_cluster, delete_bucket]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_gke.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_gke.py
index bb1044da62..8048a23283 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_gke.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_gke.py
@@ -144,13 +144,13 @@ with DAG(
>> delete_gke_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_hadoop.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_hadoop.py
index c1f5423041..e4370b2f79 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_hadoop.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_hadoop.py
@@ -123,14 +123,14 @@ with DAG(
>> [delete_cluster, delete_bucket]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_hive.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_hive.py
index 6aadceb552..bcddd46156 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_hive.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_hive.py
@@ -124,14 +124,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_pig.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_pig.py
index 762ca2bc73..848138ae54 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_pig.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_pig.py
@@ -108,14 +108,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_presto.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_presto.py
index f5bdf25732..ea8ccb231d 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_presto.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_presto.py
@@ -115,14 +115,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_pyspark.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_pyspark.py
index 71e7245b7d..fdc22d8cae 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_pyspark.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_pyspark.py
@@ -140,14 +140,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_spark.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_spark.py
index e43a56d230..5ff5ce8ed4 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_spark.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_spark.py
@@ -111,14 +111,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_async.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_async.py
index 475e9912e8..df3d969079 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_async.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_async.py
@@ -121,14 +121,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_deferrable.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_deferrable.py
index 5e6d0b773a..20b68b0574 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_deferrable.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_deferrable.py
@@ -112,14 +112,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_sql.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_sql.py
index aef860a0e5..3b34b8dbb2 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_sql.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_spark_sql.py
@@ -108,14 +108,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_sparkr.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_sparkr.py
index 7b416ade38..39890a501b 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_sparkr.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_sparkr.py
@@ -136,14 +136,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_trino.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_trino.py
index d3f7f2a1a3..3f366f6088 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_trino.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_trino.py
@@ -117,14 +117,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_workflow.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_workflow.py
index ab465a124d..eba4ca34a7 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_workflow.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_workflow.py
@@ -104,14 +104,14 @@ with DAG(
>> instantiate_inline_workflow_template
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc/example_dataproc_workflow_deferrable.py b/providers/tests/system/google/cloud/dataproc/example_dataproc_workflow_deferrable.py
index e2319a35a1..26dd27efcb 100644
--- a/providers/tests/system/google/cloud/dataproc/example_dataproc_workflow_deferrable.py
+++ b/providers/tests/system/google/cloud/dataproc/example_dataproc_workflow_deferrable.py
@@ -108,14 +108,14 @@ with DAG(
>> instantiate_inline_workflow_template_async
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore.py b/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore.py
index ee3b5b70ad..cbd612ba80 100644
--- a/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore.py
+++ b/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore.py
@@ -192,14 +192,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore_backup.py b/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore_backup.py
index 12af45d210..dddb69bfd0 100644
--- a/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore_backup.py
+++ b/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore_backup.py
@@ -126,14 +126,14 @@ with DAG(
)
(create_service >> backup_service >> list_backups >> restore_service >> delete_backup >> delete_service)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore_hive_partition_sensor.py b/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore_hive_partition_sensor.py
index 6b06b868c7..f3317b9ce7 100644
--- a/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore_hive_partition_sensor.py
+++ b/providers/tests/system/google/cloud/dataproc_metastore/example_dataproc_metastore_hive_partition_sensor.py
@@ -232,14 +232,14 @@ with DAG(
>> [delete_dataproc_cluster, delete_metastore_service, delete_warehouse_bucket]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/datastore/example_datastore_commit.py b/providers/tests/system/google/cloud/datastore/example_datastore_commit.py
index 430d176ed7..60ec53cf1f 100644
--- a/providers/tests/system/google/cloud/datastore/example_datastore_commit.py
+++ b/providers/tests/system/google/cloud/datastore/example_datastore_commit.py
@@ -160,14 +160,14 @@ with DAG(
[delete_bucket, delete_export_operation, delete_import_operation],
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/datastore/example_datastore_query.py b/providers/tests/system/google/cloud/datastore/example_datastore_query.py
index 0b282a1feb..f6908823a8 100644
--- a/providers/tests/system/google/cloud/datastore/example_datastore_query.py
+++ b/providers/tests/system/google/cloud/datastore/example_datastore_query.py
@@ -80,14 +80,14 @@ with DAG(
allocate_ids >> begin_transaction_query >> run_query
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/datastore/example_datastore_rollback.py b/providers/tests/system/google/cloud/datastore/example_datastore_rollback.py
index 44d0dc1709..0634eabeb5 100644
--- a/providers/tests/system/google/cloud/datastore/example_datastore_rollback.py
+++ b/providers/tests/system/google/cloud/datastore/example_datastore_rollback.py
@@ -63,14 +63,14 @@ with DAG(
begin_transaction_to_rollback >> rollback_transaction
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_calendar_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_calendar_to_gcs.py
index f44e7315e2..3b3cd02889 100644
--- a/providers/tests/system/google/cloud/gcs/example_calendar_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_calendar_to_gcs.py
@@ -115,13 +115,13 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_firestore.py b/providers/tests/system/google/cloud/gcs/example_firestore.py
index 89e0e3fe1c..1860d30c0d 100644
--- a/providers/tests/system/google/cloud/gcs/example_firestore.py
+++ b/providers/tests/system/google/cloud/gcs/example_firestore.py
@@ -170,14 +170,14 @@ with DAG(
>> [delete_dataset, delete_bucket]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_acl.py b/providers/tests/system/google/cloud/gcs/example_gcs_acl.py
index 7843c0877c..7000ac4d45 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_acl.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_acl.py
@@ -108,14 +108,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_copy_delete.py b/providers/tests/system/google/cloud/gcs/example_gcs_copy_delete.py
index 6bbec540df..62c31d80b2 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_copy_delete.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_copy_delete.py
@@ -122,14 +122,14 @@ with DAG(
[delete_bucket_src, delete_bucket_dst],
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_sensor.py b/providers/tests/system/google/cloud/gcs/example_gcs_sensor.py
index 2d4da2887d..a07aff849e 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_sensor.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_sensor.py
@@ -201,14 +201,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_to_bigquery.py b/providers/tests/system/google/cloud/gcs/example_gcs_to_bigquery.py
index 5d7e393bff..efea506c90 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_to_bigquery.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_to_bigquery.py
@@ -82,13 +82,13 @@ with DAG(
>> delete_test_dataset
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_to_bigquery_async.py b/providers/tests/system/google/cloud/gcs/example_gcs_to_bigquery_async.py
index 1ca531b818..7572f2bc55 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_to_bigquery_async.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_to_bigquery_async.py
@@ -174,13 +174,13 @@ with DAG(
>> delete_test_dataset_delimiter
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_gcs_to_gcs.py
index ac1d70307a..c4c812868e 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_to_gcs.py
@@ -275,14 +275,14 @@ with DAG(
[delete_bucket_src, delete_bucket_dst, delete_work_dir(create_workdir_task)],
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_to_gdrive.py b/providers/tests/system/google/cloud/gcs/example_gcs_to_gdrive.py
index bfb200a15a..c15e24ff3f 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_to_gdrive.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_to_gdrive.py
@@ -204,13 +204,13 @@ with DAG(
>> [delete_bucket, delete_connection_task]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_to_sheets.py b/providers/tests/system/google/cloud/gcs/example_gcs_to_sheets.py
index 69947698cd..991f8347f5 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_to_sheets.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_to_sheets.py
@@ -131,13 +131,13 @@ with DAG(
>> [delete_bucket, delete_connection_task]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_transform.py b/providers/tests/system/google/cloud/gcs/example_gcs_transform.py
index 0b59119c6b..421272f8b5 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_transform.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_transform.py
@@ -95,14 +95,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_transform_timespan.py b/providers/tests/system/google/cloud/gcs/example_gcs_transform_timespan.py
index 3a5b03695e..1c7e68b9d1 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_transform_timespan.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_transform_timespan.py
@@ -114,14 +114,14 @@ with DAG(
[delete_bucket_src, delete_bucket_dst],
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py b/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py
index 6907a29fb5..abde62d35b 100644
--- a/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py
+++ b/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py
@@ -92,14 +92,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_gdrive_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_gdrive_to_gcs.py
index 6ee188bd95..39633a8fc2 100644
--- a/providers/tests/system/google/cloud/gcs/example_gdrive_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_gdrive_to_gcs.py
@@ -165,13 +165,13 @@ with DAG(
>> [delete_bucket, delete_connection_task]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_mssql_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_mssql_to_gcs.py
index bc6ac79f1e..7aece3764c 100644
--- a/providers/tests/system/google/cloud/gcs/example_mssql_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_mssql_to_gcs.py
@@ -78,14 +78,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_mysql_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_mysql_to_gcs.py
index cc53337e56..ae3fea2c23 100644
--- a/providers/tests/system/google/cloud/gcs/example_mysql_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_mysql_to_gcs.py
@@ -295,13 +295,13 @@ with DAG(
mysql_to_gcs >> [delete_gcs_bucket, delete_firewall_rule, delete_gce_instance, delete_connection_task]
delete_gce_instance >> delete_persistent_disk
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_oracle_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_oracle_to_gcs.py
index c727eef1f6..ba4d7025ac 100644
--- a/providers/tests/system/google/cloud/gcs/example_oracle_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_oracle_to_gcs.py
@@ -64,13 +64,13 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_s3_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_s3_to_gcs.py
index 487dbbce55..59f88f78b2 100644
--- a/providers/tests/system/google/cloud/gcs/example_s3_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_s3_to_gcs.py
@@ -110,14 +110,14 @@ with DAG(
>> delete_gcs_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_sftp_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_sftp_to_gcs.py
index 994e7e8242..b2c2504265 100644
--- a/providers/tests/system/google/cloud/gcs/example_sftp_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_sftp_to_gcs.py
@@ -119,14 +119,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_sheets.py b/providers/tests/system/google/cloud/gcs/example_sheets.py
index 7d17379819..001ebb69d6 100644
--- a/providers/tests/system/google/cloud/gcs/example_sheets.py
+++ b/providers/tests/system/google/cloud/gcs/example_sheets.py
@@ -145,13 +145,13 @@ with DAG(
>> [delete_bucket, delete_connection_task]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_sheets_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_sheets_to_gcs.py
index 08688c3325..7c54606033 100644
--- a/providers/tests/system/google/cloud/gcs/example_sheets_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_sheets_to_gcs.py
@@ -120,13 +120,13 @@ with DAG(
>> [delete_bucket, delete_connection_task]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/gcs/example_trino_to_gcs.py b/providers/tests/system/google/cloud/gcs/example_trino_to_gcs.py
index af92e55048..e76a66ea22 100644
--- a/providers/tests/system/google/cloud/gcs/example_trino_to_gcs.py
+++ b/providers/tests/system/google/cloud/gcs/example_trino_to_gcs.py
@@ -221,13 +221,13 @@ with DAG(
>> [delete_dataset, delete_bucket]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine.py b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine.py
index e9fe3f6836..36d21ff701 100644
--- a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine.py
+++ b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine.py
@@ -113,14 +113,14 @@ with DAG(
create_cluster >> [pod_task, pod_task_xcom] >> delete_cluster
pod_task_xcom >> pod_task_xcom_result
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_async.py b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_async.py
index f5cb8f5707..9283f77dbd 100644
--- a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_async.py
+++ b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_async.py
@@ -116,14 +116,14 @@ with DAG(
create_cluster >> [pod_task, pod_task_xcom_async] >> delete_cluster
pod_task_xcom_async >> pod_task_xcom_result
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_job.py b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_job.py
index a4c56c17e5..9ff3ef32a7 100644
--- a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_job.py
+++ b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_job.py
@@ -179,14 +179,14 @@ with DAG(
delete_cluster,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_kueue.py b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_kueue.py
index 06c23432a9..44dbaae553 100644
--- a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_kueue.py
+++ b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_kueue.py
@@ -179,14 +179,14 @@ with DAG(
>> delete_cluster
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_resource.py b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_resource.py
index 784ba99486..28a6ac985b 100644
--- a/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_resource.py
+++ b/providers/tests/system/google/cloud/kubernetes_engine/example_kubernetes_engine_resource.py
@@ -101,14 +101,14 @@ with DAG(
create_cluster >> create_resource_task >> delete_resource_task >> delete_cluster
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/life_sciences/example_life_sciences.py b/providers/tests/system/google/cloud/life_sciences/example_life_sciences.py
index 170eeb39ad..e5024562b7 100644
--- a/providers/tests/system/google/cloud/life_sciences/example_life_sciences.py
+++ b/providers/tests/system/google/cloud/life_sciences/example_life_sciences.py
@@ -131,14 +131,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/ml_engine/example_mlengine.py b/providers/tests/system/google/cloud/ml_engine/example_mlengine.py
index f0eaf7cb30..a735d3015a 100644
--- a/providers/tests/system/google/cloud/ml_engine/example_mlengine.py
+++ b/providers/tests/system/google/cloud/ml_engine/example_mlengine.py
@@ -280,13 +280,13 @@ with models.DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/natural_language/example_natural_language.py b/providers/tests/system/google/cloud/natural_language/example_natural_language.py
index cc1aba0f16..4eb542d9ae 100644
--- a/providers/tests/system/google/cloud/natural_language/example_natural_language.py
+++ b/providers/tests/system/google/cloud/natural_language/example_natural_language.py
@@ -119,13 +119,13 @@ with DAG(
analyze_sentiment >> analyze_sentiment_result
analyze_classify_text >> analyze_classify_text_result
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/pubsub/example_pubsub.py b/providers/tests/system/google/cloud/pubsub/example_pubsub.py
index 93ab1cde6e..9ff909a3b4 100644
--- a/providers/tests/system/google/cloud/pubsub/example_pubsub.py
+++ b/providers/tests/system/google/cloud/pubsub/example_pubsub.py
@@ -146,14 +146,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/pubsub/example_pubsub_deferrable.py b/providers/tests/system/google/cloud/pubsub/example_pubsub_deferrable.py
index a902ea5617..f889ff16a2 100644
--- a/providers/tests/system/google/cloud/pubsub/example_pubsub_deferrable.py
+++ b/providers/tests/system/google/cloud/pubsub/example_pubsub_deferrable.py
@@ -101,14 +101,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/spanner/example_spanner.py b/providers/tests/system/google/cloud/spanner/example_spanner.py
index b11a5cd61f..ba9f727e1f 100644
--- a/providers/tests/system/google/cloud/spanner/example_spanner.py
+++ b/providers/tests/system/google/cloud/spanner/example_spanner.py
@@ -161,14 +161,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/speech_to_text/example_speech_to_text.py b/providers/tests/system/google/cloud/speech_to_text/example_speech_to_text.py
index f2382a6053..16603bf65a 100644
--- a/providers/tests/system/google/cloud/speech_to_text/example_speech_to_text.py
+++ b/providers/tests/system/google/cloud/speech_to_text/example_speech_to_text.py
@@ -89,14 +89,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/sql_to_sheets/example_sql_to_sheets.py b/providers/tests/system/google/cloud/sql_to_sheets/example_sql_to_sheets.py
index d34531ba52..65c88b38cc 100644
--- a/providers/tests/system/google/cloud/sql_to_sheets/example_sql_to_sheets.py
+++ b/providers/tests/system/google/cloud/sql_to_sheets/example_sql_to_sheets.py
@@ -315,13 +315,13 @@ with DAG(
]
delete_gce_instance >> delete_persistent_disk
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/stackdriver/example_stackdriver.py b/providers/tests/system/google/cloud/stackdriver/example_stackdriver.py
index 202c9bcfbd..e720ac1f16 100644
--- a/providers/tests/system/google/cloud/stackdriver/example_stackdriver.py
+++ b/providers/tests/system/google/cloud/stackdriver/example_stackdriver.py
@@ -228,13 +228,13 @@ with DAG(
delete_alert_policy_2,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_aws.py b/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_aws.py
index 4d6a010954..cf76f6eddd 100644
--- a/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_aws.py
+++ b/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_aws.py
@@ -254,14 +254,14 @@ with DAG(
]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_gcp.py b/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_gcp.py
index b2cef2f831..960a9c0679 100644
--- a/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_gcp.py
+++ b/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_gcp.py
@@ -195,13 +195,13 @@ with DAG(
>> [delete_transfer, delete_bucket_src, delete_bucket_dst]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_gcs_to_gcs.py b/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_gcs_to_gcs.py
index e670eba755..46289ea66c 100644
--- a/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_gcs_to_gcs.py
+++ b/providers/tests/system/google/cloud/storage_transfer/example_cloud_storage_transfer_service_gcs_to_gcs.py
@@ -106,13 +106,13 @@ with DAG(
>> [delete_bucket_src, delete_bucket_dst]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/tasks/example_queue.py b/providers/tests/system/google/cloud/tasks/example_queue.py
index 9797a29d04..614bdfd352 100644
--- a/providers/tests/system/google/cloud/tasks/example_queue.py
+++ b/providers/tests/system/google/cloud/tasks/example_queue.py
@@ -161,14 +161,14 @@ with DAG(
delete_queue,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/tasks/example_tasks.py b/providers/tests/system/google/cloud/tasks/example_tasks.py
index 0eae95fd10..f4ab2f53b0 100644
--- a/providers/tests/system/google/cloud/tasks/example_tasks.py
+++ b/providers/tests/system/google/cloud/tasks/example_tasks.py
@@ -152,14 +152,14 @@ with DAG(
random_string, create_queue, create_task, tasks_get, list_tasks, run_task, delete_task, delete_queue
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/text_to_speech/example_text_to_speech.py b/providers/tests/system/google/cloud/text_to_speech/example_text_to_speech.py
index 0227049508..437db735c2 100644
--- a/providers/tests/system/google/cloud/text_to_speech/example_text_to_speech.py
+++ b/providers/tests/system/google/cloud/text_to_speech/example_text_to_speech.py
@@ -78,14 +78,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/transfers/example_gcs_to_sftp.py b/providers/tests/system/google/cloud/transfers/example_gcs_to_sftp.py
index 8d3f194163..dd1a5945f2 100644
--- a/providers/tests/system/google/cloud/transfers/example_gcs_to_sftp.py
+++ b/providers/tests/system/google/cloud/transfers/example_gcs_to_sftp.py
@@ -171,14 +171,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/transfers/example_gdrive_to_local.py b/providers/tests/system/google/cloud/transfers/example_gdrive_to_local.py
index abfcbabce0..2e5c6f8e8f 100644
--- a/providers/tests/system/google/cloud/transfers/example_gdrive_to_local.py
+++ b/providers/tests/system/google/cloud/transfers/example_gdrive_to_local.py
@@ -167,13 +167,13 @@ with DAG(
>> [delete_bucket, delete_connection_task]
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/transfers/example_postgres_to_gcs.py b/providers/tests/system/google/cloud/transfers/example_postgres_to_gcs.py
index 91bde02798..0f9d5be7ea 100644
--- a/providers/tests/system/google/cloud/transfers/example_postgres_to_gcs.py
+++ b/providers/tests/system/google/cloud/transfers/example_postgres_to_gcs.py
@@ -290,13 +290,13 @@ with DAG(
postgres_to_gcs >> [delete_gcs_bucket, delete_firewall_rule, delete_gce_instance, delete_connection_task]
delete_gce_instance >> delete_persistent_disk
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/translate/example_translate.py b/providers/tests/system/google/cloud/translate/example_translate.py
index 13a3ee061e..44af8bf5d1 100644
--- a/providers/tests/system/google/cloud/translate/example_translate.py
+++ b/providers/tests/system/google/cloud/translate/example_translate.py
@@ -56,14 +56,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/translate_speech/example_translate_speech.py b/providers/tests/system/google/cloud/translate_speech/example_translate_speech.py
index 4fe34d0b86..2e023d68ae 100644
--- a/providers/tests/system/google/cloud/translate_speech/example_translate_speech.py
+++ b/providers/tests/system/google/cloud/translate_speech/example_translate_speech.py
@@ -110,14 +110,14 @@ with DAG(
>> delete_bucket
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_forecasting_training.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_forecasting_training.py
index 8dd0ecc9c9..e22622dd58 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_forecasting_training.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_forecasting_training.py
@@ -150,13 +150,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_image_training.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_image_training.py
index bf96da992a..77629b0bc7 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_image_training.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_image_training.py
@@ -135,13 +135,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_list_training.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_list_training.py
index ca6c65e1a6..cd75035793 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_list_training.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_list_training.py
@@ -52,13 +52,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_tabular_training.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_tabular_training.py
index fb32c7475c..106e622aba 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_tabular_training.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_tabular_training.py
@@ -140,13 +140,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_video_training.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_video_training.py
index 6c3db89382..ea68e55fa2 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_video_training.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_auto_ml_video_training.py
@@ -145,13 +145,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_batch_prediction_job.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_batch_prediction_job.py
index 78bb9ffa6b..06a56f3796 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_batch_prediction_job.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_batch_prediction_job.py
@@ -231,13 +231,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_container.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_container.py
index 1295d20983..316e0d3a7c 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_container.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_container.py
@@ -208,13 +208,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_job.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_job.py
index 3fa6169cb2..eb2a080a92 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_job.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_job.py
@@ -250,13 +250,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_job_python_package.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_job_python_package.py
index fbc40b8882..f1b471d446 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_job_python_package.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_custom_job_python_package.py
@@ -211,13 +211,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_dataset.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_dataset.py
index 3e1b98a223..af4fbb88c2 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_dataset.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_dataset.py
@@ -269,13 +269,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_endpoint.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_endpoint.py
index 12b1181a79..13e24b869c 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_endpoint.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_endpoint.py
@@ -206,13 +206,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_generative_model.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_generative_model.py
index 7e696184ee..a1140afc19 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_generative_model.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_generative_model.py
@@ -155,14 +155,14 @@ with DAG(
)
# [END how_to_cloud_vertex_ai_run_evaluation_operator]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_generative_model_tuning.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_generative_model_tuning.py
index affba0b6ae..9ce1c77350 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_generative_model_tuning.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_generative_model_tuning.py
@@ -56,13 +56,13 @@ with DAG(
)
# [END how_to_cloud_vertex_ai_supervised_fine_tuning_train_operator]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_hyperparameter_tuning_job.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_hyperparameter_tuning_job.py
index 55254741d9..4fabdbc6ac 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_hyperparameter_tuning_job.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_hyperparameter_tuning_job.py
@@ -181,13 +181,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_list_custom_jobs.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_list_custom_jobs.py
index 75de719b6a..230aeca14a 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_list_custom_jobs.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_list_custom_jobs.py
@@ -51,13 +51,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_model_service.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_model_service.py
index 3ad5537a10..272069b2f8 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_model_service.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_model_service.py
@@ -345,13 +345,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_pipeline_job.py b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_pipeline_job.py
index 5ab29ee999..bc97ff6a58 100644
--- a/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_pipeline_job.py
+++ b/providers/tests/system/google/cloud/vertex_ai/example_vertex_ai_pipeline_job.py
@@ -177,13 +177,13 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/video_intelligence/example_video_intelligence.py b/providers/tests/system/google/cloud/video_intelligence/example_video_intelligence.py
index e630fd1471..21ee27cd9b 100644
--- a/providers/tests/system/google/cloud/video_intelligence/example_video_intelligence.py
+++ b/providers/tests/system/google/cloud/video_intelligence/example_video_intelligence.py
@@ -154,14 +154,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vision/example_vision_annotate_image.py b/providers/tests/system/google/cloud/vision/example_vision_annotate_image.py
index 1e09fb7fe4..178920e83a 100644
--- a/providers/tests/system/google/cloud/vision/example_vision_annotate_image.py
+++ b/providers/tests/system/google/cloud/vision/example_vision_annotate_image.py
@@ -191,14 +191,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vision/example_vision_autogenerated.py b/providers/tests/system/google/cloud/vision/example_vision_autogenerated.py
index 11fd3cdd54..e84af0fdba 100644
--- a/providers/tests/system/google/cloud/vision/example_vision_autogenerated.py
+++ b/providers/tests/system/google/cloud/vision/example_vision_autogenerated.py
@@ -268,14 +268,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/vision/example_vision_explicit.py b/providers/tests/system/google/cloud/vision/example_vision_explicit.py
index 0c71be95bd..2e480fec4f 100644
--- a/providers/tests/system/google/cloud/vision/example_vision_explicit.py
+++ b/providers/tests/system/google/cloud/vision/example_vision_explicit.py
@@ -279,14 +279,14 @@ with DAG(
delete_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "teardown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/cloud/workflows/example_workflows.py b/providers/tests/system/google/cloud/workflows/example_workflows.py
index 6d13484acb..b933d75a6f 100644
--- a/providers/tests/system/google/cloud/workflows/example_workflows.py
+++ b/providers/tests/system/google/cloud/workflows/example_workflows.py
@@ -227,14 +227,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/datacatalog/example_datacatalog_entries.py b/providers/tests/system/google/datacatalog/example_datacatalog_entries.py
index 0f84361d4f..644adadeef 100644
--- a/providers/tests/system/google/datacatalog/example_datacatalog_entries.py
+++ b/providers/tests/system/google/datacatalog/example_datacatalog_entries.py
@@ -200,14 +200,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/datacatalog/example_datacatalog_search_catalog.py b/providers/tests/system/google/datacatalog/example_datacatalog_search_catalog.py
index a77a023f32..0ff6cf0716 100644
--- a/providers/tests/system/google/datacatalog/example_datacatalog_search_catalog.py
+++ b/providers/tests/system/google/datacatalog/example_datacatalog_search_catalog.py
@@ -223,14 +223,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/datacatalog/example_datacatalog_tag_templates.py b/providers/tests/system/google/datacatalog/example_datacatalog_tag_templates.py
index af49d3de08..19bf62c88c 100644
--- a/providers/tests/system/google/datacatalog/example_datacatalog_tag_templates.py
+++ b/providers/tests/system/google/datacatalog/example_datacatalog_tag_templates.py
@@ -183,14 +183,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/datacatalog/example_datacatalog_tags.py b/providers/tests/system/google/datacatalog/example_datacatalog_tags.py
index fdbdaa4510..0f74b49e10 100644
--- a/providers/tests/system/google/datacatalog/example_datacatalog_tags.py
+++ b/providers/tests/system/google/datacatalog/example_datacatalog_tags.py
@@ -233,14 +233,14 @@ with DAG(
# ### Everything below this line is not part of example ###
# ### Just for system tests purpose ###
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/leveldb/example_leveldb.py b/providers/tests/system/google/leveldb/example_leveldb.py
index 8b11d9e87b..52a543f048 100644
--- a/providers/tests/system/google/leveldb/example_leveldb.py
+++ b/providers/tests/system/google/leveldb/example_leveldb.py
@@ -60,14 +60,14 @@ with DAG(
# [END howto_operator_leveldb_put_key]
get_key_leveldb_task >> put_key_leveldb_task
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/marketing_platform/example_analytics_admin.py b/providers/tests/system/google/marketing_platform/example_analytics_admin.py
index 16ce6f8190..3536e7b945 100644
--- a/providers/tests/system/google/marketing_platform/example_analytics_admin.py
+++ b/providers/tests/system/google/marketing_platform/example_analytics_admin.py
@@ -219,13 +219,13 @@ with DAG(
# TEST TEARDOWN
>> delete_connection_task
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/marketing_platform/example_campaign_manager.py b/providers/tests/system/google/marketing_platform/example_campaign_manager.py
index 0c3b26e5e9..70e8c208d5 100644
--- a/providers/tests/system/google/marketing_platform/example_campaign_manager.py
+++ b/providers/tests/system/google/marketing_platform/example_campaign_manager.py
@@ -322,13 +322,13 @@ with DAG(
>> delete_connection(connection_id=CONNECTION_ID)
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/marketing_platform/example_search_ads.py b/providers/tests/system/google/marketing_platform/example_search_ads.py
index 5d86a3ad16..cb733f5bcf 100644
--- a/providers/tests/system/google/marketing_platform/example_search_ads.py
+++ b/providers/tests/system/google/marketing_platform/example_search_ads.py
@@ -102,7 +102,7 @@ with DAG(
(query_report >> get_field >> search_fields >> get_custom_column >> list_custom_columns)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/google/suite/example_local_to_drive.py b/providers/tests/system/google/suite/example_local_to_drive.py
index ac8ebfc5aa..397e65fa2b 100644
--- a/providers/tests/system/google/suite/example_local_to_drive.py
+++ b/providers/tests/system/google/suite/example_local_to_drive.py
@@ -141,13 +141,13 @@ with DAG(
>> delete_connection_task
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/http/example_http.py b/providers/tests/system/http/example_http.py
index 9842394360..a2900ad582 100644
--- a/providers/tests/system/http/example_http.py
+++ b/providers/tests/system/http/example_http.py
@@ -157,7 +157,7 @@ task_get_paginated = HttpOperator(
task_get_op_response_filter >> task_put_op >> task_del_op >> task_post_op_formenc
task_post_op_formenc >> task_get_paginated
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/influxdb/example_influxdb.py b/providers/tests/system/influxdb/example_influxdb.py
index 8e4d486742..9b67b05d0a 100644
--- a/providers/tests/system/influxdb/example_influxdb.py
+++ b/providers/tests/system/influxdb/example_influxdb.py
@@ -61,13 +61,13 @@ with DAG(
) as dag:
test_influxdb_hook()
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/influxdb/example_influxdb_query.py b/providers/tests/system/influxdb/example_influxdb_query.py
index 95940e8c8f..8c4c548b84 100644
--- a/providers/tests/system/influxdb/example_influxdb_query.py
+++ b/providers/tests/system/influxdb/example_influxdb_query.py
@@ -43,7 +43,7 @@ with DAG(
# [END howto_operator_influxdb]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/jdbc/example_jdbc_queries.py b/providers/tests/system/jdbc/example_jdbc_queries.py
index ce9234a6fb..5e73c22b3c 100644
--- a/providers/tests/system/jdbc/example_jdbc_queries.py
+++ b/providers/tests/system/jdbc/example_jdbc_queries.py
@@ -59,13 +59,13 @@ with DAG(
delete_data >> insert_data >> run_this_last
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/jenkins/example_jenkins_job_trigger.py b/providers/tests/system/jenkins/example_jenkins_job_trigger.py
index 6b7fab62a0..a24b48dda6 100644
--- a/providers/tests/system/jenkins/example_jenkins_job_trigger.py
+++ b/providers/tests/system/jenkins/example_jenkins_job_trigger.py
@@ -72,7 +72,7 @@ with DAG(
# job_trigger >> grab_artifact_from_jenkins()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_adf_run_pipeline.py b/providers/tests/system/microsoft/azure/example_adf_run_pipeline.py
index a6eae4d714..e65524b3c7 100644
--- a/providers/tests/system/microsoft/azure/example_adf_run_pipeline.py
+++ b/providers/tests/system/microsoft/azure/example_adf_run_pipeline.py
@@ -108,13 +108,13 @@ with DAG(
# Task dependency created via `XComArgs`:
# run_pipeline2 >> pipeline_run_sensor
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_adls_create.py b/providers/tests/system/microsoft/azure/example_adls_create.py
index 3f76525b6b..2c5828da44 100644
--- a/providers/tests/system/microsoft/azure/example_adls_create.py
+++ b/providers/tests/system/microsoft/azure/example_adls_create.py
@@ -46,13 +46,13 @@ with models.DAG(
upload_data >> delete_file
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_adls_delete.py b/providers/tests/system/microsoft/azure/example_adls_delete.py
index 34abc6a9b2..fcfe1793f2 100644
--- a/providers/tests/system/microsoft/azure/example_adls_delete.py
+++ b/providers/tests/system/microsoft/azure/example_adls_delete.py
@@ -46,13 +46,13 @@ with models.DAG(
upload_file >> remove_file
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_adls_list.py b/providers/tests/system/microsoft/azure/example_adls_list.py
index 594b161aaa..26e65e0cf8 100644
--- a/providers/tests/system/microsoft/azure/example_adls_list.py
+++ b/providers/tests/system/microsoft/azure/example_adls_list.py
@@ -42,13 +42,13 @@ with models.DAG(
)
# [END howto_operator_adls_list]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_azure_batch_operator.py b/providers/tests/system/microsoft/azure/example_azure_batch_operator.py
index 944c737f96..33d4603c16 100644
--- a/providers/tests/system/microsoft/azure/example_azure_batch_operator.py
+++ b/providers/tests/system/microsoft/azure/example_azure_batch_operator.py
@@ -57,7 +57,7 @@ with DAG(
)
# [END howto_azure_batch_operator]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_azure_container_instances.py b/providers/tests/system/microsoft/azure/example_azure_container_instances.py
index ee9b025193..fe7ccfd48e 100644
--- a/providers/tests/system/microsoft/azure/example_azure_container_instances.py
+++ b/providers/tests/system/microsoft/azure/example_azure_container_instances.py
@@ -90,7 +90,7 @@ with DAG(
cpu=1.0,
task_id="start_container_with_azure_container_volume",
)
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_azure_cosmosdb.py b/providers/tests/system/microsoft/azure/example_azure_cosmosdb.py
index 1d4bb60b05..7dc9524ef4 100644
--- a/providers/tests/system/microsoft/azure/example_azure_cosmosdb.py
+++ b/providers/tests/system/microsoft/azure/example_azure_cosmosdb.py
@@ -64,13 +64,13 @@ with DAG(
t1 >> t2
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_azure_service_bus.py b/providers/tests/system/microsoft/azure/example_azure_service_bus.py
index bfa77db40d..091fdff3c2 100644
--- a/providers/tests/system/microsoft/azure/example_azure_service_bus.py
+++ b/providers/tests/system/microsoft/azure/example_azure_service_bus.py
@@ -172,13 +172,13 @@ with DAG(
delete_service_bus_queue,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_azure_synapse.py b/providers/tests/system/microsoft/azure/example_azure_synapse.py
index c7f7800e57..571cd5b46f 100644
--- a/providers/tests/system/microsoft/azure/example_azure_synapse.py
+++ b/providers/tests/system/microsoft/azure/example_azure_synapse.py
@@ -69,7 +69,7 @@ with DAG(
)
# [END howto_operator_azure_synapse]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_fileshare.py b/providers/tests/system/microsoft/azure/example_fileshare.py
index bfa9819dbe..49909ae38f 100644
--- a/providers/tests/system/microsoft/azure/example_fileshare.py
+++ b/providers/tests/system/microsoft/azure/example_fileshare.py
@@ -55,13 +55,13 @@ with DAG(
) as dag:
create_fileshare() >> delete_fileshare()
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_local_to_adls.py b/providers/tests/system/microsoft/azure/example_local_to_adls.py
index d540aaf34d..f8eace7577 100644
--- a/providers/tests/system/microsoft/azure/example_local_to_adls.py
+++ b/providers/tests/system/microsoft/azure/example_local_to_adls.py
@@ -47,13 +47,13 @@ with models.DAG(
upload_file >> delete_file
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_local_to_wasb.py b/providers/tests/system/microsoft/azure/example_local_to_wasb.py
index 7e368ca839..859c339758 100644
--- a/providers/tests/system/microsoft/azure/example_local_to_wasb.py
+++ b/providers/tests/system/microsoft/azure/example_local_to_wasb.py
@@ -49,13 +49,13 @@ with DAG(
upload >> delete
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_msfabric.py b/providers/tests/system/microsoft/azure/example_msfabric.py
index b111302528..0f65df2f72 100644
--- a/providers/tests/system/microsoft/azure/example_msfabric.py
+++ b/providers/tests/system/microsoft/azure/example_msfabric.py
@@ -51,13 +51,13 @@ with models.DAG(
)
# [END howto_operator_ms_fabric_create_item_schedule]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_msgraph.py b/providers/tests/system/microsoft/azure/example_msgraph.py
index 33ee004685..239ebb76ab 100644
--- a/providers/tests/system/microsoft/azure/example_msgraph.py
+++ b/providers/tests/system/microsoft/azure/example_msgraph.py
@@ -49,13 +49,13 @@ with models.DAG(
site_task >> site_pages_task
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_powerbi.py b/providers/tests/system/microsoft/azure/example_powerbi.py
index b5a982a115..a2d9d67547 100644
--- a/providers/tests/system/microsoft/azure/example_powerbi.py
+++ b/providers/tests/system/microsoft/azure/example_powerbi.py
@@ -97,13 +97,13 @@ with models.DAG(
workspaces_task >> workspaces_info_task >> check_workspace_status_task
refresh_dataset_task >> refresh_dataset_history_task
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_powerbi_dataset_refresh.py b/providers/tests/system/microsoft/azure/example_powerbi_dataset_refresh.py
index c02cec3e57..5453caff64 100644
--- a/providers/tests/system/microsoft/azure/example_powerbi_dataset_refresh.py
+++ b/providers/tests/system/microsoft/azure/example_powerbi_dataset_refresh.py
@@ -76,13 +76,13 @@ with DAG(
refresh_powerbi_dataset,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_s3_to_wasb.py b/providers/tests/system/microsoft/azure/example_s3_to_wasb.py
index 88ae64324d..699da911d0 100644
--- a/providers/tests/system/microsoft/azure/example_s3_to_wasb.py
+++ b/providers/tests/system/microsoft/azure/example_s3_to_wasb.py
@@ -103,13 +103,13 @@ with DAG(dag_id=DAG_ID, start_date=datetime(2024, 1, 1), schedule="@once", catch
remove_s3_bucket,
)
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure when "tearDown" task with trigger
# rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_sftp_to_wasb.py b/providers/tests/system/microsoft/azure/example_sftp_to_wasb.py
index 42e5968304..0ed953a585 100644
--- a/providers/tests/system/microsoft/azure/example_sftp_to_wasb.py
+++ b/providers/tests/system/microsoft/azure/example_sftp_to_wasb.py
@@ -75,13 +75,13 @@ with DAG(
transfer_files_to_sftp_step >> transfer_files_to_azure >> delete_blob_file_step >> delete_sftp_file()
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_synapse_run_pipeline.py b/providers/tests/system/microsoft/azure/example_synapse_run_pipeline.py
index de87d6b4c2..589f840e46 100644
--- a/providers/tests/system/microsoft/azure/example_synapse_run_pipeline.py
+++ b/providers/tests/system/microsoft/azure/example_synapse_run_pipeline.py
@@ -43,13 +43,13 @@ with DAG(
# [END howto_operator_azure_synapse_run_pipeline]
begin >> run_pipeline1
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/azure/example_wasb_sensors.py b/providers/tests/system/microsoft/azure/example_wasb_sensors.py
index 56c1dce34d..c4dd7c8fa0 100644
--- a/providers/tests/system/microsoft/azure/example_wasb_sensors.py
+++ b/providers/tests/system/microsoft/azure/example_wasb_sensors.py
@@ -62,7 +62,7 @@ with DAG(
# [END wasb_prefix_sensor]
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/mssql/example_mssql.py b/providers/tests/system/microsoft/mssql/example_mssql.py
index 12e2815c59..8e29d805ab 100644
--- a/providers/tests/system/microsoft/mssql/example_mssql.py
+++ b/providers/tests/system/microsoft/mssql/example_mssql.py
@@ -146,12 +146,12 @@ with DAG(
)
# [END mssql_operator_howto_guide]
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/microsoft/winrm/example_winrm.py b/providers/tests/system/microsoft/winrm/example_winrm.py
index 1de1cb8c0f..1c7e31ed67 100644
--- a/providers/tests/system/microsoft/winrm/example_winrm.py
+++ b/providers/tests/system/microsoft/winrm/example_winrm.py
@@ -64,13 +64,13 @@ with DAG(
[t1, t2, t3] >> run_this_last
- from dev.tests_common.test_utils.watcher import watcher
+ from tests_common.test_utils.watcher import watcher
# This test needs watcher in order to properly mark success/failure
# when "tearDown" task with trigger rule is part of the DAG
list(dag.tasks) >> watcher()
-from dev.tests_common.test_utils.system_tests import get_test_run # noqa: E402
+from tests_common.test_utils.system_tests import get_test_run # noqa: E402
# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)
diff --git a/providers/tests/system/mysql/example_mysql.py b/providers/tests/system/mysql/example_mysql.py
index 6f0f884197..2f1f37f573 100644
--- a/providers/tests/system/mysql/example_mysql.py
+++ b/providers/tests/system/mysql/example_mysql.py
@@ -59,7 +59,7 @@ with DAG(
drop_table_mysql_task >> mysql_task
... 5822 lines suppressed ...