mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
7a85579917
The `digest` data type, used throughout docker for image verification and identity, has been broken out into `opencontainers/go-digest`. This PR updates the dependencies and moves uses over to the new type. Signed-off-by: Stephen J Day <stephen.day@docker.com>
104 lines
2.2 KiB
Go
104 lines
2.2 KiB
Go
package layer
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"math/rand"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"syscall"
|
|
"testing"
|
|
|
|
"github.com/opencontainers/go-digest"
|
|
)
|
|
|
|
func randomLayerID(seed int64) ChainID {
|
|
r := rand.New(rand.NewSource(seed))
|
|
|
|
return ChainID(digest.FromBytes([]byte(fmt.Sprintf("%d", r.Int63()))))
|
|
}
|
|
|
|
func newFileMetadataStore(t *testing.T) (*fileMetadataStore, string, func()) {
|
|
td, err := ioutil.TempDir("", "layers-")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
fms, err := NewFSMetadataStore(td)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return fms.(*fileMetadataStore), td, func() {
|
|
if err := os.RemoveAll(td); err != nil {
|
|
t.Logf("Failed to cleanup %q: %s", td, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func assertNotDirectoryError(t *testing.T, err error) {
|
|
perr, ok := err.(*os.PathError)
|
|
if !ok {
|
|
t.Fatalf("Unexpected error %#v, expected path error", err)
|
|
}
|
|
|
|
if perr.Err != syscall.ENOTDIR {
|
|
t.Fatalf("Unexpected error %s, expected %s", perr.Err, syscall.ENOTDIR)
|
|
}
|
|
}
|
|
|
|
func TestCommitFailure(t *testing.T) {
|
|
fms, td, cleanup := newFileMetadataStore(t)
|
|
defer cleanup()
|
|
|
|
if err := ioutil.WriteFile(filepath.Join(td, "sha256"), []byte("was here first!"), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
tx, err := fms.StartTransaction()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := tx.SetSize(0); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = tx.Commit(randomLayerID(5))
|
|
if err == nil {
|
|
t.Fatalf("Expected error committing with invalid layer parent directory")
|
|
}
|
|
assertNotDirectoryError(t, err)
|
|
}
|
|
|
|
func TestStartTransactionFailure(t *testing.T) {
|
|
fms, td, cleanup := newFileMetadataStore(t)
|
|
defer cleanup()
|
|
|
|
if err := ioutil.WriteFile(filepath.Join(td, "tmp"), []byte("was here first!"), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
_, err := fms.StartTransaction()
|
|
if err == nil {
|
|
t.Fatalf("Expected error starting transaction with invalid layer parent directory")
|
|
}
|
|
assertNotDirectoryError(t, err)
|
|
|
|
if err := os.Remove(filepath.Join(td, "tmp")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
tx, err := fms.StartTransaction()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if expected := filepath.Join(td, "tmp"); strings.HasPrefix(expected, tx.String()) {
|
|
t.Fatalf("Unexpected transaction string %q, expected prefix %q", tx.String(), expected)
|
|
}
|
|
|
|
if err := tx.Cancel(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|