fiddlek/go/named/named_test.go

package named

import (
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
	"go.skia.org/infra/fiddlek/go/store"
	"go.skia.org/infra/go/testutils/unittest"
)

type namedMock struct {
	lookup map[string]string
}

func (n *namedMock) GetHashFromName(name string) (string, error) {
	if hash, ok := n.lookup[name]; !ok {
		return "", fmt.Errorf("Not found")
	} else {
		return hash, nil
	}
}

func (n *namedMock) WriteName(name, hash, user, status string) error {
	if name == "bad_name_to_trigger_fail" {
		return fmt.Errorf("Failed to write.")
	}
	return nil
}

func TestNamed(t *testing.T) {
	unittest.SmallTest(t)
	mock := &namedMock{
		lookup: map[string]string{
			"star":      "cbb8dee39e9f1576cd97c2d504db8eee",
			"text_only": "92e8c9418a28113de73524d8b26d1c3c",
			"bad_hash":  "cbb8d",
		},
	}
	names := New(mock)
	_, err := names.DereferenceID("@missing")
	assert.Error(t, err)

	_, err = names.DereferenceID("@bad_hash")
	assert.Error(t, err)

	_, err = names.DereferenceID("cbb8")
	assert.Error(t, err, "Invalid looking fiddle hashes should fail.")

	hash, err := names.DereferenceID("@star")
	assert.NoError(t, err)
	assert.Equal(t, hash, "cbb8dee39e9f1576cd97c2d504db8eee")

	hash, err = names.DereferenceID("cbb8dee39e9f1576cd97c2d504db8eee")
	assert.NoError(t, err)
	assert.Equal(t, hash, "cbb8dee39e9f1576cd97c2d504db8eee")

	_, _, err = names.DereferenceImageID("@missing.pdf")
	assert.Error(t, err)

	_, _, err = names.DereferenceImageID("@bad_hash.pdf")
	assert.Error(t, err)

	_, _, err = names.DereferenceImageID("@star.png")
	assert.Error(t, err, "All .png's should have a prefix of either _raster or _gpu.")

	mediaHash, media, err := names.DereferenceImageID("@star.pdf")
	assert.NoError(t, err)
	assert.Equal(t, mediaHash, "cbb8dee39e9f1576cd97c2d504db8eee")
	assert.Equal(t, media, store.PDF)

	mediaHash, media, err = names.DereferenceImageID("@star_gpu.png")
	assert.NoError(t, err)
	assert.Equal(t, mediaHash, "cbb8dee39e9f1576cd97c2d504db8eee")
	assert.Equal(t, media, store.GPU)

	mediaHash, media, err = names.DereferenceImageID("cbb8dee39e9f1576cd97c2d504db8eee_raster.png")
	assert.NoError(t, err)
	assert.Equal(t, mediaHash, "cbb8dee39e9f1576cd97c2d504db8eee")
	assert.Equal(t, media, store.CPU)

	mediaHash, media, err = names.DereferenceImageID("@star_gpu.webm")
	assert.NoError(t, err)
	assert.Equal(t, mediaHash, "cbb8dee39e9f1576cd97c2d504db8eee")
	assert.Equal(t, media, store.ANIM_GPU)

	mediaHash, media, err = names.DereferenceImageID("@star_cpu.webm")
	assert.NoError(t, err)
	assert.Equal(t, mediaHash, "cbb8dee39e9f1576cd97c2d504db8eee")
	assert.Equal(t, media, store.ANIM_CPU)

	mediaHash, media, err = names.DereferenceImageID("@star_glinfo.txt")
	assert.NoError(t, err)
	assert.Equal(t, mediaHash, "cbb8dee39e9f1576cd97c2d504db8eee")
	assert.Equal(t, media, store.GLINFO)

	mediaHash, media, err = names.DereferenceImageID("@star.txt")
	assert.NoError(t, err)
	assert.Equal(t, mediaHash, "cbb8dee39e9f1576cd97c2d504db8eee")
	assert.Equal(t, media, store.TXT)

	mediaHash, media, err = names.DereferenceImageID("@text_only.txt")
	assert.NoError(t, err)
	assert.Equal(t, mediaHash, "92e8c9418a28113de73524d8b26d1c3c")
	assert.Equal(t, media, store.TXT)
}

func TestAdd(t *testing.T) {
	unittest.SmallTest(t)
	mock := &namedMock{
		lookup: map[string]string{
			"star":     "cbb8dee39e9f1576cd97c2d504db8eee",
			"bad_hash": "cbb8d",
		},
	}

	names := New(mock)
	err := names.Add("a_good_name", "cbb8dee39e9f1576cd97c2d504db8eee", "user", true)
	assert.NoError(t, err)

	err = names.Add("no spaces in names", "cbb8dee39e9f1576cd97c2d504db8eee", "user", true)
	assert.Error(t, err)

	err = names.Add("a_good_name", "cbb8_bash_hash", "user", true)
	assert.Error(t, err)

	err = names.Add("star", "cbb8dee39e9f1576cd97c2d504db8eee", "user", false)
	assert.Equal(t, err, DuplicateNameErr)

	err = names.Add("star", "cbb8dee39e9f1576cd97c2d504db8eee", "user", true)
	assert.NoError(t, err)
}