1c181429da
These packages really impact on test readability with crazy action at a distance. In addition to this removal of goconvey reduced the test run time for leveldb on average by about 40-50%.
107 lines
2.8 KiB
Go
107 lines
2.8 KiB
Go
// Copyright 2014 The Cayley Authors. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package iterator
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/cayley/graph"
|
|
)
|
|
|
|
var simpleStore = &store{data: []string{"0", "1", "2", "3", "4", "5"}}
|
|
|
|
func simpleFixedIterator() *Fixed {
|
|
f := newFixed()
|
|
for i := 0; i < 5; i++ {
|
|
f.Add(i)
|
|
}
|
|
return f
|
|
}
|
|
|
|
func checkIteratorContains(ts graph.TripleStore, it graph.Iterator, expected []string, t *testing.T) {
|
|
var actual []string
|
|
actual = nil
|
|
for {
|
|
val, ok := it.Next()
|
|
if !ok {
|
|
break
|
|
}
|
|
actual = append(actual, ts.NameOf(val))
|
|
}
|
|
actualSet := actual[:]
|
|
for _, a := range expected {
|
|
found := false
|
|
for j, b := range actualSet {
|
|
if a == b {
|
|
actualSet = append(actualSet[:j], actualSet[j+1:]...)
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
t.Error("Couldn't find", a, "in actual output.\nActual:", actual, "\nExpected: ", expected, "\nRemainder: ", actualSet)
|
|
return
|
|
}
|
|
}
|
|
if len(actualSet) != 0 {
|
|
t.Error("Actual output has more than expected.\nActual:", actual, "\nExpected: ", expected, "\nRemainder: ", actualSet)
|
|
}
|
|
}
|
|
|
|
func TestWorkingIntValueComparison(t *testing.T) {
|
|
ts := simpleStore
|
|
fixed := simpleFixedIterator()
|
|
vc := NewComparison(fixed, kCompareLT, int64(3), ts)
|
|
checkIteratorContains(ts, vc, []string{"0", "1", "2"}, t)
|
|
}
|
|
|
|
func TestFailingIntValueComparison(t *testing.T) {
|
|
ts := simpleStore
|
|
fixed := simpleFixedIterator()
|
|
vc := NewComparison(fixed, kCompareLT, int64(0), ts)
|
|
checkIteratorContains(ts, vc, []string{}, t)
|
|
}
|
|
|
|
func TestWorkingGT(t *testing.T) {
|
|
ts := simpleStore
|
|
fixed := simpleFixedIterator()
|
|
vc := NewComparison(fixed, kCompareGT, int64(2), ts)
|
|
checkIteratorContains(ts, vc, []string{"3", "4"}, t)
|
|
}
|
|
|
|
func TestWorkingGTE(t *testing.T) {
|
|
ts := simpleStore
|
|
fixed := simpleFixedIterator()
|
|
vc := NewComparison(fixed, kCompareGTE, int64(2), ts)
|
|
checkIteratorContains(ts, vc, []string{"2", "3", "4"}, t)
|
|
}
|
|
|
|
func TestVCICheck(t *testing.T) {
|
|
ts := simpleStore
|
|
fixed := simpleFixedIterator()
|
|
vc := NewComparison(fixed, kCompareGTE, int64(2), ts)
|
|
if vc.Check(1) {
|
|
t.Error("1 is less than 2, should be GTE")
|
|
}
|
|
if !vc.Check(2) {
|
|
t.Error("2 is GTE 2")
|
|
}
|
|
if !vc.Check(3) {
|
|
t.Error("3 is GTE 2")
|
|
}
|
|
if vc.Check(5) {
|
|
t.Error("5 is not in the underlying iterator")
|
|
}
|
|
}
|