github.com/demonoid81/moby@v0.0.0-20200517203328-62dd8e17c460/testutil/fakecontext/context.go (about)

     1  package fakecontext // import "github.com/demonoid81/moby/testutil/fakecontext"
     2  
     3  import (
     4  	"bytes"
     5  	"io"
     6  	"io/ioutil"
     7  	"os"
     8  	"path/filepath"
     9  	"testing"
    10  
    11  	"github.com/demonoid81/moby/pkg/archive"
    12  )
    13  
    14  // New creates a fake build context
    15  func New(t testing.TB, dir string, modifiers ...func(*Fake) error) *Fake {
    16  	t.Helper()
    17  	fakeContext := &Fake{Dir: dir}
    18  	if dir == "" {
    19  		if err := newDir(fakeContext); err != nil {
    20  			t.Fatal(err)
    21  		}
    22  	}
    23  
    24  	for _, modifier := range modifiers {
    25  		if err := modifier(fakeContext); err != nil {
    26  			t.Fatal(err)
    27  		}
    28  	}
    29  
    30  	return fakeContext
    31  }
    32  
    33  func newDir(fake *Fake) error {
    34  	tmp, err := ioutil.TempDir("", "fake-context")
    35  	if err != nil {
    36  		return err
    37  	}
    38  	if err := os.Chmod(tmp, 0755); err != nil {
    39  		return err
    40  	}
    41  	fake.Dir = tmp
    42  	return nil
    43  }
    44  
    45  // WithFile adds the specified file (with content) in the build context
    46  func WithFile(name, content string) func(*Fake) error {
    47  	return func(ctx *Fake) error {
    48  		return ctx.Add(name, content)
    49  	}
    50  }
    51  
    52  // WithDockerfile adds the specified content as Dockerfile in the build context
    53  func WithDockerfile(content string) func(*Fake) error {
    54  	return WithFile("Dockerfile", content)
    55  }
    56  
    57  // WithFiles adds the specified files in the build context, content is a string
    58  func WithFiles(files map[string]string) func(*Fake) error {
    59  	return func(fakeContext *Fake) error {
    60  		for file, content := range files {
    61  			if err := fakeContext.Add(file, content); err != nil {
    62  				return err
    63  			}
    64  		}
    65  		return nil
    66  	}
    67  }
    68  
    69  // WithBinaryFiles adds the specified files in the build context, content is binary
    70  func WithBinaryFiles(files map[string]*bytes.Buffer) func(*Fake) error {
    71  	return func(fakeContext *Fake) error {
    72  		for file, content := range files {
    73  			if err := fakeContext.Add(file, content.String()); err != nil {
    74  				return err
    75  			}
    76  		}
    77  		return nil
    78  	}
    79  }
    80  
    81  // Fake creates directories that can be used as a build context
    82  type Fake struct {
    83  	Dir string
    84  }
    85  
    86  // Add a file at a path, creating directories where necessary
    87  func (f *Fake) Add(file, content string) error {
    88  	return f.addFile(file, []byte(content))
    89  }
    90  
    91  func (f *Fake) addFile(file string, content []byte) error {
    92  	fp := filepath.Join(f.Dir, filepath.FromSlash(file))
    93  	dirpath := filepath.Dir(fp)
    94  	if dirpath != "." {
    95  		if err := os.MkdirAll(dirpath, 0755); err != nil {
    96  			return err
    97  		}
    98  	}
    99  	return ioutil.WriteFile(fp, content, 0644)
   100  
   101  }
   102  
   103  // Delete a file at a path
   104  func (f *Fake) Delete(file string) error {
   105  	fp := filepath.Join(f.Dir, filepath.FromSlash(file))
   106  	return os.RemoveAll(fp)
   107  }
   108  
   109  // Close deletes the context
   110  func (f *Fake) Close() error {
   111  	return os.RemoveAll(f.Dir)
   112  }
   113  
   114  // AsTarReader returns a ReadCloser with the contents of Dir as a tar archive.
   115  func (f *Fake) AsTarReader(t testing.TB) io.ReadCloser {
   116  	t.Helper()
   117  	reader, err := archive.TarWithOptions(f.Dir, &archive.TarOptions{})
   118  	if err != nil {
   119  		t.Fatalf("Failed to create tar from %s: %s", f.Dir, err)
   120  	}
   121  	return reader
   122  }