package psql import ( "fmt" "io/ioutil" "log" "os" "strings" "testing" "github.com/dosco/super-graph/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) } } }