From ea17e2368a16659353b7ab5453a9e8eba6df625b Mon Sep 17 00:00:00 2001
From: Ashwin <ashwin2007ray@gmail.com>
Date: Thu, 4 Feb 2016 08:04:01 +0000
Subject: [PATCH] unit test for uidassigner (changed)
---
server/uidassigner/main_test.go | 20 ++++++++++++++------
1 file changed, 14 insertions(+), 6 deletions(-)
diff --git a/server/uidassigner/main_test.go b/server/uidassigner/main_test.go
index 91fb6537..13a16176 100644
--- a/server/uidassigner/main_test.go
+++ b/server/uidassigner/main_test.go
@@ -1,6 +1,7 @@
package main
import (
+ "fmt"
"io/ioutil"
"math"
"os"
@@ -15,11 +16,12 @@ import (
)
func TestQuery(t *testing.T) {
- var instanceIdx uint64 = 0
+
var numInstances uint64 = 2
var mod uint64 = math.MaxUint64 / numInstances
- var minIdx uint64 = instanceIdx * mod
+ var minIdx0 uint64 = 0 * mod
+ var minIdx1 uint64 = 1 * mod
logrus.SetLevel(logrus.DebugLevel)
@@ -40,13 +42,19 @@ func TestQuery(t *testing.T) {
list := []string{"alice", "bob", "mallory", "ash", "man", "dgraph"}
for _, str := range list {
- if farm.Fingerprint64([]byte(str))%numInstances != instanceIdx {
- continue
+ 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, instanceIdx, numInstances)
- if uid < minIdx || uid > minIdx+mod-1 {
+ 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")
}
}
--
GitLab