Add TestCreateFileFailures for more coverage

dolmen-go · Mar 28, 2024 · 35c9aa3 · 35c9aa3
1 parent 001f737
commit 35c9aa3
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/codegen_test.go b/codegen_test.go
@@ -3,6 +3,7 @@ package codegen_test
 import (
 	"log"
 	"os"
+	"strings"
 	"testing"
 
 	"github.com/dolmen-go/codegen"
@@ -46,3 +47,10 @@ func TestParseFailures(t *testing.T) {
 		t.Fatal("MustParse should panic.")
 	}
 }
+
+func TestCreateFileFailures(t *testing.T) {
+	err := codegen.CreateFile("tmp_test.go", codegen.MustParse("package main_test\n").Template, 0)
+	if err == nil || !strings.Contains(err.Error(), "https://golang.org/s/generatedcode") {
+		t.Fatal("Error expected")
+	}
+}