Newer
Older
*
* 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 query
import (
Manish R Jain
committed
"fmt"
"io/ioutil"
"os"
"testing"
"time"
"github.com/dgraph-io/dgraph/commit"
Manish R Jain
committed
"github.com/dgraph-io/dgraph/gql"
"github.com/dgraph-io/dgraph/query/graph"
"github.com/dgraph-io/dgraph/store"
"github.com/dgraph-io/dgraph/task"
"github.com/dgraph-io/dgraph/worker"
Manish R Jain
committed
"github.com/google/flatbuffers/go"
)
func setErr(err *error, nerr error) {
if err != nil {
return
}
*err = nerr
}
func addEdge(t *testing.T, edge x.DirectedEdge, l *posting.List) {
if err := l.AddMutation(edge, posting.Set); err != nil {
t.Error(err)
}
}
func checkName(t *testing.T, r *task.Result, idx int, expected string) {
var tv task.Value
if ok := r.Values(&tv, idx); !ok {
t.Error("Unable to retrieve value")
}
var iname interface{}
if err := posting.ParseValue(&iname, tv.ValBytes()); err != nil {
if name != expected {
t.Errorf("Expected: %v. Got: %v", expected, name)
}
}
func checkSingleValue(t *testing.T, child *SubGraph,
attr string, value string) {
if child.Attr != attr || len(child.Result) == 0 {
t.Error("Expected attr name with some.Result")
Manish R Jain
committed
t.Errorf("Expected value length 1. Got: %v", r.ValuesLength())
}
if r.UidmatrixLength() != 1 {
t.Errorf("Expected uidmatrix length 1. Got: %v", r.UidmatrixLength())
Manish R Jain
committed
var ul task.UidList
if ok := r.Uidmatrix(&ul, 0); !ok {
t.Errorf("While parsing uidlist")
}
if ul.UidsLength() != 0 {
t.Errorf("Expected uids length 0. Got: %v", ul.UidsLength())
}
checkName(t, r, 0, value)
}
Manish R Jain
committed
func TestNewGraph(t *testing.T) {
var ex uint64
ex = 101
dir, err := ioutil.TempDir("", "storetest_")
Manish R Jain
committed
if err != nil {
t.Error(err)
Manish R Jain
committed
}
ps := new(store.Store)
ps.Init(dir)
Manish R Jain
committed
if err != nil {
t.Error(err)
}
worker.Init(ps, nil, 0, 1)
Manish R Jain
committed
r := new(task.Result)
Manish R Jain
committed
if r.UidmatrixLength() != 1 {
t.Errorf("Expected length 1. Got: %v", r.UidmatrixLength())
}
var ul task.UidList
if ok := r.Uidmatrix(&ul, 0); !ok {
t.Errorf("Unable to parse uidlist at index 0")
}
if ul.UidsLength() != 1 {
t.Errorf("Expected length 1. Got: %v", ul.UidsLength())
}
if ul.Uids(0) != ex {
t.Errorf("Expected uid: %v. Got: %v", ex, ul.Uids(0))
}
}
func populateGraph(t *testing.T) (string, *store.Store) {
Manish R Jain
committed
// logrus.SetLevel(logrus.DebugLevel)
dir, err := ioutil.TempDir("", "storetest_")
if err != nil {
t.Error(err)
ps := new(store.Store)
worker.Init(ps, nil, 0, 1)
clog := commit.NewLogger(dir, "mutations", 50<<20)
clog.Init()
// So, user we're interested in has uid: 1.
// She has 4 friends: 23, 24, 25, 31, and 101
ValueId: 23,
Source: "testing",
Timestamp: time.Now(),
}
addEdge(t, edge, posting.GetOrCreate(posting.Key(1, "friend"), ps))
addEdge(t, edge, posting.GetOrCreate(posting.Key(1, "friend"), ps))
addEdge(t, edge, posting.GetOrCreate(posting.Key(1, "friend"), ps))
addEdge(t, edge, posting.GetOrCreate(posting.Key(1, "friend"), ps))
addEdge(t, edge, posting.GetOrCreate(posting.Key(1, "friend"), ps))
// Now let's add a few properties for the main user.
addEdge(t, edge, posting.GetOrCreate(posting.Key(1, "name"), ps))
addEdge(t, edge, posting.GetOrCreate(posting.Key(1, "gender"), ps))
addEdge(t, edge, posting.GetOrCreate(posting.Key(1, "status"), ps))
// Now let's add a name for each of the friends, except 101.
addEdge(t, edge, posting.GetOrCreate(posting.Key(23, "name"), ps))
addEdge(t, edge, posting.GetOrCreate(posting.Key(24, "name"), ps))
addEdge(t, edge, posting.GetOrCreate(posting.Key(25, "name"), ps))
addEdge(t, edge, posting.GetOrCreate(posting.Key(31, "name"), ps))
Manish R Jain
committed
edge.Value = "mich"
addEdge(t, edge, posting.GetOrCreate(posting.Key(1, "_xid_"), ps))
Manish R Jain
committed
defer os.RemoveAll(dir)
// Alright. Now we have everything set up. Let's create the query.
Manish R Jain
committed
query := `
{
me(_uid_: 0x01) {
friend {
name
}
name
gender
status
}
}
`
Manish R Jain
committed
if err != nil {
t.Error(err)
}
if err != nil {
t.Error(err)
}
ch := make(chan error)
err = <-ch
if err != nil {
t.Error(err)
}
if len(sg.Children) != 4 {
t.Errorf("Expected len 4. Got: %v", len(sg.Children))
}
child := sg.Children[0]
if child.Attr != "friend" {
t.Errorf("Expected attr friend. Got: %v", child.Attr)
}
if len(child.Result) == 0 {
t.Errorf("Expected some.Result.")
Manish R Jain
committed
return
Manish R Jain
committed
if r.UidmatrixLength() != 1 {
t.Errorf("Expected 1 matrix. Got: %v", r.UidmatrixLength())
}
var ul task.UidList
if ok := r.Uidmatrix(&ul, 0); !ok {
t.Errorf("While parsing uidlist")
}
if ul.UidsLength() != 5 {
t.Errorf("Expected 5 friends. Got: %v", ul.UidsLength())
Manish R Jain
committed
if ul.Uids(0) != 23 || ul.Uids(1) != 24 || ul.Uids(2) != 25 ||
ul.Uids(3) != 31 || ul.Uids(4) != 101 {
t.Errorf("Friend ids don't match")
}
if len(child.Children) != 1 || child.Children[0].Attr != "name" {
t.Errorf("Expected attr name")
}
child = child.Children[0]
uo = flatbuffers.GetUOffsetT(child.Result)
r.Init(child.Result, uo)
if r.ValuesLength() != 5 {
t.Errorf("Expected 5 names of 5 friends")
}
checkName(t, r, 0, "Rick Grimes")
checkName(t, r, 1, "Glenn Rhee")
checkName(t, r, 2, "Daryl Dixon")
checkName(t, r, 3, "Andrea")
{
var tv task.Value
if ok := r.Values(&tv, 4); !ok {
t.Error("Unable to retrieve value")
}
if tv.ValLength() != 1 || tv.ValBytes()[0] != 0x00 {
t.Error("Expected a null byte")
}
}
checkSingleValue(t, sg.Children[1], "name", "Michonne")
checkSingleValue(t, sg.Children[2], "gender", "female")
checkSingleValue(t, sg.Children[3], "status", "alive")
Manish R Jain
committed
defer os.RemoveAll(dir)
// Alright. Now we have everything set up. Let's create the query.
Manish R Jain
committed
query := `
{
me(_uid_:0x01) {
name
gender
status
friend {
name
}
}
}
`
Manish R Jain
committed
if err != nil {
t.Error(err)
}
if err != nil {
t.Error(err)
}
ch := make(chan error)
Manish R Jain
committed
Manish R Jain
committed
var l Latency
js, err := sg.ToJson(&l)
Manish R Jain
committed
if err != nil {
t.Error(err)
}
fmt.Printf(string(js))
func getProperty(properties []*graph.Property, prop string) (v *graph.Value) {
for _, p := range properties {
if p.Prop == prop {
return p.Val
}
}
return v
}
func TestToProtocolBuffer(t *testing.T) {
dir, _ := populateGraph(t)
defer os.RemoveAll(dir)
query := `
{
me(_uid_:0x01) {
name
gender
status
friend {
name
}
}
}
`
gq, _, err := gql.Parse(query)
if err != nil {
t.Error(err)
}
sg, err := ToSubGraph(gq)
if err != nil {
t.Error(err)
}
ch := make(chan error)
err = <-ch
if err != nil {
t.Error(err)
}
if err != nil {
t.Error(err)
}
if gr.Attribute != "_root_" {
t.Errorf("Expected attribute _root_, Got: %v", gr.Attribute)
if gr.Uid != 1 {
t.Errorf("Expected uid 1, Got: %v", gr.Uid)
if gr.Xid != "mich" {
t.Errorf("Expected xid mich, Got: %v", gr.Xid)
if len(gr.Properties) != 3 {
t.Errorf("Expected values map to contain 3 properties, Got: %v",
if getProperty(gr.Properties, "name").Str != "Michonne" {
t.Errorf("Expected property name to have value Michonne, Got: %v",
getProperty(gr.Properties, "name").Str)
if len(gr.Children) != 10 {
t.Errorf("Expected 10 children, Got: %v", len(gr.Children))
child := gr.Children[0]
if child.Uid != 23 {
t.Errorf("Expected uid 23, Got: %v", gr.Uid)
}
t.Errorf("Expected attribute friend, Got: %v", child.Attribute)
t.Errorf("Expected values map to contain 1 property, Got: %v",
if getProperty(child.Properties, "name").Str != "Rick Grimes" {
t.Errorf("Expected property name to have value Rick Grimes, Got: %v",
getProperty(child.Properties, "name").Str)
}
if len(child.Children) != 0 {
t.Errorf("Expected 0 children, Got: %v", len(child.Children))
}
child = gr.Children[5]
if child.Uid != 23 {
t.Errorf("Expected uid 23, Got: %v", gr.Uid)
}
if child.Attribute != "friend" {
t.Errorf("Expected attribute friend, Got: %v", child.Attribute)
}
if len(child.Properties) != 0 {
t.Errorf("Expected values map to contain 0 properties, Got: %v",
len(child.Properties))
if len(child.Children) != 0 {
t.Errorf("Expected 0 children, Got: %v", len(child.Children))
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
func benchmarkToJson(file string, b *testing.B) {
b.ReportAllocs()
var sg SubGraph
var l Latency
f, err := ioutil.ReadFile(file)
if err != nil {
b.Error(err)
}
buf := bytes.NewBuffer(f)
dec := gob.NewDecoder(buf)
err = dec.Decode(&sg)
if err != nil {
b.Error(err)
}
b.ResetTimer()
for i := 0; i < b.N; i++ {
if _, err := sg.ToJson(&l); err != nil {
b.Fatal(err)
}
}
}
func BenchmarkToJson(b *testing.B) {
benchmarkToJson("benchmark/actors10.txt", b)
benchmarkToJson("benchmark/actors100.txt", b)
benchmarkToJson("benchmark/actors1000.txt", b)
benchmarkToJson("benchmark/directors10.txt", b)
benchmarkToJson("benchmark/directors100.txt", b)
benchmarkToJson("benchmark/directors1000.txt", b)
}
func benchmarkToProtocolBuffer(file string, b *testing.B) {
b.ReportAllocs()
var sg SubGraph
var l Latency
f, err := ioutil.ReadFile(file)
if err != nil {
b.Error(err)
}
buf := bytes.NewBuffer(f)
dec := gob.NewDecoder(buf)
err = dec.Decode(&sg)
if err != nil {
b.Error(err)
}
b.ResetTimer()
for i := 0; i < b.N; i++ {
if _, err := sg.ToProtocolBuffer(&l); err != nil {
b.Fatal(err)
}
}
}
func BenchmarkToProtocolBuffer(b *testing.B) {
benchmarkToProtocolBuffer("benchmark/actors10.txt", b)
benchmarkToProtocolBuffer("benchmark/actors100.txt", b)
benchmarkToProtocolBuffer("benchmark/actors1000.txt", b)
benchmarkToProtocolBuffer("benchmark/directors10.txt", b)
benchmarkToProtocolBuffer("benchmark/directors100.txt", b)
benchmarkToProtocolBuffer("benchmark/directors1000.txt", b)
}