super-graph/core/internal/psql/psql_test.go

package psql

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"strings"
	"testing"

	"github.com/dosco/super-graph/core/internal/qcode"
)

const (
	errNotExpected = "Generated SQL did not match what was expected"
	headerMarker   = "=== RUN"
	commentMarker  = "---"
)

var (
	qcompile *qcode.Compiler
	pcompile *Compiler
	expected map[string][]string
)

func TestMain(m *testing.M) {
	var err error

	qcompile, err = qcode.NewCompiler(qcode.Config{
		Blocklist: []string{
			"secret",
			"password",
			"token",
		},
	})
	if err != nil {
		log.Fatal(err)
	}

	err = qcompile.AddRole("user", "product", qcode.TRConfig{
		Query: qcode.QueryConfig{
			Columns: []string{"id", "name", "price", "users", "customers"},
			Filters: []string{
				"{ price: { gt: 0 } }",
				"{ price: { lt: 8 } }",
			},
		},
		Insert: qcode.InsertConfig{
			Presets: map[string]string{
				"user_id":    "$user_id",
				"created_at": "now",
				"updated_at": "now",
			},
		},
		Update: qcode.UpdateConfig{
			Filters: []string{"{ user_id: { eq: $user_id } }"},
			Presets: map[string]string{"updated_at": "now"},
		},
		Delete: qcode.DeleteConfig{
			Filters: []string{
				"{ price: { gt: 0 } }",
				"{ price: { lt: 8 } }",
			},
		},
	})
	if err != nil {
		log.Fatal(err)
	}

	err = qcompile.AddRole("anon", "product", qcode.TRConfig{
		Query: qcode.QueryConfig{
			Columns: []string{"id", "name"},
		},
	})
	if err != nil {
		log.Fatal(err)
	}

	err = qcompile.AddRole("anon1", "product", qcode.TRConfig{
		Query: qcode.QueryConfig{
			Columns:          []string{"id", "name", "price"},
			DisableFunctions: true,
		},
	})
	if err != nil {
		log.Fatal(err)
	}

	err = qcompile.AddRole("user", "users", qcode.TRConfig{
		Query: qcode.QueryConfig{
			Columns: []string{"id", "full_name", "avatar", "email", "products"},
		},
	})
	if err != nil {
		log.Fatal(err)
	}

	err = qcompile.AddRole("bad_dude", "users", qcode.TRConfig{
		Query: qcode.QueryConfig{
			Filters:          []string{"false"},
			DisableFunctions: true,
		},
		Insert: qcode.InsertConfig{
			Filters: []string{"false"},
		},
		Update: qcode.UpdateConfig{
			Filters: []string{"false"},
		},
	})
	if err != nil {
		log.Fatal(err)
	}

	err = qcompile.AddRole("user", "mes", qcode.TRConfig{
		Query: qcode.QueryConfig{
			Columns: []string{"id", "full_name", "avatar"},
			Filters: []string{
				"{ id: { eq: $user_id } }",
			},
		},
	})
	if err != nil {
		log.Fatal(err)
	}

	err = qcompile.AddRole("user", "customers", qcode.TRConfig{
		Query: qcode.QueryConfig{
			Columns: []string{"id", "email", "full_name", "products"},
		},
	})

	if err != nil {
		log.Fatal(err)
	}

	schema := getTestSchema()

	vars := NewVariables(map[string]string{
		"admin_account_id": "5",
	})

	pcompile = NewCompiler(Config{
		Schema: schema,
		Vars:   vars,
	})

	expected = make(map[string][]string)

	b, err := ioutil.ReadFile("tests.sql")
	if err != nil {
		log.Fatal(err)
	}
	text := string(b)
	lines := strings.Split(text, "\n")

	var h string

	for _, v := range lines {
		switch {
		case strings.HasPrefix(v, headerMarker):
			h = strings.TrimSpace(v[len(headerMarker):])

		case strings.HasPrefix(v, commentMarker):
			break

		default:
			v := strings.TrimSpace(v)
			if len(v) != 0 {
				expected[h] = append(expected[h], v)
			}
		}
	}
	os.Exit(m.Run())
}

func compileGQLToPSQL(t *testing.T, gql string, vars Variables, role string) {
	generateTestFile := false

	if generateTestFile {
		var sqlStmts []string

		for i := 0; i < 100; i++ {
			qc, err := qcompile.Compile([]byte(gql), role)
			if err != nil {
				t.Fatal(err)
			}

			_, sqlB, err := pcompile.CompileEx(qc, vars)
			if err != nil {
				t.Fatal(err)
			}

			sql := string(sqlB)

			match := false
			for _, s := range sqlStmts {
				if sql == s {
					match = true
					break
				}
			}

			if !match {
				s := string(sql)
				sqlStmts = append(sqlStmts, s)
				fmt.Println(s)
			}
		}

		return
	}

	for i := 0; i < 200; i++ {
		qc, err := qcompile.Compile([]byte(gql), role)
		if err != nil {
			t.Fatal(err)
		}

		_, sqlStmt, err := pcompile.CompileEx(qc, vars)
		if err != nil {
			t.Fatal(err)
		}

		failed := true

		for _, sql := range expected[t.Name()] {
			if string(sqlStmt) == sql {
				failed = false
			}
		}

		if failed {
			fmt.Println(string(sqlStmt))
			t.Fatal(errNotExpected)
		}
	}
}