From 1a153b8e80eaec71727c80be54ba613a8425d408 Mon Sep 17 00:00:00 2001
From: Brad Rydzewski <brad.rydzewski@gmail.com>
Date: Thu, 13 Mar 2014 17:07:33 -0700
Subject: [PATCH] test coverage for dockerfile package

---
 pkg/build/dockerfile/dockerfile_test.go | 63 +++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 pkg/build/dockerfile/dockerfile_test.go

diff --git a/pkg/build/dockerfile/dockerfile_test.go b/pkg/build/dockerfile/dockerfile_test.go
new file mode 100644
index 000000000..fb1b017d5
--- /dev/null
+++ b/pkg/build/dockerfile/dockerfile_test.go
@@ -0,0 +1,63 @@
+package dockerfile
+
+import (
+	"testing"
+)
+
+func TestWrite(t *testing.T) {
+
+	var f = New("ubuntu")
+	var got, want = f.String(), "FROM ubuntu\n"
+	if got != want {
+		t.Errorf("Exepected New() returned %s, got %s", want, got)
+	}
+
+	f = &Dockerfile{}
+	f.WriteAdd("src", "target")
+	got, want = f.String(), "ADD src target\n"
+	if got != want {
+		t.Errorf("Exepected WriteAdd returned %s, got %s", want, got)
+	}
+
+	f = &Dockerfile{}
+	f.WriteFrom("ubuntu")
+	got, want = f.String(), "FROM ubuntu\n"
+	if got != want {
+		t.Errorf("Exepected WriteFrom returned %s, got %s", want, got)
+	}
+
+	f = &Dockerfile{}
+	f.WriteRun("whoami")
+	got, want = f.String(), "RUN whoami\n"
+	if got != want {
+		t.Errorf("Exepected WriteRun returned %s, got %s", want, got)
+	}
+
+	f = &Dockerfile{}
+	f.WriteUser("root")
+	got, want = f.String(), "USER root\n"
+	if got != want {
+		t.Errorf("Exepected WriteUser returned %s, got %s", want, got)
+	}
+
+	f = &Dockerfile{}
+	f.WriteEnv("FOO", "BAR")
+	got, want = f.String(), "ENV FOO BAR\n"
+	if got != want {
+		t.Errorf("Exepected WriteEnv returned %s, got %s", want, got)
+	}
+
+	f = &Dockerfile{}
+	f.WriteWorkdir("/home/ubuntu")
+	got, want = f.String(), "WORKDIR /home/ubuntu\n"
+	if got != want {
+		t.Errorf("Exepected WriteWorkdir returned %s, got %s", want, got)
+	}
+
+	f = &Dockerfile{}
+	f.WriteEntrypoint("/root")
+	got, want = f.String(), "ENTRYPOINT /root\n"
+	if got != want {
+		t.Errorf("Exepected WriteEntrypoint returned %s, got %s", want, got)
+	}
+}
\ No newline at end of file