From ac345d0e6635d2809b730e3e659f369e3571511b Mon Sep 17 00:00:00 2001
From: Anil Belur <askb23@gmail.com>
Date: Sat, 10 Mar 2018 22:21:56 +1000
Subject: [PATCH] Add support for upstream downstream job filter

Add support for upstream downstream job filter for views.

Change-Id: I8869f804a6f85f8bee7720d785f03492c87ff959
Signed-off-by: Anil Belur <askb23@gmail.com>
---
 jenkins_jobs/modules/view_list.py             | 26 ++++++++++++++++
 ...w_list_upstream_downstream_filter_full.xml | 30 +++++++++++++++++++
 ..._list_upstream_downstream_filter_full.yaml | 13 ++++++++
 ...ew_list_upstream_downstream_filter_min.xml | 30 +++++++++++++++++++
 ...w_list_upstream_downstream_filter_min.yaml | 10 +++++++
 5 files changed, 109 insertions(+)
 create mode 100644 tests/views/fixtures/view_list_upstream_downstream_filter_full.xml
 create mode 100644 tests/views/fixtures/view_list_upstream_downstream_filter_full.yaml
 create mode 100644 tests/views/fixtures/view_list_upstream_downstream_filter_min.xml
 create mode 100644 tests/views/fixtures/view_list_upstream_downstream_filter_min.yaml

diff --git a/jenkins_jobs/modules/view_list.py b/jenkins_jobs/modules/view_list.py
index 1d17b4341..e9946f951 100644
--- a/jenkins_jobs/modules/view_list.py
+++ b/jenkins_jobs/modules/view_list.py
@@ -187,6 +187,17 @@ to the :ref:`view_list` definition.
                 * **permission-check**: ('str'): Match user permissions.
                   (default MustMatchAll)
 
+        * **upstream-downstream** (`dict`)
+            :upstream-downstream:
+                * **include-upstream** ('bool'): Jobs that match upstream.
+                  (default False)
+                * **include-downstream** ('bool'): Jobs that match downstream.
+                  (default False)
+                * **recursive** ('bool'): Jobs that are recursive.
+                  (default False)
+                * **exclude-originals** ('bool'): Jobs that are originals.
+                  (default False)
+
     * **columns** (`list`): List of columns to be shown in view.
     * **regex** (`str`): . Regular expression for selecting jobs
       (optional)
@@ -325,6 +336,21 @@ class List(jenkins_jobs.modules.base.Base):
                 convert_mapping_to_xml(js_xml, js_data, mapping,
                                        fail_required=True)
 
+            if jobfilter == 'upstream-downstream':
+                ud_xml = XML.SubElement(job_filter_xml,
+                    'hudson.views.UpstreamDownstreamJobsFilter')
+                ud_xml.set('plugin', 'view-job-filters')
+                ud_data = jobfilters.get('upstream-downstream')
+                mapping = [
+                    ('include-upstream', 'includeUpstream',
+                        False),
+                    ('include-downstream', 'includeDownstream', False),
+                    ('recursive', 'recursive', False),
+                    ('exclude-originals', 'excludeOriginals', False),
+                ]
+                convert_mapping_to_xml(ud_xml, ud_data, mapping,
+                                       fail_required=True)
+
             if jobfilter == 'fallback':
                 fb_xml = XML.SubElement(job_filter_xml,
                                         'hudson.views.AddRemoveFallbackFilter')
diff --git a/tests/views/fixtures/view_list_upstream_downstream_filter_full.xml b/tests/views/fixtures/view_list_upstream_downstream_filter_full.xml
new file mode 100644
index 000000000..160d8f50d
--- /dev/null
+++ b/tests/views/fixtures/view_list_upstream_downstream_filter_full.xml
@@ -0,0 +1,30 @@
+<?xml version="1.0" encoding="utf-8"?>
+<hudson.model.ListView>
+  <name>list-view-name01</name>
+  <description>Sample description</description>
+  <filterExecutors>true</filterExecutors>
+  <filterQueue>true</filterQueue>
+  <properties class="hudson.model.View$PropertyList"/>
+  <jobNames>
+    <comparator class="hudson.util.CaseInsensitiveComparator"/>
+  </jobNames>
+  <jobFilters>
+    <hudson.views.UpstreamDownstreamJobsFilter plugin="view-job-filters">
+      <includeUpstream>true</includeUpstream>
+      <includeDownstream>true</includeDownstream>
+      <recursive>true</recursive>
+      <excludeOriginals>true</excludeOriginals>
+    </hudson.views.UpstreamDownstreamJobsFilter>
+  </jobFilters>
+  <columns>
+    <hudson.views.StatusColumn/>
+    <hudson.views.WeatherColumn/>
+    <hudson.views.JobColumn/>
+    <hudson.views.LastSuccessColumn/>
+    <hudson.views.LastFailureColumn/>
+    <hudson.views.LastDurationColumn/>
+    <hudson.views.BuildButtonColumn/>
+  </columns>
+  <recurse>true</recurse>
+  <statusFilter>false</statusFilter>
+</hudson.model.ListView>
diff --git a/tests/views/fixtures/view_list_upstream_downstream_filter_full.yaml b/tests/views/fixtures/view_list_upstream_downstream_filter_full.yaml
new file mode 100644
index 000000000..2e56e2068
--- /dev/null
+++ b/tests/views/fixtures/view_list_upstream_downstream_filter_full.yaml
@@ -0,0 +1,13 @@
+name: list-view-name01
+view-type: list
+description: 'Sample description'
+filter-executors: true
+filter-queue: true
+job-filters:
+  upstream-downstream:
+    include-downstream: true
+    include-upstream: true
+    recursive: true
+    exclude-originals: true
+recurse: true
+status-filter: false
diff --git a/tests/views/fixtures/view_list_upstream_downstream_filter_min.xml b/tests/views/fixtures/view_list_upstream_downstream_filter_min.xml
new file mode 100644
index 000000000..6fcf2a09e
--- /dev/null
+++ b/tests/views/fixtures/view_list_upstream_downstream_filter_min.xml
@@ -0,0 +1,30 @@
+<?xml version="1.0" encoding="utf-8"?>
+<hudson.model.ListView>
+  <name>list-view-name01</name>
+  <description>Sample description</description>
+  <filterExecutors>true</filterExecutors>
+  <filterQueue>true</filterQueue>
+  <properties class="hudson.model.View$PropertyList"/>
+  <jobNames>
+    <comparator class="hudson.util.CaseInsensitiveComparator"/>
+  </jobNames>
+  <jobFilters>
+    <hudson.views.UpstreamDownstreamJobsFilter plugin="view-job-filters">
+      <includeUpstream>false</includeUpstream>
+      <includeDownstream>false</includeDownstream>
+      <recursive>false</recursive>
+      <excludeOriginals>false</excludeOriginals>
+    </hudson.views.UpstreamDownstreamJobsFilter>
+  </jobFilters>
+  <columns>
+    <hudson.views.StatusColumn/>
+    <hudson.views.WeatherColumn/>
+    <hudson.views.JobColumn/>
+    <hudson.views.LastSuccessColumn/>
+    <hudson.views.LastFailureColumn/>
+    <hudson.views.LastDurationColumn/>
+    <hudson.views.BuildButtonColumn/>
+  </columns>
+  <recurse>true</recurse>
+  <statusFilter>false</statusFilter>
+</hudson.model.ListView>
diff --git a/tests/views/fixtures/view_list_upstream_downstream_filter_min.yaml b/tests/views/fixtures/view_list_upstream_downstream_filter_min.yaml
new file mode 100644
index 000000000..099189d7c
--- /dev/null
+++ b/tests/views/fixtures/view_list_upstream_downstream_filter_min.yaml
@@ -0,0 +1,10 @@
+name: list-view-name01
+view-type: list
+description: 'Sample description'
+filter-executors: true
+filter-queue: true
+job-filters:
+  upstream-downstream:
+    include-downstream: false
+recurse: true
+status-filter: false