str comparison

1 files changed, 4 insertions, 1 deletions

diff --git a/srv_integration_test.go b/srv_integration_test.go
index eb25220..a26dc85 100644
--- a/srv_integration_test.go
+++ b/srv_integration_test.go
@@ -5,6 +5,8 @@ package dnsutils
 
 import (
 	"github.com/stretchr/testify/assert"
+	"strconv"
+	"strings"
 	"testing"
 )
 
@@ -26,7 +28,8 @@ func TestResolveKDC(t *testing.T) {
 	for _, s := range expected {
 		var found bool
 		for _, v := range res {
-			if s == v {
+			srvStr := strings.TrimRight(v.Target, ".") + ":" + strconv.Itoa(int(v.Port))
+			if s == srvStr {
 				found = true
 				break
 			}