llgo cmptest (#391)

internal/build/cmptest.go

@@ -0,0 +1,78 @@
+/*
+ * Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package build
+
+import (
+	"bytes"
+	"errors"
+	"fmt"
+	"io"
+	"log"
+	"os"
+	"os/exec"
+)
+
+func cmpTest(dir, pkgPath, llApp string, runArgs []string) {
+	var goOut, goErr bytes.Buffer
+	var llgoOut, llgoErr bytes.Buffer
+	var llgoRunErr = runApp(runArgs, dir, &llgoOut, &llgoErr, llApp)
+	var goRunErr = runApp(runArgs, dir, &goOut, &goErr, "go", "run", pkgPath)
+	checkEqual("output", llgoOut.Bytes(), goOut.Bytes())
+	checkEqual("stderr", llgoErr.Bytes(), goErr.Bytes())
+	checkEqualRunErr(llgoRunErr, goRunErr)
+}
+
+func checkEqualRunErr(llgoRunErr, goRunErr error) {
+	if llgoRunErr == goRunErr {
+		return
+	}
+	fmt.Fprintln(os.Stderr, "=> Exit:", llgoRunErr)
+	fmt.Fprintln(os.Stderr, "\n=> Expected Exit:", goRunErr)
+}
+
+func checkEqual(prompt string, a, expected []byte) {
+	if bytes.Equal(a, expected) {
+		return
+	}
+
+	fmt.Fprintln(os.Stderr, "=> Result of", prompt)
+	os.Stderr.Write(a)
+
+	fmt.Fprintln(os.Stderr, "\n=> Expected", prompt)
+	os.Stderr.Write(expected)
+
+	fatal(errors.New("checkEqual: unexpected " + prompt))
+}
+
+func runApp(runArgs []string, dir string, stdout, stderr io.Writer, app string, args ...string) error {
+	if len(runArgs) > 0 {
+		if len(args) > 0 {
+			args = append(args, runArgs...)
+		} else {
+			args = runArgs
+		}
+	}
+	cmd := exec.Command(app, args...)
+	cmd.Dir = dir
+	cmd.Stdout = stdout
+	cmd.Stderr = stderr
+	return cmd.Run()
+}
+
+func fatal(err error) {
+	log.Panicln(err)
+}