CBPS-1154: Default to using max bucket mem quota on Capella

Change-Id: I49245d15eb574537b32b9b61ed18843fd3903158
Reviewed-on: https://review.couchbase.org/c/perfrunner/+/190918
Tested-by: Build Bot <[email protected]>
Reviewed-by: Laura Silaja <[email protected]>

Author: d-nagy

perfrunner/helpers/cluster.py

@@ -315,13 +315,23 @@ def delete_buckets(self):
 
     def create_buckets(self):
         mem_quota = self.test_config.cluster.mem_quota
+
+        if mem_quota == 0 and self.capella_infra:
+            logger.info('No memory quota provided for buckets on provisioned Capella cluster. '
+                        'Getting free memory available for buckets...')
+            mem_info = self.rest.get_bucket_mem_available(next(self.cluster_spec.masters))
+            mem_quota = mem_info['free']
+            logger.info('Free memory for buckets (per node): {}MB'.format(mem_quota))
+
         if self.test_config.cluster.num_buckets > 7:
             self.increase_bucket_limit(self.test_config.cluster.num_buckets + 3)
 
         if self.test_config.cluster.eventing_metadata_bucket_mem_quota:
             mem_quota -= (self.test_config.cluster.eventing_metadata_bucket_mem_quota +
                           self.test_config.cluster.eventing_bucket_mem_quota)
+
         per_bucket_quota = mem_quota // self.test_config.cluster.num_buckets
+
         if self.dynamic_infra:
             self.remote.delete_all_buckets()
             for bucket_name in self.test_config.buckets:

perfrunner/helpers/rest.py

@@ -2102,6 +2102,14 @@ def create_db_user(self, cluster_id: str, username: str, password: str):
                                                     username, password)
         return resp
 
+    def get_bucket_mem_available(self, host: str):
+        cluster_id = self.hostname_to_cluster_id(host)
+        resp = self.dedicated_client.get_buckets(self.tenant_id, self.project_id, cluster_id)
+        return {
+            'free': resp.json()['freeMemoryInMb'],
+            'total': resp.json()['totalMemoryInMb']
+        }
+
     @retry
     def create_bucket(self, host: str,
                       name: str,