add error test

ota42y · Dec 20, 2015 · d3ed2ca · d3ed2ca
1 parent 5cc0c8c
commit d3ed2ca
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/generate_test.go b/generate_test.go
@@ -101,6 +101,20 @@ import (
 
 }
 
+func TestGenerateError(t *testing.T) {
+	invalidData := "invalid"
+	input := strings.NewReader(invalidData)
+
+	b, err := Generate(input, "", "")
+	if err == nil {
+		t.Error("if invalid toml data, should return error, but nil")
+	}
+
+	if b != nil {
+		t.Errorf("if invalid toml data, should return nil, but %v", b)
+	}
+}
+
 func TestGenerate(t *testing.T) {
 	structName := "Normal"
 	pkgName := "example"