From 2fae9b0a69465c3365c8fdeac7034b68edd89e0c Mon Sep 17 00:00:00 2001 From: Mathias Fredriksson Date: Tue, 18 Jul 2023 20:26:03 +0300 Subject: [PATCH] chore(scripts/dbgen): only add arg validation for dbfake (#8578) --- scripts/dbgen/main.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/dbgen/main.go b/scripts/dbgen/main.go index 0eeda09ba9..8a7976e321 100644 --- a/scripts/dbgen/main.go +++ b/scripts/dbgen/main.go @@ -220,7 +220,9 @@ func orderAndStubDatabaseFunctions(filePath, receiver, structName string, stub f for _, fn := range funcs { var bodyStmts []dst.Stmt - if len(fn.Func.Params.List) == 2 && fn.Func.Params.List[1].Names[0].Name == "arg" { + + // Add input validation, only relevant for dbfake. + if strings.Contains(filePath, "dbfake") && len(fn.Func.Params.List) == 2 && fn.Func.Params.List[1].Names[0].Name == "arg" { /* err := validateDatabaseType(arg) if err != nil {