From ed6ca6c5589015ebe2734ddfb5d8bd0b4ee0d5dc Mon Sep 17 00:00:00 2001
From: childe <heychilde@gmail.com>
Date: Tue, 6 Mar 2018 10:29:39 +0800
Subject: [PATCH] new file:   assign_test.go

---
 assign_test.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 assign_test.go

diff --git a/assign_test.go b/assign_test.go
new file mode 100644
index 0000000..bf6cf0e
--- /dev/null
+++ b/assign_test.go
@@ -0,0 +1,39 @@
+package healer
+
+import "testing"
+
+func TestAssgin(t *testing.T) {
+	var (
+		s          = &RangeAssignmentStrategy{}
+		members    []string
+		partitions []int32
+		rst        map[string][]int32
+	)
+	members = []string{"1", "2"}
+	partitions = []int32{0, 1, 2, 3, 4}
+	rst = s.assignPartitions(members, partitions)
+
+	if len(rst) != 2 {
+		t.Error("rst length != 2")
+	}
+	if len(rst["1"]) != 3 {
+		t.Error("partitions in memeber 1 != 3")
+	}
+	if len(rst["2"]) != 2 {
+		t.Error("partitions in memeber 2 != 2")
+	}
+
+	members = []string{"1", "2"}
+	partitions = []int32{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
+	rst = s.assignPartitions(members, partitions)
+
+	if len(rst) != 2 {
+		t.Error("rst length != 2")
+	}
+	if len(rst["1"]) != 5 {
+		t.Error("partitions in memeber 1 != 5")
+	}
+	if len(rst["2"]) != 5 {
+		t.Error("partitions in memeber 2 != 5")
+	}
+}