dedupe-linker/base/base_test.go

package base

import (
	"bytes"
	"fmt"
	"io/ioutil"
	"os"
	"path"
	"testing"
)

func TestSumPath(t *testing.T) {
	expected := "/var/dedup/blobs/sha1/de/deadbeef"
	b := Base{Path: "/var/dedup", HashName: "sha1"}
	if bp := b.blobPath("deadbeef"); bp != expected {
		t.Errorf("expected %q, got %q", expected, bp)
	}
}

func TestGetPut(t *testing.T) {
	var (
		srcDir, destDir string
		err             error
	)
	if srcDir, err = ioutil.TempDir("", "dedupe-linker-src"); err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(srcDir)
	if destDir, err = ioutil.TempDir("", "dedupe-linker-dest"); err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(destDir)

	b, err := NewBase(destDir, "sha1")
	if err != nil {
		t.Fatal(err)
	}

	rHash := "deadbeaf"
	rMsg := "this is the dead beef"

	r := bytes.NewReader([]byte(rMsg))
	sum, err := b.PutBlob(r, 0666)
	if err != nil {
		t.Error(err)
	}

	fi, err := b.Stat(rHash)
	if err != nil {
		t.Error(err)
	}
	fmt.Printf("%#v\n", fi.Sys())

	if err = b.LinkTo(path.Join(srcDir, "beef1.txt"), rHash); err != nil {
		t.Error(err)
	}
	fi2, err := os.Stat(path.Join(srcDir, "beef1.txt"))
	if err != nil {
		t.Error(err)
	}
	fmt.Printf("%#v\n", fi2.Sys())

	if err = b.LinkTo(path.Join(srcDir, "beef1.txt"), rHash); err != nil && !os.IsExist(err) {
		t.Error(err)
	}

	if rHash != sum {
		t.Errorf("expected %s; got %s", rHash, sum)
	}
}