Browse Source

etcdhttp: use peer.String() in test

Xiang Li 11 years ago
parent
commit
908d326e22
1 changed files with 9 additions and 11 deletions
  1. 9 11
      etcdserver/etcdhttp/peers_test.go

+ 9 - 11
etcdserver/etcdhttp/peers_test.go

@@ -5,17 +5,15 @@ import "testing"
 //TODO: full testing for peer set
 func TestPeerSet(t *testing.T) {
 	p := &Peers{}
-
-	p.Set("0x1=1.1.1.1")
-	if p.Pick(0x1) != "http://1.1.1.1" {
-		t.Errorf("pick = %s, want http://1.1.1.1", p.Pick(0x1))
-	}
-
-	p.Set("0x2=2.2.2.2")
-	if p.Pick(0x1) != "" {
-		t.Errorf("pick = %s, want empty string", p.Pick(0x1))
+	tests := []string{
+		"1=1.1.1.1",
+		"2=2.2.2.2",
+		"1=1.1.1.1&1=1.1.1.2&2=2.2.2.2",
 	}
-	if p.Pick(0x2) != "http://2.2.2.2" {
-		t.Errorf("pick = %s, want http://2.2.2.2", p.Pick(0x2))
+	for i, tt := range tests {
+		p.Set(tt)
+		if p.String() != tt {
+			t.Errorf("#%d: string = %s, want %s", i, p.String(), tt)
+		}
 	}
 }