ProjectOctopus/agent-go/executor/FunctionalExecutor_test.go

package executor

import (
	"bytes"
	"fmt"
	"os"
	"os/exec"
	"strings"
	"testing"
)

var closeSELinux = []string{
	"sed",
	"-i",
	"s/SELINUX=enforcing/SELINUX=disabled/g",
	"/etc/selinux/config",
}

var callShellScript = []string{
	"/bin/bash",
	"/root/IdeaProjects/ProjectOctopus/agent-go/tmp/simple.sh",
}

var shutdownFirewalld = []string{
	// wrong usage of &&
	"systemctl", "stop", "firewalld", "&&", "systemctl", "disable", "firewalld",
}

var wgetCommand = []string{
	"wget",
	"--timeout=10",
	"https://oss-s1.107421.xyz/octopus_ssh_banner",
	"-O",
	"/home/wdd/IdeaProjects/ProjectOctopus/agent-go/executor/script/123",
}

var pipelineCommandFalse = []string{
	"ls",
	"/etc",
	"|",
	"grep",
	"passwd",
}

var pipelineCommand = [][]string{
	{
		"cat",
		"/home/wdd/IdeaProjects/ProjectOctopus/agent-go/executor/script/123",
	},
	{
		"grep",
		"passwd",
	},
}

var pipelineCommandMore = "apt-cache madison docker-ce | awk {print$3} | cut -d: -f2"
var pipelineCommandSecond = "systemctl status -q docker.service | grep found."
var pipelineCommandThird = "ls /etc/ | grep passwd | head -2"
var pipelineCommandFourth = "echo dsadsad | tee /home/wdd/IdeaProjects/ProjectOctopus/agent-go/executor/script/123"

var ifconfigCommand = []string{"ifconfig"}

func TestReadTimeOutput(t *testing.T) {

	ReadTimeCommandExecutor(ifconfigCommand)

}

func TestAllCommandExecutor(t *testing.T) {
	ok, result := AllCommandExecutor(wgetCommand)

	t.Logf("执行结果为 => %#v", ok)
	t.Logf("执行日志为 => %#v", result)

}

func TestPureResultSingleExecute(t *testing.T) {

	PureResultSingleExecute(closeSELinux)

}

func TestPipelineCommandExecutorSingle(t *testing.T) {

	cmd1 := exec.Command("ls", "/etc/")
	cmd2 := exec.Command("grep", "passwd")

	var outputBuf1 bytes.Buffer
	cmd1.Stdout = &outputBuf1
	if err := cmd1.Start(); err != nil {
		fmt.Printf("Error: The first command can not be startup %s\n", err)
		return
	}
	if err := cmd1.Wait(); err != nil {
		fmt.Printf("Error: Couldn't wait for the first command: %s\n", err)
		return
	}
	cmd2.Stdin = &outputBuf1
	var outputBuf2 bytes.Buffer
	cmd2.Stdout = &outputBuf2
	if err := cmd2.Start(); err != nil {
		fmt.Printf("Error: The second command can not be startup: %s\n", err)
		return
	}
	if err := cmd2.Wait(); err != nil {
		fmt.Printf("Error: Couldn't wait for the second command: %s\n", err)
		return
	}

	s := outputBuf2.String()
	split := strings.Split(s, "\n")

	BasicPrettyPrint(true, split)

}

func TestPipelineCommandExecutor(t *testing.T) {

	//PipelineCommandExecutor(pipelineCommand)
	pipelineStringToCommand := BasicTransPipelineCommand(pipelineCommandSecond)

	t.Logf("pipelineCommmand are => %#v", pipelineStringToCommand)

	ok, resultLog := PipelineCommandExecutor(pipelineStringToCommand)

	t.Logf("command execute ok is => %#v", ok)
	t.Logf("command result are 下 \n")
	BasicPrettyPrint(ok, resultLog)
}

func TestSimple(t *testing.T) {

	for i := 0; i < 10; i++ {
		var output bytes.Buffer

		fmt.Printf("output address  is => %p \n", &output)

	}
	fmt.Println()
	var output bytes.Buffer
	fmt.Printf("out is => %#v\n", output)

	command := exec.Command("ls")
	command.StdoutPipe()

}

func TestFileBasedPipe(t *testing.T) {
	reader, writer, err := os.Pipe()
	if err != nil {
		fmt.Printf("Error: Couldn't create the named pipe: %s\n", err)
	}
	go func() {

		output := make([]byte, 100)
		n, err := reader.Read(output)
		if err != nil {
			fmt.Printf("Error: Couldn't read data from the named pipe: %s\n", err)
		}
		fmt.Printf("Read %d byte(s). [file-based pipe]\n", n)

		fmt.Printf("Read content are => %s\n", string(output))

	}()

	input := make([]byte, 26)
	for i := 65; i <= 90; i++ {
		input[i-65] = byte(i)
	}
	n, err := writer.Write(input)
	if err != nil {
		fmt.Printf("Error: Couldn't write data to the named pipe: %s\n", err)
	}
	fmt.Printf("Written %d byte(s). [file-based pipe]\n", n)
	//time.Sleep(200 * time.Millisecond)

}