main_test.go

package main

import (
	"fmt"
	"io/ioutil"
	"math"
	"os"
	"testing"

	"github.com/Sirupsen/logrus"
	"github.com/dgraph-io/dgraph/commit"
	"github.com/dgraph-io/dgraph/posting"
	"github.com/dgraph-io/dgraph/rdf"
	"github.com/dgraph-io/dgraph/store"
	"github.com/dgryski/go-farm"
)

func TestQuery(t *testing.T) {

	var numInstances uint64 = 2

	var mod uint64 = math.MaxUint64 / numInstances
	var minIdx0 uint64 = 0 * mod
	var minIdx1 uint64 = 1 * mod

	logrus.SetLevel(logrus.DebugLevel)

	dir, err := ioutil.TempDir("", "storetest_")
	if err != nil {
		t.Error(err)
		return
	}
	defer os.RemoveAll(dir)
	ps := new(store.Store)
	ps.Init(dir)
	clog := commit.NewLogger(dir, "mutations", 50<<20)
	clog.Init()
	defer clog.Close()

	posting.Init(ps, clog)

	list := []string{"alice", "bob", "mallory", "ash", "man", "dgraph"}

	for _, str := range list {
		if farm.Fingerprint64([]byte(str))%numInstances == 0 {
			uid, err := rdf.GetUid(str, 0, numInstances)
			if uid < minIdx0 || uid > minIdx0+mod-1 {
				t.Error("Not the correct UID", err)
			}
			fmt.Println("Instance-0", str, uid)
			t.Logf("Correct UID")
		} else {
			uid, err := rdf.GetUid(str, 1, numInstances)
			if uid < minIdx1 || uid > minIdx1+mod-1 {
				t.Error("Not the correct UID", err)
			}
			fmt.Println("Instance-1", str, uid)
			t.Logf("Correct UID")
		}
	}
}