package fakecontext // import "github.com/docker/docker/integration-cli/cli/build/fakecontext" import ( "bytes" "io" "io/ioutil" "os" "path/filepath" "github.com/docker/docker/pkg/archive" ) type testingT interface { Fatal(args ...interface{}) Fatalf(string, ...interface{}) } // New creates a fake build context func New(t testingT, dir string, modifiers ...func(*Fake) error) *Fake { fakeContext := &Fake{Dir: dir} if dir == "" { if err := newDir(fakeContext); err != nil { t.Fatal(err) } } for _, modifier := range modifiers { if err := modifier(fakeContext); err != nil { t.Fatal(err) } } return fakeContext } func newDir(fake *Fake) error { tmp, err := ioutil.TempDir("", "fake-context") if err != nil { return err } if err := os.Chmod(tmp, 0755); err != nil { return err } fake.Dir = tmp return nil } // WithFile adds the specified file (with content) in the build context func WithFile(name, content string) func(*Fake) error { return func(ctx *Fake) error { return ctx.Add(name, content) } } // WithDockerfile adds the specified content as Dockerfile in the build context func WithDockerfile(content string) func(*Fake) error { return WithFile("Dockerfile", content) } // WithFiles adds the specified files in the build context, content is a string func WithFiles(files map[string]string) func(*Fake) error { return func(fakeContext *Fake) error { for file, content := range files { if err := fakeContext.Add(file, content); err != nil { return err } } return nil } } // WithBinaryFiles adds the specified files in the build context, content is binary func WithBinaryFiles(files map[string]*bytes.Buffer) func(*Fake) error { return func(fakeContext *Fake) error { for file, content := range files { if err := fakeContext.Add(file, string(content.Bytes())); err != nil { return err } } return nil } } // Fake creates directories that can be used as a build context type Fake struct { Dir string } // Add a file at a path, creating directories where necessary func (f *Fake) Add(file, content string) error { return f.addFile(file, []byte(content)) } func (f *Fake) addFile(file string, content []byte) error { fp := filepath.Join(f.Dir, filepath.FromSlash(file)) dirpath := filepath.Dir(fp) if dirpath != "." { if err := os.MkdirAll(dirpath, 0755); err != nil { return err } } return ioutil.WriteFile(fp, content, 0644) } // Delete a file at a path func (f *Fake) Delete(file string) error { fp := filepath.Join(f.Dir, filepath.FromSlash(file)) return os.RemoveAll(fp) } // Close deletes the context func (f *Fake) Close() error { return os.RemoveAll(f.Dir) } // AsTarReader returns a ReadCloser with the contents of Dir as a tar archive. func (f *Fake) AsTarReader(t testingT) io.ReadCloser { reader, err := archive.TarWithOptions(f.Dir, &archive.TarOptions{}) if err != nil { t.Fatalf("Failed to create tar from %s: %s", f.Dir, err) } return reader }