github.com/docker/cnab-to-oci@v0.3.0-beta4/remotes/mount_test.go (about)

     1  package remotes
     2  
     3  import (
     4  	"context"
     5  	"net/http"
     6  	"net/http/httptest"
     7  	"net/url"
     8  	"strings"
     9  	"testing"
    10  
    11  	"github.com/containerd/containerd/remotes/docker"
    12  	"github.com/docker/distribution/reference"
    13  	ocischemav1 "github.com/opencontainers/image-spec/specs-go/v1"
    14  	"gotest.tools/assert"
    15  )
    16  
    17  type twoStepReader struct {
    18  	first        []byte
    19  	second       []byte
    20  	hasReadFirst bool
    21  }
    22  
    23  func (r *twoStepReader) Read(p []byte) (n int, err error) {
    24  	if r.hasReadFirst {
    25  		return copy(p, r.second), nil
    26  	}
    27  	r.hasReadFirst = true
    28  	return copy(p, r.first), nil
    29  }
    30  func (r *twoStepReader) Close() error {
    31  	return nil
    32  }
    33  
    34  func TestRemoteReaderAtShortReads(t *testing.T) {
    35  	helloWorld := []byte("Hello world!")
    36  	r := &twoStepReader{
    37  		first:  helloWorld[:5],
    38  		second: helloWorld[5:],
    39  	}
    40  	tested := &remoteReaderAt{
    41  		ReadCloser: r,
    42  		size:       int64(len(helloWorld)),
    43  	}
    44  
    45  	actual := make([]byte, len(helloWorld))
    46  	n, err := tested.ReadAt(actual, 0)
    47  	assert.NilError(t, err)
    48  	assert.Equal(t, n, len(helloWorld))
    49  	assert.DeepEqual(t, helloWorld, actual)
    50  }
    51  
    52  func TestMountOnPush(t *testing.T) {
    53  	hasMounted := false
    54  	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    55  		// Testing if mount is called, mount API call is in the form of:
    56  		// POST http://<REGISTRY>/<REPO>/<IMAGE>/blobs/uploads?from=<REPO2>/<IMAGE2>
    57  		if strings.Contains(r.URL.EscapedPath(), "library/test/blobs/uploads/") && strings.Contains(r.URL.Query().Get("from"), "library/busybox") {
    58  			hasMounted = true
    59  		}
    60  		// We don't really care what we send here.
    61  		w.WriteHeader(404)
    62  	})
    63  	server := httptest.NewServer(handler)
    64  	defer server.Close()
    65  
    66  	resolver := docker.NewResolver(docker.ResolverOptions{
    67  		PlainHTTP: true,
    68  	})
    69  
    70  	u, err := url.Parse(server.URL)
    71  	assert.NilError(t, err)
    72  
    73  	r, err := resolver.Pusher(context.TODO(), u.Hostname()+":"+u.Port()+"/library/test")
    74  	assert.NilError(t, err)
    75  
    76  	ref, err := reference.WithName(u.Hostname() + "/library/busybox")
    77  	assert.NilError(t, err)
    78  
    79  	desc := ocischemav1.Descriptor{}
    80  	_, _ = pushWithAnnotation(context.TODO(), r, ref, desc)
    81  	assert.Equal(t, hasMounted, true)
    82  }