| 106 | } |
| 107 | |
| 108 | func TestRender_Header(t *testing.T) { |
| 109 | r := New() |
| 110 | buf := &bytes.Buffer{} |
| 111 | |
| 112 | header := &models.Header{ |
| 113 | Package: "testpkg", |
| 114 | Imports: []*models.Import{ |
| 115 | {Path: `"testing"`}, |
| 116 | }, |
| 117 | } |
| 118 | |
| 119 | err := r.Header(buf, header) |
| 120 | if err != nil { |
| 121 | t.Errorf("Render.Header() error = %v", err) |
| 122 | } |
| 123 | |
| 124 | output := buf.String() |
| 125 | if output == "" { |
| 126 | t.Error("Render.Header() produced empty output") |
| 127 | } |
| 128 | |
| 129 | // Check that output contains package declaration |
| 130 | if !bytes.Contains([]byte(output), []byte("package testpkg")) { |
| 131 | t.Error("Render.Header() output does not contain package declaration") |
| 132 | } |
| 133 | } |
| 134 | |
| 135 | func TestRender_TestFunction(t *testing.T) { |
| 136 | r := New() |