Added IPv6 tests

schollz · Feb 6, 2019 · e947981 · e947981
1 parent 81fc856
commit e947981
Showing 1 changed file with 38 additions and 32 deletions.
diff --git a/peerdiscovery_test.go b/peerdiscovery_test.go
@@ -8,42 +8,48 @@ import (
 )
 
 func TestDiscovery(t *testing.T) {
-	// should not be able to "discover" itself
-	discoveries, err := Discover()
-	assert.Nil(t, err)
-	assert.Zero(t, len(discoveries))
-
-	// should be able to "discover" itself
-	discoveries, err = Discover(Settings{
-		Limit:     -1,
-		AllowSelf: true,
-		Payload:   []byte("payload"),
-		Delay:     500 * time.Millisecond,
-		TimeLimit: 1 * time.Second,
-	})
-	assert.Nil(t, err)
-	assert.NotZero(t, len(discoveries))
-}
+	for _, version := range []IPVersion{IPv4, IPv6} {
+		// should not be able to "discover" itself
+		discoveries, err := Discover()
+		assert.Nil(t, err)
+		assert.Zero(t, len(discoveries))
 
-func TestDiscoverySelf(t *testing.T) {
-	// broadcast self to self
-	go func() {
-		_, err := Discover(Settings{
+		// should be able to "discover" itself
+		discoveries, err = Discover(Settings{
 			Limit:     -1,
+			AllowSelf: true,
 			Payload:   []byte("payload"),
-			Delay:     10 * time.Millisecond,
+			Delay:     500 * time.Millisecond,
 			TimeLimit: 1 * time.Second,
+			IPVersion: version,
+		})
+		assert.Nil(t, err)
+		assert.NotZero(t, len(discoveries))
+	}
+}
+
+func TestDiscoverySelf(t *testing.T) {
+	for _, version := range []IPVersion{IPv4, IPv6} {
+		// broadcast self to self
+		go func() {
+			_, err := Discover(Settings{
+				Limit:     -1,
+				Payload:   []byte("payload"),
+				Delay:     10 * time.Millisecond,
+				TimeLimit: 1 * time.Second,
+				IPVersion: version,
+			})
+			assert.Nil(t, err)
+		}()
+		discoveries, err := Discover(Settings{
+			Limit:            1,
+			Payload:          []byte("payload"),
+			Delay:            500 * time.Millisecond,
+			TimeLimit:        1 * time.Second,
+			DisableBroadcast: true,
+			AllowSelf:        true,
 		})
 		assert.Nil(t, err)
-	}()
-	discoveries, err := Discover(Settings{
-		Limit:            1,
-		Payload:          []byte("payload"),
-		Delay:            500 * time.Millisecond,
-		TimeLimit:        1 * time.Second,
-		DisableBroadcast: true,
-		AllowSelf:        true,
-	})
-	assert.Nil(t, err)
-	assert.NotZero(t, len(discoveries))
+		assert.NotZero(t, len(discoveries))
+	}
 }