"log"
"os"
"os/exec"
+ "strconv"
"strings"
)
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(false, false)
-}
-
-// StartKeepWithParams is used to start keep servers while specifying
-// needMore and enforcePermissions parameters.
-func StartKeepWithParams(needMore bool, 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 needMore {
- cmdArgs = append(cmdArgs, "--need-more")
- }
+ cmdArgs := []string{"run_test_server.py", "start_keep", "--num-keep-servers", strconv.Itoa(numKeepServers)}
if enforcePermissions {
cmdArgs = append(cmdArgs, "--keep-enforce-permissions")
}
}
}
-func StopKeep() {
+// 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()
- exec.Command("python", "run_test_server.py", "stop_keep").Run()
+ exec.Command("python", "run_test_server.py", "stop_keep", "--num-keep-servers", strconv.Itoa(numKeepServers))
}