public/tests/go/cat_test.go

package student_test

import (
	"errors"
	"io/ioutil"
	"log"
	"os"
	"os/exec"
	"strings"
	"testing"

	solutions "./solutions"

	"github.com/01-edu/z01"
)

//executes commands
func execC(name string, args ...string) (string, error) {
	out, err := exec.Command(name, args...).Output()

	output := string(out)
	if err == nil {
		return output, nil
	}
	if output == "" {
		return "", z01.Wrap(err, "Command failed")
	}
	return "", errors.New(output)
}

func TestCat(t *testing.T) {
	var table []string
	pathFileName := "./student/cat/quest8.txt"
	pathFileName2 := "./student/cat/quest8T.txt"

	solutions.CheckFile(t, pathFileName)
	solutions.CheckFile(t, pathFileName2)

	table = append(table, pathFileName, pathFileName+" "+pathFileName2, "asd")

	for _, s := range table {
		z01.ChallengeMain(t, strings.Fields(s)...)
	}
	_, err := execC("go", "build", "-o", "cat_student", "./student/cat/main.go")
	_, err = execC("go", "build", "-o", "cat_solution", "./solutions/cat/main.go")
	if err != nil {
		log.Fatal(err.Error())
	}
	pwd, err := os.Getwd()

	if err != nil {
		t.Fatalf(err)
	}

	for i := 0; i < 2; i++ {
		randStdin := z01.RandAlnum()
		cmd := exec.Command("sh", "-c", pwd+"/cat_solution")
		solutionResult := execStdin(cmd, randStdin)
		cmdS := exec.Command(pwd + "/cat_student")
		studentResult := execStdin(cmdS, randStdin)

		if solutionResult != studentResult {
			t.Fatalf("./cat prints %s instead of %s\n", studentResult, solutionResult)
		}
	}
	execC("rm", "cat_student", "cat_solution")
}

func execStdin(cmd *exec.Cmd, randomStdin string) string {
	stdin, err := cmd.StdinPipe()
	if err != nil {
		log.Fatal(err)
	}
	stdout, err := cmd.StdoutPipe()
	if err != nil {
		log.Fatal(err)
	}
	if err := cmd.Start(); err != nil {
		log.Fatal(err)
	}
	_, err = stdin.Write([]byte(randomStdin))
	if err != nil {
		log.Fatal(err)
	}
	stdin.Close()

	out, _ := ioutil.ReadAll(stdout)
	if err := cmd.Wait(); err != nil {
		log.Fatal(err)
	}
	return string(out)
}