Allow configuring worker count in Trino

PiperOrigin-RevId: 846453023

Author: hubatish

perfkitbenchmarker/providers/gcp/trino.py

@@ -46,10 +46,18 @@ def _Create(self):
         'trino',
         'https://trinodb.github.io/charts',
     ]
+    yaml_config = {
+        'server': {
+            'workers': self.node_count,
+        }
+    }
+    filename = vm_util.WriteYaml([yaml_config])
     vm_util.IssueCommand(cmd)
     cmd = [
         'helm',
         'install',
+        '-f',
+        filename,
         self.name,
         'trino/trino',
         '--kubeconfig',

tests/providers/gcp/trino_test.py

@@ -3,6 +3,7 @@
 from absl import flags
 from absl.testing import flagsaver
 import mock
+from perfkitbenchmarker import vm_util
 from perfkitbenchmarker.providers.gcp import trino
 from tests import pkb_common_test_case
 
@@ -35,6 +36,13 @@ def testBuildAndCompile(self):
     self.assertEqual(db.name, 'pkb-123')
 
   def testCreate(self):
+    self.enter_context(
+        mock.patch.object(
+            vm_util,
+            'WriteTemporaryFile',
+            return_value='trino.yaml',
+        )
+    )
     db = trino.Trino(EDW_SERVICE_SPEC)
     mock_cmd = self.MockIssueCommand({'': [('', '', 0)]})
     db._Create()
@@ -49,6 +57,8 @@ def testCreate(self):
         mock.call([
             'helm',
             'install',
+            '-f',
+            'trino.yaml',
             'pkb-123',
             'trino/trino',
             '--kubeconfig',