diff --git a/.github/workflows/tunasync.yml b/.github/workflows/tunasync.yml index 39f6079..aaa3487 100644 --- a/.github/workflows/tunasync.yml +++ b/.github/workflows/tunasync.yml @@ -36,10 +36,9 @@ jobs: - name: Setup test dependencies run: | - sudo apt update - sudo apt install -y cgroup-bin - /usr/bin/docker pull alpine - /usr/bin/docker images + sudo apt-get update + sudo apt-get install -y cgroup-bin + docker pull alpine:3.8 lssubsys -am sudo cgcreate -a $USER -t $USER -g cpu:tunasync sudo cgcreate -a $USER -t $USER -g memory:tunasync diff --git a/worker/docker_test.go b/worker/docker_test.go new file mode 100644 index 0000000..e556967 --- /dev/null +++ b/worker/docker_test.go @@ -0,0 +1,127 @@ +package worker + +import ( + "fmt" + "io/ioutil" + "os" + "os/exec" + "path/filepath" + "testing" + "time" + + "github.com/codeskyblue/go-sh" + . "github.com/smartystreets/goconvey/convey" +) + +func cmdRun(p string, args []string) { + cmd := exec.Command(p, args...) + + out, err := cmd.CombinedOutput() + if err != nil { + logger.Debugf("cmdRun failed %s", err) + return + } + logger.Debugf("cmdRun: ", string(out)) +} + +func getDockerByName(name string) (string, error) { + // docker ps -f 'name=$name' --format '{{.Names}}' + out, err := sh.Command( + "docker", "ps", "-a", + "--filter", "name="+name, + "--format", "{{.Names}}", + ).Output() + if err == nil { + logger.Debugf("docker ps: '%s'", string(out)) + } + return string(out), err +} + +func TestDocker(t *testing.T) { + Convey("Docker Should Work", t, func(ctx C) { + tmpDir, err := ioutil.TempDir("", "tunasync") + defer os.RemoveAll(tmpDir) + So(err, ShouldBeNil) + cmdScript := filepath.Join(tmpDir, "cmd.sh") + tmpFile := filepath.Join(tmpDir, "log_file") + expectedOutput := "HELLO_WORLD" + + c := cmdConfig{ + name: "tuna-docker", + upstreamURL: "http://mirrors.tuna.moe/", + command: "/bin/cmd.sh", + workingDir: tmpDir, + logDir: tmpDir, + logFile: tmpFile, + interval: 600 * time.Second, + env: map[string]string{ + "TEST_CONTENT": expectedOutput, + }, + } + + cmdScriptContent := `#!/bin/sh +echo ${TEST_CONTENT} +sleep 20 +` + err = ioutil.WriteFile(cmdScript, []byte(cmdScriptContent), 0755) + So(err, ShouldBeNil) + + provider, err := newCmdProvider(c) + So(err, ShouldBeNil) + + d := &dockerHook{ + emptyHook: emptyHook{ + provider: provider, + }, + image: "alpine:3.8", + volumes: []string{ + fmt.Sprintf("%s:%s", cmdScript, "/bin/cmd.sh"), + }, + } + provider.AddHook(d) + So(provider.Docker(), ShouldNotBeNil) + + err = d.preExec() + So(err, ShouldBeNil) + + cmdRun("docker", []string{"images"}) + exitedErr := make(chan error, 1) + go func() { + err = provider.Run() + logger.Debugf("provider.Run() exited") + if err != nil { + logger.Errorf("provider.Run() failed: %v", err) + } + exitedErr <- err + }() + cmdRun("ps", []string{"aux"}) + + // Wait for docker running + time.Sleep(8 * time.Second) + + cmdRun("ps", []string{"aux"}) + + // assert container running + names, err := getDockerByName(d.Name()) + So(err, ShouldBeNil) + // So(names, ShouldEqual, d.Name()+"\n") + + err = provider.Terminate() + // So(err, ShouldBeNil) + + cmdRun("ps", []string{"aux"}) + <-exitedErr + + // container should be terminated and removed + names, err = getDockerByName(d.Name()) + So(err, ShouldBeNil) + So(names, ShouldEqual, "") + + // check log content + loggedContent, err := ioutil.ReadFile(provider.LogFile()) + So(err, ShouldBeNil) + So(string(loggedContent), ShouldEqual, expectedOutput+"\n") + + d.postExec() + }) +} diff --git a/worker/runner.go b/worker/runner.go index e47417e..144f7e1 100644 --- a/worker/runner.go +++ b/worker/runner.go @@ -104,7 +104,7 @@ func newCmdJob(provider mirrorProvider, cmdAndArgs []string, workingDir string, } func (c *cmdJob) Start() error { - // logger.Debugf("Command start: %v", c.cmd.Args) + logger.Debugf("Command start: %v", c.cmd.Args) c.finished = make(chan empty, 1) return c.cmd.Start() }