upload_test.go

package rest

import (
	"context"
	"crypto/rand"
	"crypto/sha256"
	"encoding/hex"
	"io"
	"log"
	"testing"
)

type emptyReader struct{}

func (*emptyReader) Read(b []byte) (int, error) {
	return 0, io.EOF
}

func TestUpload(t *testing.T) {
	// input file (16MB, non seekable)
	var input io.Reader
	input = &io.LimitedReader{R: rand.Reader, N: 16 * 1024 * 1024}
	// compute sha256 of bytes we send
	hash := sha256.New()
	input = io.TeeReader(input, hash)

	ctx := context.Background()
	res, err := Upload(ctx, "Misc/Debug:testUpload", "POST", Param{"filename": "test.bin"}, input, "application/octet-stream")
	//res, err := Upload(ctx, "Shell/Bit:upload", "POST", Param{"filename": "test.bin"}, input, "application/octet-stream")

	if err != nil {
		t.Fatalf("failed to do upload: %s", err)
	}

	log.Printf("expected hash = %s", hex.EncodeToString(hash.Sum(nil)))
	log.Printf("res = %s", res.Data)
}

func TestUploadEmpty(t *testing.T) {
	// input file (0 non seekable)
	var input io.Reader
	input = &emptyReader{}

	ctx := context.Background()
	res, err := Upload(ctx, "Misc/Debug:testUpload", "POST", Param{"filename": "empty.bin"}, input, "application/octet-stream")
	//res, err := Upload(ctx, "Shell/Bit:upload", "POST", Param{"filename": "test.bin"}, input, "application/octet-stream")

	if err != nil {
		t.Fatalf("failed to do upload: %s", err)
	}

	log.Printf("res = %s", res.Data)
}