Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix example import tests after move of providers to new structure #46217

Merged
merged 1 commit into from
Jan 28, 2025
Merged

Fix example import tests after move of providers to new structure #46217

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions providers/apache/beam/tests/system/apache/beam/example_beam.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,14 +21,15 @@

from __future__ import annotations

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
from providers.apache.beam.tests.system.apache.beam.utils import (
GCS_INPUT,
GCS_JAR_DIRECT_RUNNER_BUCKET_NAME,
GCS_JAR_DIRECT_RUNNER_OBJECT_NAME,
START_DATE,
)

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator

with models.DAG(
Expand Down
7 changes: 4 additions & 3 deletions providers/apache/beam/tests/system/apache/beam/example_beam_java_flink.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,14 +21,15 @@

from __future__ import annotations

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
from providers.apache.beam.tests.system.apache.beam.utils import (
GCS_INPUT,
GCS_JAR_FLINK_RUNNER_BUCKET_NAME,
GCS_JAR_FLINK_RUNNER_OBJECT_NAME,
START_DATE,
)

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator

with models.DAG(
Expand Down
7 changes: 4 additions & 3 deletions providers/apache/beam/tests/system/apache/beam/example_beam_java_spark.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,14 +21,15 @@

from __future__ import annotations

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
from providers.apache.beam.tests.system.apache.beam.utils import (
GCS_INPUT,
GCS_JAR_SPARK_RUNNER_BUCKET_NAME,
GCS_JAR_SPARK_RUNNER_OBJECT_NAME,
START_DATE,
)

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator

with models.DAG(
Expand Down
7 changes: 4 additions & 3 deletions providers/apache/beam/tests/system/apache/beam/example_go.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,7 @@

from __future__ import annotations

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator
from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
from providers.apache.beam.tests.system.apache.beam.utils import (
DEFAULT_ARGS,
GCP_PROJECT_ID,
GCS_GO,
Expand All @@ -32,6 +30,9 @@
GCS_TMP,
START_DATE,
)

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator
from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration

with models.DAG(
Expand Down
7 changes: 4 additions & 3 deletions providers/apache/beam/tests/system/apache/beam/example_go_dataflow.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,7 @@

from __future__ import annotations

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator
from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
from providers.apache.beam.tests.system.apache.beam.utils import (
DEFAULT_ARGS,
GCP_PROJECT_ID,
GCS_GO_DATAFLOW_ASYNC,
Expand All @@ -32,6 +30,9 @@
GCS_TMP,
START_DATE,
)

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator
from airflow.providers.google.cloud.hooks.dataflow import DataflowJobStatus
from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration
from airflow.providers.google.cloud.sensors.dataflow import DataflowJobStatusSensor
Expand Down
7 changes: 4 additions & 3 deletions providers/apache/beam/tests/system/apache/beam/example_java_dataflow.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,16 +21,17 @@

from __future__ import annotations

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
from providers.apache.beam.tests.system.apache.beam.utils import (
GCS_JAR_DATAFLOW_RUNNER_BUCKET_NAME,
GCS_JAR_DATAFLOW_RUNNER_OBJECT_NAME,
GCS_OUTPUT,
GCS_STAGING,
GCS_TMP,
START_DATE,
)

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator

with models.DAG(
Expand Down
7 changes: 4 additions & 3 deletions providers/apache/beam/tests/system/apache/beam/example_python.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,7 @@

from __future__ import annotations

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
from providers.apache.beam.tests.system.apache.beam.utils import (
DEFAULT_ARGS,
GCP_PROJECT_ID,
GCS_OUTPUT,
Expand All @@ -32,6 +30,9 @@
GCS_TMP,
START_DATE,
)

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration

with models.DAG(
Expand Down
7 changes: 4 additions & 3 deletions providers/apache/beam/tests/system/apache/beam/example_python_async.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,7 @@

from __future__ import annotations

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
from providers.apache.beam.tests.system.apache.beam.utils import (
DEFAULT_ARGS,
GCP_PROJECT_ID,
GCS_OUTPUT,
Expand All @@ -32,6 +30,9 @@
GCS_TMP,
START_DATE,
)

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration

with models.DAG(
Expand Down
7 changes: 4 additions & 3 deletions providers/apache/beam/tests/system/apache/beam/example_python_dataflow.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,7 @@

from __future__ import annotations

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
from providers.apache.beam.tests.system.apache.beam.utils import (
DEFAULT_ARGS,
GCP_PROJECT_ID,
GCS_OUTPUT,
Expand All @@ -32,6 +30,9 @@
GCS_TMP,
START_DATE,
)

from airflow import models
from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
from airflow.providers.google.cloud.hooks.dataflow import DataflowJobStatus
from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration
from airflow.providers.google.cloud.sensors.dataflow import DataflowJobStatusSensor
Expand Down
3 changes: 2 additions & 1 deletion providers/openlineage/tests/system/openlineage/example_openlineage.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,9 @@
import os
from datetime import datetime

from providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator

from airflow import DAG
from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator
from airflow.providers.standard.operators.python import PythonOperator


Expand Down
3 changes: 2 additions & 1 deletion providers/openlineage/tests/system/openlineage/example_openlineage_mapped_sensor.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,9 +19,10 @@
import os
from datetime import datetime, timedelta

from providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator

from airflow import DAG
from airflow.models import Variable
from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator
from airflow.providers.standard.operators.python import PythonOperator
from airflow.providers.standard.sensors.time_delta import TimeDeltaSensor

Expand Down
2 changes: 1 addition & 1 deletion providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,8 @@
from airflow.providers.google.cloud.operators.gcs import GCSCreateBucketOperator, GCSDeleteBucketOperator
from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator
from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator
from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator
from airflow.utils.trigger_rule import TriggerRule
from providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator

from providers.tests.system.google import DEFAULT_GCP_SYSTEM_TEST_PROJECT_ID

Expand Down
Loading