attr: flag test methods

attr/attr_test.go

@@ -42,3 +42,34 @@ func TestAttributeContainsTextFlags(t *testing.T) {
 		}
 	}
 }
+
+func TestAttributeTestMethods(t *testing.T) {
+	cases := []struct {
+		Attribute Attribute
+		Predicate func(Attribute) bool
+		Expect    bool
+	}{
+		// Confirm logic works as expected.
+		{DUPOK | NOSPLIT, Attribute.DUPOK, true},
+		{DUPOK | NOSPLIT, Attribute.NOSPLIT, true},
+		{DUPOK | NOSPLIT, Attribute.NOFRAME, false},
+
+		// Basic test for every method.
+		{NOPROF, Attribute.NOPROF, true},
+		{DUPOK, Attribute.DUPOK, true},
+		{NOSPLIT, Attribute.NOSPLIT, true},
+		{RODATA, Attribute.RODATA, true},
+		{NOPTR, Attribute.NOPTR, true},
+		{WRAPPER, Attribute.WRAPPER, true},
+		{NEEDCTXT, Attribute.NEEDCTXT, true},
+		{TLSBSS, Attribute.TLSBSS, true},
+		{NOFRAME, Attribute.NOFRAME, true},
+		{REFLECTMETHOD, Attribute.REFLECTMETHOD, true},
+		{TOPFRAME, Attribute.TOPFRAME, true},
+	}
+	for _, c := range cases {
+		if c.Predicate(c.Attribute) != c.Expect {
+			t.Errorf("%s: expected %#v", c.Attribute.Asm(), c.Expect)
+		}
+	}
+}

attr/make_textflag.go

@@ -197,4 +197,10 @@ func PrintFlagAttributes(w io.Writer, fs []Flag) {
 		fmt.Fprintf(w, "\t%s: %q,\n", f.Name, f.Name)
 	}
 	fmt.Fprintf(w, "}\n")
+
+	// Flag test methods.
+	for _, f := range fs {
+		fmt.Fprintf(w, "\n// %s reports whether the %s flag is set.\n", f.Name, f.Name)
+		fmt.Fprintf(w, "func (a Attribute) %s() bool { return (a & %s) != 0 }\n", f.Name, f.Name)
+	}
 }