Function TestStore

factstore/simplecolumn_test.go:119–172

(t *testing.T)

}

func TestStore(t *testing.T) {
	m := testStore(t)
	sc := SimpleColumn{true /* deterministic */}
	var buf bytes.Buffer
	if err := sc.WriteTo(m, &buf); err != nil {
		t.Fatal(err)
	}

	s, err := NewSimpleColumnStore(func() (io.ReadCloser, error) {
		return io.NopCloser(bytes.NewReader(buf.Bytes())), nil
	})

	if err != nil {
		t.Fatal(err)
	}

	if len(s.predicateFactCount) != len(m.ListPredicates()) {
		t.Errorf("NewSimpleColumnStore: got %d predicates want %d", len(s.predicateFactCount), len(m.ListPredicates()))
	}

	var foundBaz bool
	s.GetFacts(atom("baz()"), func(a ast.Atom) error {
		if !a.Equals(atom("baz()")) {
			t.Errorf("GetFacts(baz()): got %v want baz()", a)
		}
		foundBaz = true
		return nil
	})
	if !foundBaz {
		t.Errorf("GetFacts(baz()): got nothing want baz()")
	}

	tests := []struct {
		query string
		want  int
	}{
		{"foo(X)", wantNumFooFacts},
		{"bar(X, Y, Z)", wantNumBarFacts},
		{"qaz(X, Y, Z)", wantNumQazFacts},
	}
	for _, test := range tests {
		var count int
		s.GetFacts(atom(test.query), func(a ast.Atom) error {
			if !m.Contains(a) {
				t.Errorf("GetFacts(%s): unexpected fact: %v", test.query, a)
			}
			count++
			return nil
		})
		if count != test.want {
			t.Errorf("GetFacts(%s): got %d want %d facts", test.query, count, test.want)
		}
	}
}

func TestFiltered(t *testing.T) {
	m := testStore(t)