Add availability zone option to launch agent.

JIRA: STORPERF-178

Change-Id: I1a5c1a0d1c3b700a0858f644b01553bf0b5675de
Signed-off-by: Taseer Ahmed <taseer94@gmail.com>

1 files changed, 13 insertions, 0 deletions

diff --git a/docker/storperf-master/storperf/storperf_master.py b/docker/storperf-master/storperf/storperf_master.py
index 866a98e..8c2a7b4 100644
--- a/docker/storperf-master/storperf/storperf_master.py
+++ b/docker/storperf-master/storperf/storperf_master.py
@@ -168,6 +168,19 @@ class StorPerfMaster(object):
             value)
 
     @property
+    def availability_zone(self):
+        return self.configuration_db.get_configuration_value(
+            'stack',
+            'availability_zone')
+
+    @availability_zone.setter
+    def availability_zone(self, value):
+        self.configuration_db.set_configuration_value(
+            'stack',
+            'availability_zone',
+            value)
+
+    @property
     def volume_quota(self):
         self._attach_to_openstack()
         quotas = self._cinder_client.quotas.get(