Update buf rulegen

aaliddell · aaliddell · commit 77881e12b96a · 2024-01-02T11:28:49.000Z
diff --git a/tools/rulegen/buf.go b/tools/rulegen/buf.go
@@ -7,6 +7,7 @@ load(
 )
 load(
     ":buf.bzl",
+    "TEST_ATTRS",
     "{{ .Rule.Name }}_script_impl",
 )
 
@@ -37,12 +38,13 @@ load(
 def {{ .Rule.Name }}(name, **kwargs):
     {{ .Rule.Name }}_script(
         name = name + ".sh",
-        **kwargs
+        **{k: v for k, v in kwargs.items() if k not in TEST_ATTRS}
     )
 
     native.sh_test(
         name = name,
         srcs = [name + ".sh"],
+        **{k: v for k, v in kwargs.items() if k in TEST_ATTRS}
     )`)
 
 var bufBreakingExampleTemplate = mustTemplate(`load("@rules_proto_grpc_{{ .Lang.Name }}//:defs.bzl", "{{ .Rule.Name }}")

Original file line number	Diff line number	Diff line change
`@@ -7,6 +7,7 @@ load(`
`7`	`7`	`)`
`8`	`8`	`load(`
`9`	`9`	`":buf.bzl",`
	`10`	`+ "TEST_ATTRS",`
`10`	`11`	`"{{ .Rule.Name }}_script_impl",`
`11`	`12`	`)`
`12`	`13`
`@@ -37,12 +38,13 @@ load(`
`37`	`38`	`def {{ .Rule.Name }}(name, **kwargs):`
`38`	`39`	`{{ .Rule.Name }}_script(`
`39`	`40`	`name = name + ".sh",`
`40`		`- **kwargs`
	`41`	`+ **{k: v for k, v in kwargs.items() if k not in TEST_ATTRS}`
`41`	`42`	`)`
`42`	`43`
`43`	`44`	`native.sh_test(`
`44`	`45`	`name = name,`
`45`	`46`	`srcs = [name + ".sh"],`
	`47`	`+ **{k: v for k, v in kwargs.items() if k in TEST_ATTRS}`
`46`	`48`	)`)
`47`	`49`
`48`	`50`	var bufBreakingExampleTemplate = mustTemplate(`load("@rules_proto_grpc_{{ .Lang.Name }}//:defs.bzl", "{{ .Rule.Name }}")