adrianolaselva / csvql

package csvqlctl_test

import (

	"adrianolaselva.github.io/csvql/cmd/csvqlctl"

	"bytes"

	"context"

	"fmt"

	"github.com/spf13/cobra"

	"github.com/stretchr/testify/assert"

	"os"

	"path/filepath"

	"strings"

	"testing"

)

const (

	FileModeDefault os.FileMode = 0644

)

//func TestShouldExecuteWithSuccess(t *testing.T) {

//	tests := []struct {

//		args      []string

//		filePath  string

//		fileName  string

//		data      string

//		delimiter string

//		out       string

//		queries   []string

//	}{

//		{

//			args:     []string{"-f", "./../../.tmp/0001.csv", "-d", ";"},

//			filePath: "./../../.tmp",

//			fileName: "0001.csv",

//			data: strings.Join([]string{

//				"id;name;email",

//				"0001;teste_1;[email protected]",

//				"0002;teste_2;[email protected]",

//				"0003;teste_3;[email protected]",

//				"0004;teste_4;[email protected]",

//				"0005;teste_5;[email protected]",

//			}, "\n"),

//			delimiter: ";",

//			out:       "",

//			queries: []string{

//				"select * from rows;",

//			},

//		},

//	}

//

//	ctx := context.TODO()

//	cmd, err := csvqlctl.New().Command()

//	assert.NoError(t, err)

//

//	for _, test := range tests {

//		_ = os.MkdirAll(test.filePath, os.ModePerm)

//		err := os.WriteFile(filepath.Join(test.filePath, test.fileName), bytes.NewBufferString(test.data).Bytes(), FileModeDefault)

//		assert.NoError(t, err)

//

//		//buf := new(bytes.Buffer)

//		//cmd.SetOut(buf)

//		//cmd.SetErr(buf)

//

//		//cmd.SetArgs(test.args)

//		//

//		//err = cmd.Execute()

//		//assert.NoError(t, err)

//		//

//		////cmd.SetIn(strings.NewReader("select * from rows;^D"))

//		//

//		//_, err = cmd.InOrStdin().Read([]byte("select * from rows;"))

//

//		//fmt.Printf("result: %s", buf.String())

//

//		//

//		//_, err = cmd.InOrStdin().Read([]byte("select * from rows"))

//		//assert.NoError(t, err)

//

//		err = executeCommandWithContext(ctx, cmd, test.args...)

//		assert.NoError(t, err)

//

//		//_, err = cmd.OutOrStdout().Write([]byte("select * from rows;\x0D"))

//		//assert.NoError(t, err)

//

//		//cmd.Print("select * from rows;\x0D")

//		cmd.Println("select * from rows;")

//

//		//_, err = cmd.InOrStdin().Read([]byte("select * from rows;\x0D"))

//		//assert.NoError(t, err)

//

//		//fmt.Printf("rs> out: %s\n", out)

//

//		//out := cmd.OutOrStdout()

//		//assert.NoError(t, err)

//		//

//		//c.Printf("select * from rows;")

//

//		for i := 0; i < 10; i++ {

//			if out := cmd.OutOrStdout(); out != os.Stdout {

//				fmt.Printf("out: %s\n", out)

//			}

//

//			if in := cmd.InOrStdin(); in != os.Stdin {

//				fmt.Printf("in:%s\n", in)

//			}

//		}

//

//		//fmt.Println(buf.String())

//

//		//cmd.SetIn(strings.NewReader("select * from rows;"))

//

//		//fmt.Printf("->%s\n", scanner.Text())

//

//		//cmd.InOrStdin().Read([]byte("select * from rows;"))

//		//

//		//for i := 0; i < 10; i++ {

//		//	fmt.Printf("[%s]\n", cmd.InOrStdin())

//		//	time.Sleep(1 * time.Second)

//		//}

//

//		//ctx.Done()

//

//		//cmd.SetIn(strings.NewReader("select * from rows;"))

//

//		//if out := cmd.InOrStdin(); out != os.Stdout {

//		//	fmt.Printf("%s", out)

//		//	//t.Errorf("Expected setting input to nil to revert back to stdin")

//		//}

//

//		//fmt.Printf("->out: %s\n", out)

//	}

//}

func executeCommandWithContext(ctx context.Context, root *cobra.Command, args ...string) error {

	//buf := new(bytes.Buffer)

	root.SetOut(os.Stdout)

	root.SetErr(os.Stderr)

	root.SetArgs(args)

	err := root.ExecuteContext(ctx)

	return err

}

func TestShouldExecuteWithSuccess(t *testing.T) {

	tests := []struct {

		args      []string

		filePath  string

		fileName  string

		data      string

		delimiter string

		out       string

		queries   []string

	}{

		{

			args:     []string{"-f", "./../../.tmp/0001.csv", "-d", ";"},

			filePath: "./../../.tmp",

			fileName: "0001.csv",

			data: strings.Join([]string{

				"id;name;email",

				"0001;teste_1;[email protected]",

				"0002;teste_2;[email protected]",

				"0003;teste_3;[email protected]",

				"0004;teste_4;[email protected]",

				"0005;teste_5;[email protected]",

			}, "\n"),

			delimiter: ";",

			out:       "",

			queries: []string{

				"select * from rows;",

			},

		},

	}

	//ctx := context.TODO()

	cmd, err := csvqlctl.New().Command()

	assert.NoError(t, err)

	for _, test := range tests {

		_ = os.MkdirAll(test.filePath, os.ModePerm)

		err := os.WriteFile(filepath.Join(test.filePath, test.fileName), bytes.NewBufferString(test.data).Bytes(), FileModeDefault)

		assert.NoError(t, err)

		buf := new(bytes.Buffer)

		cmd.SetOut(buf)

		cmd.SetErr(buf)

		cmd.SetArgs(test.args)

		err = cmd.Execute()

		assert.NoError(t, err)

		for _, q := range test.queries {

			_, err = cmd.OutOrStdout().Write([]byte(q))

			assert.NoError(t, err)

			if out := cmd.OutOrStdout(); out != os.Stdout {

				assert.Equal(t, q, fmt.Sprintf("%s", out))