github.com/demonoid81/moby@v0.0.0-20200517203328-62dd8e17c460/layer/filestore_test.go (about)

     1  package layer // import "github.com/demonoid81/moby/layer"
     2  
     3  import (
     4  	"fmt"
     5  	"io/ioutil"
     6  	"math/rand"
     7  	"os"
     8  	"path/filepath"
     9  	"strings"
    10  	"syscall"
    11  	"testing"
    12  
    13  	"github.com/demonoid81/moby/pkg/stringid"
    14  	digest "github.com/opencontainers/go-digest"
    15  )
    16  
    17  func randomLayerID(seed int64) ChainID {
    18  	r := rand.New(rand.NewSource(seed))
    19  
    20  	return ChainID(digest.FromBytes([]byte(fmt.Sprintf("%d", r.Int63()))))
    21  }
    22  
    23  func newFileMetadataStore(t *testing.T) (*fileMetadataStore, string, func()) {
    24  	td, err := ioutil.TempDir("", "layers-")
    25  	if err != nil {
    26  		t.Fatal(err)
    27  	}
    28  	fms, err := newFSMetadataStore(td)
    29  	if err != nil {
    30  		t.Fatal(err)
    31  	}
    32  
    33  	return fms, td, func() {
    34  		if err := os.RemoveAll(td); err != nil {
    35  			t.Logf("Failed to cleanup %q: %s", td, err)
    36  		}
    37  	}
    38  }
    39  
    40  func assertNotDirectoryError(t *testing.T, err error) {
    41  	perr, ok := err.(*os.PathError)
    42  	if !ok {
    43  		t.Fatalf("Unexpected error %#v, expected path error", err)
    44  	}
    45  
    46  	if perr.Err != syscall.ENOTDIR {
    47  		t.Fatalf("Unexpected error %s, expected %s", perr.Err, syscall.ENOTDIR)
    48  	}
    49  }
    50  
    51  func TestCommitFailure(t *testing.T) {
    52  	fms, td, cleanup := newFileMetadataStore(t)
    53  	defer cleanup()
    54  
    55  	if err := ioutil.WriteFile(filepath.Join(td, "sha256"), []byte("was here first!"), 0644); err != nil {
    56  		t.Fatal(err)
    57  	}
    58  
    59  	tx, err := fms.StartTransaction()
    60  	if err != nil {
    61  		t.Fatal(err)
    62  	}
    63  
    64  	if err := tx.SetSize(0); err != nil {
    65  		t.Fatal(err)
    66  	}
    67  
    68  	err = tx.Commit(randomLayerID(5))
    69  	if err == nil {
    70  		t.Fatalf("Expected error committing with invalid layer parent directory")
    71  	}
    72  	assertNotDirectoryError(t, err)
    73  }
    74  
    75  func TestStartTransactionFailure(t *testing.T) {
    76  	fms, td, cleanup := newFileMetadataStore(t)
    77  	defer cleanup()
    78  
    79  	if err := ioutil.WriteFile(filepath.Join(td, "tmp"), []byte("was here first!"), 0644); err != nil {
    80  		t.Fatal(err)
    81  	}
    82  
    83  	_, err := fms.StartTransaction()
    84  	if err == nil {
    85  		t.Fatalf("Expected error starting transaction with invalid layer parent directory")
    86  	}
    87  	assertNotDirectoryError(t, err)
    88  
    89  	if err := os.Remove(filepath.Join(td, "tmp")); err != nil {
    90  		t.Fatal(err)
    91  	}
    92  
    93  	tx, err := fms.StartTransaction()
    94  	if err != nil {
    95  		t.Fatal(err)
    96  	}
    97  
    98  	if expected := filepath.Join(td, "tmp"); strings.HasPrefix(expected, tx.String()) {
    99  		t.Fatalf("Unexpected transaction string %q, expected prefix %q", tx.String(), expected)
   100  	}
   101  
   102  	if err := tx.Cancel(); err != nil {
   103  		t.Fatal(err)
   104  	}
   105  }
   106  
   107  func TestGetOrphan(t *testing.T) {
   108  	fms, td, cleanup := newFileMetadataStore(t)
   109  	defer cleanup()
   110  
   111  	layerRoot := filepath.Join(td, "sha256")
   112  	if err := os.MkdirAll(layerRoot, 0755); err != nil {
   113  		t.Fatal(err)
   114  	}
   115  
   116  	tx, err := fms.StartTransaction()
   117  	if err != nil {
   118  		t.Fatal(err)
   119  	}
   120  
   121  	layerid := randomLayerID(5)
   122  	err = tx.Commit(layerid)
   123  	if err != nil {
   124  		t.Fatal(err)
   125  	}
   126  	layerPath := fms.getLayerDirectory(layerid)
   127  	if err := ioutil.WriteFile(filepath.Join(layerPath, "cache-id"), []byte(stringid.GenerateRandomID()), 0644); err != nil {
   128  		t.Fatal(err)
   129  	}
   130  
   131  	orphanLayers, err := fms.getOrphan()
   132  	if err != nil {
   133  		t.Fatal(err)
   134  	}
   135  	if len(orphanLayers) != 0 {
   136  		t.Fatalf("Expected to have zero orphan layers")
   137  	}
   138  
   139  	layeridSplit := strings.Split(layerid.String(), ":")
   140  	newPath := filepath.Join(layerRoot, fmt.Sprintf("%s-%s-removing", layeridSplit[1], stringid.GenerateRandomID()))
   141  	err = os.Rename(layerPath, newPath)
   142  	if err != nil {
   143  		t.Fatal(err)
   144  	}
   145  	orphanLayers, err = fms.getOrphan()
   146  	if err != nil {
   147  		t.Fatal(err)
   148  	}
   149  	if len(orphanLayers) != 1 {
   150  		t.Fatalf("Expected to have one orphan layer")
   151  	}
   152  }