generator: verified

pkg/generator/generator_test.go

@@ -1,11 +1,31 @@
 package generator_test
 
 import (
+	"fmt"
 	"testing"
 
 	"github.com/iliadenisov/galaxy/pkg/generator"
 )
 
-func Test_Generator(t *testing.T) {
-	generator.Generate()
+func TestGenerator(t *testing.T) {
+	for players := 10; players <= 50; players++ {
+		_, err := generator.Generate(func(ms *generator.MapSetting) { ms.Players = uint32(players) })
+		if err != nil {
+			t.Errorf("generate: %s", err)
+			break
+		}
+	}
+}
+
+func BenchmarkGenerator(b *testing.B) {
+	i := 0
+	for b.Loop() {
+		i++
+		b.Run(fmt.Sprintf("instance #%02d", i), func(b *testing.B) {
+			_, err := generator.Generate()
+			if err != nil {
+				b.Error(err)
+			}
+		})
+	}
 }