order_test.go

package query

import (
	"strings"
	"testing"
)

type orderTestCase struct {
	order  Order
	keys   []string
	expect []string
}

func testKeyOrder(t *testing.T, f Order, keys []string, expect []string) {
	e := make([]Entry, len(keys))
	for i, k := range keys {
		e[i] = Entry{Key: k}
	}

	res := ResultsWithEntries(Query{}, e)
	res = NaiveOrder(res, f)
	actualE := res.AllEntries()
	actual := make([]string, len(actualE))
	for i, e := range actualE {
		actual[i] = e.Key
	}

	if len(actual) != len(expect) {
		t.Error("expect != actual.", expect, actual)
	}

	if strings.Join(actual, "") != strings.Join(expect, "") {
		t.Error("expect != actual.", expect, actual)
	}
}

func TestOrderByKey(t *testing.T) {

	testKeyOrder(t, OrderByKey{}, sampleKeys, []string{
		"/a",
		"/ab",
		"/ab/c",
		"/ab/cd",
		"/abce",
		"/abcf",
	})
	testKeyOrder(t, OrderByKeyDescending{}, sampleKeys, []string{
		"/abcf",
		"/abce",
		"/ab/cd",
		"/ab/c",
		"/ab",
		"/a",
	})
}