From: Kyr Shatskyy <kyrylo.shatskyy@suse.com>
Date: Wed, 12 Aug 2020 16:17:59 +0000 (+0200)
Subject: provision/downburst: add environment variables
X-Git-Tag: 1.1.0~49^2~5
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=ccf36a2d5a067e9c53d2304abdc7d5cbe030591a;p=teuthology.git

provision/downburst: add environment variables

Add downburst environment variables to make it possible
to config disk number and disk size when locking machines
locally.

Signed-off-by: Kyr Shatskyy <kyrylo.shatskyy@suse.com>
---

diff --git a/teuthology/provision/downburst.py b/teuthology/provision/downburst.py
index 30c6d3ce8..7e27d3fc1 100644
--- a/teuthology/provision/downburst.py
+++ b/teuthology/provision/downburst.py
@@ -165,16 +165,21 @@ class Downburst(object):
         os_type = self.os_type.lower()
         mac_address = self.status['mac_address']
 
+        cpus = int(os.environ.get('DOWNBURST_CPUS', 1))
+        ram_size = os.environ.get('DOWNBURST_RAM_SIZE', '3.8G')
+        disk_size = os.environ.get('DOWNBURST_DISK_SIZE', '100G')
+        extra_disk_size = os.environ.get('DOWNBURST_EXTRA_DISK_SIZE', '100G')
+        extra_disk_number = int(os.environ.get('DOWNBURST_EXTRA_DISK_NUMBER', 4))
         file_info = {
-            'disk-size': '100G',
-            'ram': '3.8G',
-            'cpus': 1,
+            'disk-size': disk_size,
+            'ram': ram_size,
+            'cpus': cpus,
             'networks': [
                 {'source': 'front', 'mac': mac_address}],
             'distro': os_type,
             'distroversion': self.os_version,
-            'additional-disks': 4,
-            'additional-disks-size': '100G',
+            'additional-disks': extra_disk_number,
+            'additional-disks-size': extra_disk_size,
             'arch': 'x86_64',
         }
         fqdn = self.name.split('@')[1]