Chore/migrate tests

go.mod

@@ -5,8 +5,11 @@ go 1.22.1
 require (
 	github.com/antlr4-go/antlr/v4 v4.13.1
 	github.com/gkampitakis/go-snaps v0.5.4
+	github.com/sergi/go-diff v1.0.0
 )
 
+require gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 // indirect
+
 require (
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect

go.sum

@@ -35,6 +35,8 @@ github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/f
 github.com/rogpeppe/go-internal v1.12.0 h1:exVL4IDcn6na9z1rAb56Vxr+CgyK3nn3O+epU5NdKM8=
 github.com/rogpeppe/go-internal v1.12.0/go.mod h1:E+RYuTGaKKdloAfM02xzb0FW3Paa99yedzYV+kq4uf4=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/sergi/go-diff v1.0.0 h1:Kpca3qRNrduNnOQeazBd0ysaKrUJiIuISHxogkT9RPQ=
+github.com/sergi/go-diff v1.0.0/go.mod h1:0CfEIISq7TuYL3j771MWULgwwjU+GofnZX9QAmXWZgo=
 github.com/spf13/cobra v1.8.1 h1:e5/vxKd/rZsfSJMUX1agtjeTDf+qv1/JdBF8gg5k9ZM=
 github.com/spf13/cobra v1.8.1/go.mod h1:wHxEcudfqmLYa8iTfL+OuZPbBZkmvliBWKIezN3kD9Y=
 github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
@@ -57,7 +59,8 @@ golang.org/x/sys v0.18.0 h1:DBdB3niSjOA/O0blCZBqDefyWNYveAYMNF1Wum0DYQ4=
 golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
 golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
-gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

internal/ansi/ansi.go

@@ -1,22 +1,95 @@
 package ansi
 
-import "fmt"
+import (
+	"fmt"
+	"strings"
+)
 
 const resetCol = "\033[0m"
 
+func Compose(cols ...func(string) string) func(string) string {
+	return func(s string) string {
+		for _, mod := range cols {
+			s = mod(s)
+		}
+		return s
+	}
+}
+
+func replaceLast(s, oldStr, newStr string) string {
+	lastIndex := strings.LastIndex(s, oldStr)
+	if lastIndex == -1 {
+		return s
+	}
+	return s[:lastIndex] + newStr + s[lastIndex+len(oldStr):]
+}
+
 func col(s string, code int) string {
-	c := fmt.Sprintf("\033[%dm", code)
-	return c + s + resetCol
+	colorCode := fmt.Sprintf("\033[%dm", code)
+	// This trick should allow to stack colors (TODO test)
+	s = replaceLast(s, resetCol, resetCol+colorCode)
+	return colorCode + s + resetCol
 }
 
 func ColorRed(s string) string {
 	return col(s, 31)
 }
 
+func ColorWhite(s string) string {
+	return col(s, 37)
+}
+
+func ColorGreen(s string) string {
+	return col(s, 32)
+}
+
 func ColorYellow(s string) string {
 	return col(s, 33)
 }
 
 func ColorCyan(s string) string {
 	return col(s, 36)
 }
+
+func ColorLight(s string) string {
+	return col(s, 97) // Bright white → light
+}
+
+func ColorBrightBlack(s string) string {
+	return col(s, 90)
+}
+
+func ColorBrightRed(s string) string {
+	return col(s, 91)
+}
+
+func ColorBrightGreen(s string) string {
+	return col(s, 92)
+}
+
+func ColorBrightYellow(s string) string {
+	return col(s, 93)
+}
+
+// BG
+func BgDark(s string) string {
+	return col(s, 100)
+}
+
+func BgRed(s string) string {
+	return col(s, 41)
+}
+
+func BgGreen(s string) string {
+	return col(s, 42)
+}
+
+// modifiers
+
+func Bold(s string) string {
+	return col(s, 1)
+}
+
+func Underline(s string) string {
+	return col(s, 4)
+}

internal/cmd/root.go

@@ -25,6 +25,7 @@ func Execute(options CliOptions) {
 
 	rootCmd.AddCommand(lspCmd)
 	rootCmd.AddCommand(checkCmd)
+	rootCmd.AddCommand(getTestCmd())
 	rootCmd.AddCommand(getRunCmd())
 
 	if err := rootCmd.Execute(); err != nil {

internal/cmd/run.go

@@ -8,7 +8,6 @@ import (
 	"os"
 	"strings"
 
-	"github.com/formancehq/numscript/internal/ansi"
 	"github.com/formancehq/numscript/internal/flags"
 	"github.com/formancehq/numscript/internal/interpreter"
 	"github.com/formancehq/numscript/internal/parser"
@@ -21,7 +20,7 @@ const (
 	OutputFormatJson   = "json"
 )
 
-type Args struct {
+type runArgs struct {
 	VariablesOpt string
 	BalancesOpt  string
 	MetaOpt      string
@@ -38,7 +37,7 @@ type inputOpts struct {
 	Balances  interpreter.Balances         `json:"balances"`
 }
 
-func (o *inputOpts) fromRaw(opts Args) {
+func (o *inputOpts) fromRaw(opts runArgs) {
 	if opts.RawOpt == "" {
 		return
 	}
@@ -49,7 +48,7 @@ func (o *inputOpts) fromRaw(opts Args) {
 	}
 }
 
-func (o *inputOpts) fromStdin(opts Args) {
+func (o *inputOpts) fromStdin(opts runArgs) {
 	if !opts.StdinFlag {
 		return
 	}
@@ -65,7 +64,7 @@ func (o *inputOpts) fromStdin(opts Args) {
 	}
 }
 
-func (o *inputOpts) fromOptions(path string, opts Args) {
+func (o *inputOpts) fromOptions(path string, opts runArgs) {
 	if path != "" {
 		numscriptContent, err := os.ReadFile(path)
 		if err != nil {
@@ -103,7 +102,7 @@ func (o *inputOpts) fromOptions(path string, opts Args) {
 	}
 }
 
-func run(path string, opts Args) {
+func run(path string, opts runArgs) {
 	opt := inputOpts{
 		Variables: make(map[string]string),
 		Meta:      make(interpreter.AccountsMetadata),
@@ -164,25 +163,19 @@ func showJson(result *interpreter.ExecutionResult) {
 }
 
 func showPretty(result *interpreter.ExecutionResult) {
-	fmt.Println(ansi.ColorCyan("Postings:"))
-	postingsJson, err := json.MarshalIndent(result.Postings, "", "  ")
-	if err != nil {
-		panic(err)
-	}
-	fmt.Println(string(postingsJson))
 
-	fmt.Println()
+	fmt.Println("Postings:")
+	fmt.Println(interpreter.PrettyPrintPostings(result.Postings))
 
-	fmt.Println(ansi.ColorCyan("Meta:"))
-	txMetaJson, err := json.MarshalIndent(result.Metadata, "", "  ")
-	if err != nil {
-		panic(err)
+	if len(result.Metadata) != 0 {
+		fmt.Println("Meta:")
+		fmt.Println(interpreter.PrettyPrintMeta(result.Metadata))
 	}
-	fmt.Println(string(txMetaJson))
+
 }
 
 func getRunCmd() *cobra.Command {
-	opts := Args{}
+	opts := runArgs{}
 
 	cmd := cobra.Command{
 		Use:   "run",