Add tests for bitfield enums and cleanup error printing

Author: Fabianexe

generator/generator.go

@@ -189,6 +189,7 @@ func (g *Generator) Generate(f *ast.File) ([]byte, error) {
 		// Parse the enum doc statement
 		enum, pErr := g.parseEnum(ts)
 		if pErr != nil {
+			fmt.Println(pErr)
 			continue
 		}
 
@@ -358,15 +359,13 @@ func (g *Generator) parseEnum(ts *ast.TypeSpec) (*Enum, error) {
 						newData, err := strconv.ParseUint(dataVal, 0, 64)
 						if err != nil {
 							err = fmt.Errorf("failed parsing the data part of enum value '%s': %w", value, err)
-							fmt.Println(err)
 							return nil, err
 						}
 						data = newData
 					} else {
 						newData, err := strconv.ParseInt(dataVal, 0, 64)
 						if err != nil {
 							err = fmt.Errorf("failed parsing the data part of enum value '%s': %w", value, err)
-							fmt.Println(err)
 							return nil, err
 						}
 						data = newData

generator/generator_test.go

@@ -469,6 +469,52 @@ func TestQuotedStrings(t *testing.T) {
 	}
 }
 
+func TestBitFieldOfStringFailure(t *testing.T) {
+	input := `package test
+	// This is a pre-enum comment that needs (to be handled properly)
+	// ENUM(
+	//	abc
+	//  ghi 
+	//). This is an extra string comment (With parentheses of it's own)
+	// And (another line) with Parentheses
+	type Animal string
+	`
+	g := NewGenerator(WithBitfield())
+	f, err := parser.ParseFile(g.fileSet, "TestRequiredErrors", input, parser.ParseComments)
+	assert.Nil(t, err, "Error parsing input")
+
+	enums := g.inspect(f)
+	output, err := g.parseEnum(enums["Animal"])
+	assert.ErrorContains(t, err, "bitfield option is not allowed on string types")
+	assert.Empty(t, output)
+	if false { // Debugging statement
+		fmt.Println(output)
+	}
+}
+
+func TestBitFieldManuallyValueFailure(t *testing.T) {
+	input := `package test
+	// This is a pre-enum comment that needs (to be handled properly)
+	// ENUM(
+	//	abc = 1
+	//  ghi 
+	//). This is an extra string comment (With parentheses of it's own)
+	// And (another line) with Parentheses
+	type Animal int
+	`
+	g := NewGenerator(WithBitfield())
+	f, err := parser.ParseFile(g.fileSet, "TestRequiredErrors", input, parser.ParseComments)
+	assert.Nil(t, err, "Error parsing input")
+
+	enums := g.inspect(f)
+	output, err := g.parseEnum(enums["Animal"])
+	assert.ErrorContains(t, err, "manually setting values is not allowed with the bitfield option")
+	assert.Empty(t, output)
+	if false { // Debugging statement
+		fmt.Println(output)
+	}
+}
+
 func TestStringWithSingleDoubleQuoteValue(t *testing.T) {
 	input := `package test
 	// ENUM(DoubleQuote='"')

generator/options.go

@@ -213,3 +213,10 @@ func WithNoParse() Option {
 		g.NoParse = true
 	}
 }
+
+// WithBitfield is used to create bitfield enums.
+func WithBitfield() Option {
+	return func(g *GeneratorConfig) {
+		g.BitField = true
+	}
+}