Migrate Jenkins example DAGs to new design #22451 (#24138)

apache · Jun 3, 2022 · 5bd6418 · 5bd6418
1 parent 9f7ab0d
commit 5bd6418
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 19 deletions.
diff --git a/airflow/providers/jenkins/example_dags/__init__.py b/airflow/providers/jenkins/example_dags/__init__.py
diff --git a/docs/apache-airflow-providers-jenkins/index.rst b/docs/apache-airflow-providers-jenkins/index.rst
@@ -37,7 +37,7 @@ Content
     :maxdepth: 1
     :caption: Resources
 
-    Example DAGs <https://github.com/apache/airflow/tree/main/airflow/providers/jenkins/example_dags>
+    Example DAGs <https://github.com/apache/airflow/tree/main/tests/system/providers/jenkins>
     PyPI Repository <https://pypi.org/project/apache-airflow-providers-jenkins/>
     Installing from sources <installing-providers-from-sources>
 

diff --git a/...ample_dags/example_jenkins_job_trigger.py → ...rs/jenkins/example_jenkins_job_trigger.py b/...ample_dags/example_jenkins_job_trigger.py → ...rs/jenkins/example_jenkins_job_trigger.py
@@ -15,6 +15,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+import os
 from datetime import datetime
 
 from requests import Request
@@ -25,9 +26,11 @@
 from airflow.providers.jenkins.operators.jenkins_job_trigger import JenkinsJobTriggerOperator
 
 JENKINS_CONNECTION_ID = "your_jenkins_connection"
+ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID")
+DAG_ID = "test_jenkins"
 
 with DAG(
-    "test_jenkins",
+    DAG_ID,
     default_args={
         "retries": 1,
         "concurrency": 8,
@@ -65,3 +68,9 @@ def grab_artifact_from_jenkins(url):
 
     # Task dependency created via `XComArgs`:
     #   job_trigger >> grab_artifact_from_jenkins()
+
+
+from tests.system.utils import get_test_run  # noqa: E402
+
+# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
+test_run = get_test_run(dag)