repo_fake_test.go

package pkgcraft_test

import (
	"testing"

	"github.com/stretchr/testify/assert"

	. "github.com/pkgcraft/pkgcraft-go"
)

func TestNewFakeRepo(t *testing.T) {
	// empty
	repo, _ := NewFakeRepo("fake", 0, []string{})
	assert.Equal(t, repo.Len(), 0)

	// single pkg
	repo, _ = NewFakeRepo("fake", 0, []string{"cat/pkg-1"})
	assert.Equal(t, repo.Len(), 1)

	// multiple pkgs with invalid cpv
	repo, _ = NewFakeRepo("fake", 0, []string{"a/b-1", "c/d-2", "=cat/pkg-1"})
	assert.Equal(t, repo.Len(), 2)

	// invalid repo id
	_, err := NewFakeRepo("fake\xc5", 0, []string{})
	assert.NotNil(t, err)

	// CPV string with invalid utf8
	_, err = NewFakeRepo("fake", 0, []string{"cat/pkg\xc5-1"})
	assert.NotNil(t, err)
}

func TestFakeRepoExtend(t *testing.T) {
	// empty
	repo, _ := NewFakeRepo("fake", 0, []string{})
	assert.Equal(t, repo.Len(), 0)

	// add single pkg
	err := repo.Extend([]string{"cat/pkg-1"})
	assert.Equal(t, repo.Len(), 1)
	assert.Nil(t, err)

	// add multiple pkgs with overlap
	err = repo.Extend([]string{"cat/pkg-1", "cat/pkg-2"})
	assert.Equal(t, repo.Len(), 2)
	assert.Nil(t, err)

	// mutations allowed after adding to a config
	config := NewConfig()
	defer config.Close()
	err = config.AddRepo(repo)
	assert.Nil(t, err)
	err = repo.Extend([]string{"cat/pkg-3"})
	assert.Nil(t, err)
}

func TestFakeRepoIter(t *testing.T) {
	var cpvs []string

	// empty
	repo, _ := NewFakeRepo("fake", 0, []string{})
	iter := repo.Iter()
	assert.False(t, iter.HasNext())
	assert.Nil(t, iter.Next())

	// add single pkg
	err := repo.Extend([]string{"cat/pkg-1"})
	assert.Nil(t, err)
	for iter := repo.Iter(); iter.HasNext(); {
		pkg := iter.Next()
		// verify repos are equal
		assert.True(t, repo.Cmp(pkg.Repo()) == 0)
		cpvs = append(cpvs, pkg.Cpv().String())
	}
	assert.Equal(t, cpvs, []string{"cat/pkg-1"})

	// reset slice
	cpvs = cpvs[:0]

	// add multiple pkgs with overlap
	err = repo.Extend([]string{"cat/pkg-1", "cat/pkg-2"})
	assert.Nil(t, err)
	for iter := repo.Iter(); iter.HasNext(); {
		pkg := iter.Next()
		// verify repos are equal
		assert.True(t, repo.Cmp(pkg.Repo()) == 0)
		cpvs = append(cpvs, pkg.Cpv().String())
	}
	assert.Equal(t, cpvs, []string{"cat/pkg-1", "cat/pkg-2"})
}

func TestFakeRepoPkgs(t *testing.T) {
	var cpvs []string

	// empty
	repo, _ := NewFakeRepo("fake", 0, []string{})
	assert.Empty(t, repo.Pkgs())

	// add single pkg
	err := repo.Extend([]string{"cat/pkg-1"})
	assert.Nil(t, err)
	for pkg := range repo.Pkgs() {
		// verify repos are equal
		assert.True(t, repo.Cmp(pkg.Repo()) == 0)
		cpvs = append(cpvs, pkg.Cpv().String())
	}
	assert.Equal(t, cpvs, []string{"cat/pkg-1"})

	// reset slice
	cpvs = cpvs[:0]

	// add multiple pkgs with overlap
	err = repo.Extend([]string{"cat/pkg-1", "cat/pkg-2"})
	assert.Nil(t, err)
	for pkg := range repo.Pkgs() {
		// verify repos are equal
		assert.True(t, repo.Cmp(pkg.Repo()) == 0)
		cpvs = append(cpvs, pkg.Cpv().String())
	}
	assert.Equal(t, cpvs, []string{"cat/pkg-1", "cat/pkg-2"})
}

func TestFakeRepoIterRestrict(t *testing.T) {
	var cpvs []string
	restrict, _ := NewRestrict("<cat/pkg-2")

	// empty
	repo, _ := NewFakeRepo("fake", 0, []string{})
	iter := repo.IterRestrict(restrict)
	assert.False(t, iter.HasNext())
	assert.Nil(t, iter.Next())

	// add single pkg
	err := repo.Extend([]string{"cat/pkg-1"})
	assert.Nil(t, err)
	for iter := repo.IterRestrict(restrict); iter.HasNext(); {
		pkg := iter.Next()
		// verify repos are equal
		assert.True(t, repo.Cmp(pkg.Repo()) == 0)
		cpvs = append(cpvs, pkg.Cpv().String())
	}
	assert.Equal(t, cpvs, []string{"cat/pkg-1"})

	// reset slice
	cpvs = cpvs[:0]

	// add multiple pkgs with overlap
	err = repo.Extend([]string{"cat/pkg-1", "cat/pkg-2"})
	assert.Nil(t, err)
	for iter := repo.IterRestrict(restrict); iter.HasNext(); {
		pkg := iter.Next()
		// verify repos are equal
		assert.True(t, repo.Cmp(pkg.Repo()) == 0)
		cpvs = append(cpvs, pkg.Cpv().String())
	}
	assert.Equal(t, cpvs, []string{"cat/pkg-1"})
}

func TestFakeRepoRestrictPkgs(t *testing.T) {
	var cpvs []string
	restrict, _ := NewRestrict("<cat/pkg-2")

	// empty
	repo, _ := NewFakeRepo("fake", 0, []string{})
	assert.Empty(t, repo.RestrictPkgs(restrict))

	// add single pkg
	err := repo.Extend([]string{"cat/pkg-1"})
	assert.Nil(t, err)
	for pkg := range repo.RestrictPkgs(restrict) {
		// verify repos are equal
		assert.True(t, repo.Cmp(pkg.Repo()) == 0)
		cpvs = append(cpvs, pkg.Cpv().String())
	}
	assert.Equal(t, cpvs, []string{"cat/pkg-1"})

	// reset slice
	cpvs = cpvs[:0]

	// add multiple pkgs with overlap
	err = repo.Extend([]string{"cat/pkg-1", "cat/pkg-2"})
	assert.Nil(t, err)
	for pkg := range repo.RestrictPkgs(restrict) {
		// verify repos are equal
		assert.True(t, repo.Cmp(pkg.Repo()) == 0)
		cpvs = append(cpvs, pkg.Cpv().String())
	}
	assert.Equal(t, cpvs, []string{"cat/pkg-1"})
}