perf/go/shortcut/shortcuttest/shortcuttest.go

// Package shortcuttest has common code for tests of implementations of
// shortcut.Store.
package shortcuttest

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"sort"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.skia.org/infra/perf/go/shortcut"
)

// Shortcut_InsertGet does the core testing of an instance of shortcut.Store.
func InsertGet(t *testing.T, store shortcut.Store) {
	ctx := context.Background()
	// Write a shortcut, make sure the keys are out of sorted order.
	sh := &shortcut.Shortcut{
		Keys: []string{
			",arch=x86,test=testA,",
			",arch=x86,test=testC,",
			",arch=x86,test=testB,",
		},
	}
	b, err := json.Marshal(sh)
	buf := bytes.NewBuffer(b)
	id, err := store.Insert(ctx, buf)
	require.NoError(t, err)
	assert.NotEqual(t, "", id)

	// Read it back, confirm it is unchanged, except for being sorted.
	sh2, err := store.Get(ctx, id)
	require.NoError(t, err)
	assert.NotEqual(t, sh, sh2)
	sort.Strings(sh.Keys)
	assert.Equal(t, sh, sh2)
}

// Shortcut_GetNonExistent tests that we fail when retrieving an unknown
// shortcut.
func GetNonExistent(t *testing.T, store shortcut.Store) {
	ctx := context.Background()

	_, err := store.Get(ctx, "X-unknown")
	require.Error(t, err)
}

// readAll reads all the Shortcuts from the channel and returns them in a slice.
func readAll(ch <-chan *shortcut.Shortcut) []*shortcut.Shortcut {
	ret := []*shortcut.Shortcut{}
	for s := range ch {
		ret = append(ret, s)
	}
	return ret
}

// Shortcut_GetAll tests that GetAll produces a channel of all the shortcuts in
// the database.
func GetAll(t *testing.T, store shortcut.Store) {
	ctx := context.Background()
	const numShortcuts = 12
	// Write some shortcuts.
	for i := 0; i < numShortcuts; i++ {
		sh := &shortcut.Shortcut{
			Keys: []string{
				fmt.Sprintf(",arch=x86,test=test%d,", i),
			},
		}
		_, err := store.InsertShortcut(ctx, sh)
		require.NoError(t, err)
	}
	ch, err := store.GetAll(ctx)
	require.NoError(t, err)
	all := readAll(ch)
	assert.Len(t, all, numShortcuts)
	assert.True(t, strings.HasPrefix(all[0].Keys[0], ",arch=x86,test=test"))
}

// SubTestFunction is a func we will call to test one aspect of an
// implementation of regression.Store.
type SubTestFunction func(t *testing.T, store shortcut.Store)

// SubTests are all the subtests we have for regression.Store.
var SubTests = map[string]SubTestFunction{
	"Shortcut_GetAll":         GetAll,
	"Shortcut_InsertGet":      InsertGet,
	"Shortcut_GetNonExistent": GetNonExistent,
}