X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/07b382c82d7d834e801cf9dc85e2ed5ffcd7cd91..f0ea4324260fb4dc6df693d9548285bb64b3b69f:/sdk/go/arvadostest/run_servers.go

diff --git a/sdk/go/arvadostest/run_servers.go b/sdk/go/arvadostest/run_servers.go
index 68153b8b51..27c552a4e1 100644
--- a/sdk/go/arvadostest/run_servers.go
+++ b/sdk/go/arvadostest/run_servers.go
@@ -99,23 +99,15 @@ func StopAPI() {
 	exec.Command("python", "run_test_server.py", "stop").Run()
 }
 
-// StartKeep is used to start keep servers
-// with needMore = false and enforcePermissions = false
-func StartKeep() {
-	StartKeepWithParams(2, false)
-}
-
-// StartKeepWithParams is used to start keep servers while specifying
-// numKeepServers and enforcePermissions parameters.
-func StartKeepWithParams(numKeepServers int, enforcePermissions bool) {
+// StartKeep starts the given number of keep servers,
+// optionally with -enforce-permissions enabled.
+// Use numKeepServers = 2 and enforcePermissions = false under all normal circumstances.
+func StartKeep(numKeepServers int, enforcePermissions bool) {
 	cwd, _ := os.Getwd()
 	defer os.Chdir(cwd)
 	chdirToPythonTests()
 
-	cmdArgs := []string{"run_test_server.py", "start_keep"}
-	if numKeepServers != 2 {
-		cmdArgs = append(cmdArgs, "--num-keep-servers", strconv.Itoa(numKeepServers))
-	}
+	cmdArgs := []string{"run_test_server.py", "start_keep", "--num-keep-servers", strconv.Itoa(numKeepServers)}
 	if enforcePermissions {
 		cmdArgs = append(cmdArgs, "--keep-enforce-permissions")
 	}
@@ -132,21 +124,13 @@ func StartKeepWithParams(numKeepServers int, enforcePermissions bool) {
 	}
 }
 
-func StopKeep() {
-	StopKeepServers(2)
-}
-
-// StopKeepServers is used to stop keep servers while specifying numKeepServers
-func StopKeepServers(numKeepServers int) {
+// StopKeep stops keep servers that were started with StartKeep.
+// numkeepServers should be the same value that was passed to StartKeep,
+// which is 2 under all normal circumstances.
+func StopKeep(numKeepServers int) {
 	cwd, _ := os.Getwd()
 	defer os.Chdir(cwd)
 	chdirToPythonTests()
 
-	cmdArgs := []string{"run_test_server.py", "stop_keep"}
-
-	if numKeepServers != 2 {
-		cmdArgs = append(cmdArgs, "--num-keep-servers", strconv.Itoa(numKeepServers))
-	}
-
-	exec.Command("python", cmdArgs...)
+	exec.Command("python", "run_test_server.py", "stop_keep", "--num-keep-servers", strconv.Itoa(numKeepServers))
 }