fixed and tested

2 files changed, 2 insertions, 6 deletions

diff --git a/srv.go b/srv.go
index 78f1975..15ea912 100644
--- a/srv.go
+++ b/srv.go
@@ -1,10 +1,8 @@
 package dnsutils
 
 import (
-	"fmt"
 	"math/rand"
 	"net"
-	"os"
 	"sort"
 )
 
@@ -47,10 +45,8 @@ func orderSRV(addrs []*net.SRV) (int, map[int]*net.SRV) {
 	var count int
 	sort.Ints(priorities)
 	for _, p := range priorities {
-		fmt.Fprintf(os.Stderr, "priority: %v\n", p)
 		tos := weightedOrder(prioMap[p])
 		for i, s := range tos {
-			fmt.Fprintf(os.Stderr, "srv: %v\n", s.Target)
 			count += 1
 			osrv[o+i] = s
 		}
diff --git a/srv_integration_test.go b/srv_integration_test.go
index c3d1924..6ba8f51 100644
--- a/srv_integration_test.go
+++ b/srv_integration_test.go
@@ -16,8 +16,6 @@ func TestResolveKDC(t *testing.T) {
 		if err != nil {
 			t.Errorf("error resolving SRV DNS records: %v", err)
 		}
-		assert.Equal(t, 6, count, "Number of SRV records not as expected: %v", res)
-		assert.Equal(t, count, len(res), "Map size does not match: %v", res)
 		expected := []string{
 			"kdc.test.gokrb5:88",
 			"kdc1a.test.gokrb5:88",
@@ -25,6 +23,8 @@ func TestResolveKDC(t *testing.T) {
 			"kdc1b.test.gokrb5:88",
 			"kdc2b.test.gokrb5:88",
 		}
+		assert.Equal(t, len(expected), count, "Number of SRV records not as expected: %v", res)
+		assert.Equal(t, count, len(res), "Map size does not match: %v", res)
 		for _, s := range expected {
 			var found bool
 			for _, v := range res {