Restore --format table header support

Signed-off-by: Jhon Honce <jhonce@redhat.com>

1 files changed, 30 insertions, 0 deletions

diff --git a/cmd/podman/parse/template_test.go b/cmd/podman/parse/template_test.go
new file mode 100644
index 000000000..7880d9bec
--- /dev/null
+++ b/cmd/podman/parse/template_test.go
@@ -0,0 +1,30 @@
+package parse
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestEnforceRange(t *testing.T) {
+	tests := []struct {
+		input    string
+		expected string
+	}{
+		{"{{range .}}{{.ID}}{{end}}", "{{range .}}{{.ID}}{{end}}"},
+		{"{{.ID}}", "{{range .}}{{.ID}}{{end}}"},
+		{"{{ range . }}{{ .ID }}{{ end }}", "{{ range . }}{{ .ID }}{{ end }}"},
+		// EnforceRange does not verify syntax or semantics, that will happen later
+		{"{{range .}}{{.ID}}", "{{range .}}{{range .}}{{.ID}}{{end}}"},
+		{".ID", "{{range .}}.ID{{end}}"},
+	}
+
+	for _, tc := range tests {
+		tc := tc
+		label := "TestEnforceRange_" + tc.input
+		t.Run(label, func(t *testing.T) {
+			t.Parallel()
+			assert.Equal(t, tc.expected, EnforceRange(tc.input))
+		})
+	}
+}