out-of-tree/distro/debian/cache_test.go

package debian

import (
	"os"
	"path/filepath"
	"testing"

	"github.com/rapidloop/skv"

	"code.dumpstack.io/tools/out-of-tree/distro/debian/snapshot"
)

func TestCache(t *testing.T) {
	dir, err := os.MkdirTemp("", "out-of-tree_cache_test_")
	if err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(dir)

	path := filepath.Join(dir, "debian.cache")

	c, err := NewCache(path)
	if err != nil {
		t.Fatal(err)
	}

	image := snapshot.Package{}
	image.Deb.Hash = "12345"

	version := "4.17.14-1"

	dk := DebianKernel{
		Version: DebianKernelVersion{Package: version},
		Image:   image,
	}

	err = c.Put([]DebianKernel{dk})
	if err != nil {
		t.Fatal(err)
	}

	dk2s, err := c.Get(version)
	if err != nil {
		t.Fatal(err)
	}
	dk2 := dk2s[0]

	if dk.Image.Deb.Hash != dk2.Image.Deb.Hash {
		t.Fatalf("mismatch")
	}

	c.Close()

	c, err = NewCache(path)
	if err != nil {
		t.Fatal(err)
	}
	defer c.Close()

	dk3s, err := c.Get(version)
	if err != nil {
		t.Fatal(err)
	}
	dk3 := dk3s[0]

	if dk.Image.Deb.Hash != dk3.Image.Deb.Hash {
		t.Fatalf("mismatch")
	}

	_, err = c.Get("key not exist")
	if err == nil || err != skv.ErrNotFound {
		t.Fatal(err)
	}
}

func TestVersionsCache(t *testing.T) {
	dir, err := os.MkdirTemp("", "out-of-tree_cache_test_")
	if err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(dir)

	path := filepath.Join(dir, "debian.cache")

	c, err := NewCache(path)
	if err != nil {
		t.Fatal(err)
	}
	defer c.Close()

	versions := []string{"a", "b", "c"}

	err = c.PutVersions(versions)
	if err != nil {
		t.Fatal(err)
	}

	result, err := c.GetVersions()
	if err != nil {
		t.Fatal(err)
	}

	if len(versions) != len(result) {
		t.Fatal("mismatch")
	}
}