diff --git a/server/uidassigner/main_test.go b/server/uidassigner/main_test.go index 13a161769e75ef2c1ff298122f3201a93dcf8fb9..798d7e58f1f397bc647127b8f335b23fa90d3b6d 100644 --- a/server/uidassigner/main_test.go +++ b/server/uidassigner/main_test.go @@ -1,7 +1,6 @@ package main import ( - "fmt" "io/ioutil" "math" "os" @@ -16,45 +15,42 @@ import ( ) 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 + mod := math.MaxUint64 / numInstances + minIdx0 := 0 * mod + minIdx1 := 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("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("Instance-1", str, uid) t.Logf("Correct UID") } }