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 30c6d3ce8be5f05e638d72ee09616c8043197a15..7e27d3fc174ac6e3762ede024601fc12c0d15efa 100644 (file)
--- 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]