Function TestCgroup

worker/cgroup_test.go:89–328

(t *testing.T)

}

func TestCgroup(t *testing.T) {
	var cgcf *cgroupConfig
	Convey("init cgroup", t, func(ctx C) {
		_, useCurrentCgroup := os.LookupEnv("USECURCGROUP")
		cgcf = &cgroupConfig{BasePath: "/sys/fs/cgroup", Group: "tunasync", Subsystem: "cpu"}
		if useCurrentCgroup {
			cgcf.Group = ""
		}
		err := initCgroup(cgcf)
		So(err, ShouldBeNil)
		if cgcf.isUnified {
			So(cgcf.cgMgrV2, ShouldNotBeNil)
		} else {
			So(cgcf.cgMgrV1, ShouldNotBeNil)
		}

		Convey("Cgroup Should Work", func(ctx C) {
			tmpDir, err := os.MkdirTemp("", "tunasync")
			defer os.RemoveAll(tmpDir)
			So(err, ShouldBeNil)
			cmdScript := filepath.Join(tmpDir, "cmd.sh")
			daemonScript := filepath.Join(tmpDir, "daemon.sh")
			tmpFile := filepath.Join(tmpDir, "log_file")
			bgPidfile := filepath.Join(tmpDir, "bg.pid")

			c := cmdConfig{
				name:        "tuna-cgroup",
				upstreamURL: "http://mirrors.tuna.moe/",
				command:     cmdScript + " " + daemonScript,
				workingDir:  tmpDir,
				logDir:      tmpDir,
				logFile:     tmpFile,
				interval:    600 * time.Second,
				env: map[string]string{
					"BG_PIDFILE": bgPidfile,
				},
			}
			cmdScriptContent := `#!/bin/bash
redirect-std() {
    [[ -t 0 ]] && exec </dev/null
    [[ -t 1 ]] && exec >/dev/null
    [[ -t 2 ]] && exec 2>/dev/null
}

# close all non-std* fds
close-fds() {
    eval exec {3..255}\>\&-
}

# full daemonization of external command with setsid
daemonize() {
    (
        redirect-std
        cd /
        close-fds
        exec setsid "$@"
    ) &
}