From 45f24585841aff5529d9928b31f26afe989b5ed0 Mon Sep 17 00:00:00 2001
From: Michal Ovciarik <michal.ovciarik@mavenir.com>
Date: Tue, 25 Feb 2020 10:24:38 +0100
Subject: [PATCH] Add support for become in ansible playbook

Change-Id: Ia116f715042d4a4c8cc26ac45cf1a91be47435d3
---
 jenkins_jobs/modules/builders.py              |  5 ++++
 .../builders/fixtures/ansible-playbook001.xml |  1 +
 .../builders/fixtures/ansible-playbook002.xml |  1 +
 .../builders/fixtures/ansible-playbook003.xml |  1 +
 .../builders/fixtures/ansible-playbook004.xml | 24 +++++++++++++++++++
 .../fixtures/ansible-playbook004.yaml         |  7 ++++++
 6 files changed, 39 insertions(+)
 create mode 100644 tests/builders/fixtures/ansible-playbook004.xml
 create mode 100644 tests/builders/fixtures/ansible-playbook004.yaml

diff --git a/jenkins_jobs/modules/builders.py b/jenkins_jobs/modules/builders.py
index 52c5155fb..b36707f5b 100644
--- a/jenkins_jobs/modules/builders.py
+++ b/jenkins_jobs/modules/builders.py
@@ -4460,6 +4460,11 @@ def ansible_playbook(parser, xml_parent, data):
         XML.SubElement(plugin, "sudoUser").text = data.get("sudo-user", "")
     else:
         XML.SubElement(plugin, "sudo").text = "false"
+    if data.get("become", False):
+        XML.SubElement(plugin, "become").text = "true"
+        XML.SubElement(plugin, "becomeUser").text = data.get("become-user", "")
+    else:
+        XML.SubElement(plugin, "become").text = "false"
     XML.SubElement(plugin, "forks").text = str(data.get("workers", "5"))
     XML.SubElement(plugin, "unbufferedOutput").text = str(
         data.get("unbuffered-output", True)
diff --git a/tests/builders/fixtures/ansible-playbook001.xml b/tests/builders/fixtures/ansible-playbook001.xml
index 662321b3a..d19d6068c 100644
--- a/tests/builders/fixtures/ansible-playbook001.xml
+++ b/tests/builders/fixtures/ansible-playbook001.xml
@@ -13,6 +13,7 @@
       <credentialsId/>
       <vaultCredentialsId/>
       <sudo>false</sudo>
+      <become>false</become>
       <forks>5</forks>
       <unbufferedOutput>true</unbufferedOutput>
       <colorizedOutput>false</colorizedOutput>
diff --git a/tests/builders/fixtures/ansible-playbook002.xml b/tests/builders/fixtures/ansible-playbook002.xml
index 21c1ae0b8..afe136761 100644
--- a/tests/builders/fixtures/ansible-playbook002.xml
+++ b/tests/builders/fixtures/ansible-playbook002.xml
@@ -18,6 +18,7 @@ machine02.example.com
       <vaultCredentialsId>0421b950-487b-4749-aa69-d87425e14459</vaultCredentialsId>
       <sudo>true</sudo>
       <sudoUser>cloud-user</sudoUser>
+      <become>false</become>
       <forks>2</forks>
       <unbufferedOutput>false</unbufferedOutput>
       <colorizedOutput>true</colorizedOutput>
diff --git a/tests/builders/fixtures/ansible-playbook003.xml b/tests/builders/fixtures/ansible-playbook003.xml
index 314749252..8e15e776f 100644
--- a/tests/builders/fixtures/ansible-playbook003.xml
+++ b/tests/builders/fixtures/ansible-playbook003.xml
@@ -11,6 +11,7 @@
       <credentialsId/>
       <vaultCredentialsId/>
       <sudo>false</sudo>
+      <become>false</become>
       <forks>5</forks>
       <unbufferedOutput>true</unbufferedOutput>
       <colorizedOutput>false</colorizedOutput>
diff --git a/tests/builders/fixtures/ansible-playbook004.xml b/tests/builders/fixtures/ansible-playbook004.xml
new file mode 100644
index 000000000..bc4018bfb
--- /dev/null
+++ b/tests/builders/fixtures/ansible-playbook004.xml
@@ -0,0 +1,24 @@
+<?xml version="1.0" encoding="utf-8"?>
+<project>
+  <builders>
+    <org.jenkinsci.plugins.ansible.AnsiblePlaybookBuilder>
+      <playbook>path/to/playbook.yml</playbook>
+      <inventory class="org.jenkinsci.plugins.ansible.InventoryDoNotSpecify"/>
+      <limit/>
+      <tags/>
+      <skippedTags/>
+      <startAtTask/>
+      <credentialsId/>
+      <vaultCredentialsId/>
+      <sudo>false</sudo>
+      <become>true</become>
+      <becomeUser>cloud-user</becomeUser>
+      <forks>5</forks>
+      <unbufferedOutput>true</unbufferedOutput>
+      <colorizedOutput>false</colorizedOutput>
+      <hostKeyChecking>false</hostKeyChecking>
+      <additionalParameters/>
+      <copyCredentialsInWorkspace>false</copyCredentialsInWorkspace>
+    </org.jenkinsci.plugins.ansible.AnsiblePlaybookBuilder>
+  </builders>
+</project>
diff --git a/tests/builders/fixtures/ansible-playbook004.yaml b/tests/builders/fixtures/ansible-playbook004.yaml
new file mode 100644
index 000000000..528c839a1
--- /dev/null
+++ b/tests/builders/fixtures/ansible-playbook004.yaml
@@ -0,0 +1,7 @@
+---
+builders:
+  - ansible-playbook:
+      playbook: "path/to/playbook.yml"
+      inventory-type: "do-not-specify"
+      become: "yes"
+      become-user: "cloud-user"