github.com/marinho/drone@v0.2.1-0.20140504195434-d3ba962e89a7/pkg/build/dockerfile/dockerfile_test.go (about)

     1  package dockerfile
     2  
     3  import (
     4  	"testing"
     5  )
     6  
     7  func TestWrite(t *testing.T) {
     8  
     9  	var f = New("ubuntu")
    10  	var got, want = f.String(), "FROM ubuntu\n"
    11  	if got != want {
    12  		t.Errorf("Exepected New() returned %s, got %s", want, got)
    13  	}
    14  
    15  	f = &Dockerfile{}
    16  	f.WriteAdd("src", "target")
    17  	got, want = f.String(), "ADD src target\n"
    18  	if got != want {
    19  		t.Errorf("Exepected WriteAdd returned %s, got %s", want, got)
    20  	}
    21  
    22  	f = &Dockerfile{}
    23  	f.WriteFrom("ubuntu")
    24  	got, want = f.String(), "FROM ubuntu\n"
    25  	if got != want {
    26  		t.Errorf("Exepected WriteFrom returned %s, got %s", want, got)
    27  	}
    28  
    29  	f = &Dockerfile{}
    30  	f.WriteRun("whoami")
    31  	got, want = f.String(), "RUN whoami\n"
    32  	if got != want {
    33  		t.Errorf("Exepected WriteRun returned %s, got %s", want, got)
    34  	}
    35  
    36  	f = &Dockerfile{}
    37  	f.WriteUser("root")
    38  	got, want = f.String(), "USER root\n"
    39  	if got != want {
    40  		t.Errorf("Exepected WriteUser returned %s, got %s", want, got)
    41  	}
    42  
    43  	f = &Dockerfile{}
    44  	f.WriteEnv("FOO", "BAR")
    45  	got, want = f.String(), "ENV FOO BAR\n"
    46  	if got != want {
    47  		t.Errorf("Exepected WriteEnv returned %s, got %s", want, got)
    48  	}
    49  
    50  	f = &Dockerfile{}
    51  	f.WriteWorkdir("/home/ubuntu")
    52  	got, want = f.String(), "WORKDIR /home/ubuntu\n"
    53  	if got != want {
    54  		t.Errorf("Exepected WriteWorkdir returned %s, got %s", want, got)
    55  	}
    56  
    57  	f = &Dockerfile{}
    58  	f.WriteEntrypoint("/root")
    59  	got, want = f.String(), "ENTRYPOINT /root\n"
    60  	if got != want {
    61  		t.Errorf("Exepected WriteEntrypoint returned %s, got %s", want, got)
    62  	}
    63  }