gpython: add a simple test for gpython command

sbinet · sbinet · commit 01cbebd2719c · 2022-02-12T19:39:37.000+01:00
Signed-off-by: Sebastien Binet &lt;binet@cern.ch&gt;
diff --git a/main.go b/main.go
@@ -41,10 +41,12 @@ Full options:
 func main() {
 	flag.Usage = syntaxError
 	flag.Parse()
-	args := flag.Args()
+	xmain(flag.Args())
+}
 
+func xmain(args []string) {
 	opts := py.DefaultContextOpts()
-	opts.SysArgs = flag.Args()
+	opts.SysArgs = args
 	ctx := py.NewContext(opts)
 
 	if *cpuprofile != "" {
@@ -77,5 +79,4 @@ func main() {
 			log.Fatal(err)
 		}
 	}
-
 }
diff --git a/main_test.go b/main_test.go
@@ -0,0 +1,61 @@
+// Copyright 2022 The go-python Authors.  All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package main
+
+import (
+	"bytes"
+	"flag"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"testing"
+)
+
+var regen = flag.Bool("regen", false, "regenerate golden files")
+
+func TestGPython(t *testing.T) {
+
+	tmp, err := os.MkdirTemp("", "go-python-")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(tmp)
+
+	exe := filepath.Join(tmp, "out.exe")
+	cmd := exec.Command("go", "build", "-o", exe, ".")
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	err = cmd.Run()
+	if err != nil {
+		t.Fatalf("failed to compile embedding example: %+v", err)
+	}
+
+	got, err := exec.Command(exe, "testdata/hello.py").CombinedOutput()
+	if err != nil {
+		t.Fatalf("could not run gpython:\n%s\nerr: %+v", got, err)
+	}
+
+	const fname = "testdata/hello_golden.txt"
+
+	flag.Parse()
+	if *regen {
+		err = os.WriteFile(fname, got, 0644)
+		if err != nil {
+			t.Fatalf("could not write golden file: %+v", err)
+		}
+	}
+
+	want, err := os.ReadFile(fname)
+	if err != nil {
+		t.Fatalf("could not read golden file: %+v", err)
+	}
+	if !bytes.Equal(got, want) {
+		t.Fatalf("stdout differ:\ngot:\n%s\nwant:\n%s\n", got, want)
+	}
+}
+
+func TestRunFile(t *testing.T) {
+	xmain([]string{"./testdata/hello.py"})
+}
diff --git a/testdata/hello.py b/testdata/hello.py
@@ -0,0 +1 @@
+print("hello, world!")
diff --git a/testdata/hello_golden.txt b/testdata/hello_golden.txt
@@ -0,0 +1 @@
+hello, world!

Original file line number	Diff line number	Diff line change
`@@ -41,10 +41,12 @@ Full options:`
`41`	`41`	`func main() {`
`42`	`42`	`flag.Usage = syntaxError`
`43`	`43`	`flag.Parse()`
`44`		`- args := flag.Args()`
	`44`	`+ xmain(flag.Args())`
	`45`	`+}`
`45`	`46`
	`47`	`+func xmain(args []string) {`
`46`	`48`	`opts := py.DefaultContextOpts()`
`47`		`- opts.SysArgs = flag.Args()`
	`49`	`+ opts.SysArgs = args`
`48`	`50`	`ctx := py.NewContext(opts)`
`49`	`51`
`50`	`52`	`if *cpuprofile != "" {`
`@@ -77,5 +79,4 @@ func main() {`
`77`	`79`	`log.Fatal(err)`
`78`	`80`	`}`
`79`	`81`	`}`
`80`		`-`
`81`	`82`	`}`