From d9e7f372800e3650f9d328fbf4d817a34cd0810d Mon Sep 17 00:00:00 2001
From: Kristoffer Dalby <kradalby@kradalby.no>
Date: Fri, 25 Feb 2022 10:27:27 +0100
Subject: [PATCH] Uncomment previous test and update them for no boundries

---
 dns_test.go        | 27 ++++++++++-----------
 namespaces_test.go | 58 ++++++++++++++++++++++------------------------
 2 files changed, 42 insertions(+), 43 deletions(-)

diff --git a/dns_test.go b/dns_test.go
index b81bc7f..23a34ca 100644
--- a/dns_test.go
+++ b/dns_test.go
@@ -1,6 +1,8 @@
 package headscale
 
 import (
+	"fmt"
+
 	"gopkg.in/check.v1"
 	"inet.af/netaddr"
 	"tailscale.com/tailcfg"
@@ -241,20 +243,19 @@ func (s *Suite) TestDNSConfigMapResponseWithMagicDNS(c *check.C) {
 	)
 	c.Assert(dnsConfig, check.NotNil)
 
-	// TODO: Remove comment out when we have all nodes available to every node
-	// c.Assert(len(dnsConfig.Routes), check.Equals, 2)
+	c.Assert(len(dnsConfig.Routes), check.Equals, 3)
 
-	// domainRouteShared1 := fmt.Sprintf("%s.%s", namespaceShared1.Name, baseDomain)
-	// _, ok := dnsConfig.Routes[domainRouteShared1]
-	// c.Assert(ok, check.Equals, true)
-	//
-	// domainRouteShared2 := fmt.Sprintf("%s.%s", namespaceShared2.Name, baseDomain)
-	// _, ok = dnsConfig.Routes[domainRouteShared2]
-	// c.Assert(ok, check.Equals, true)
-	//
-	// domainRouteShared3 := fmt.Sprintf("%s.%s", namespaceShared3.Name, baseDomain)
-	// _, ok = dnsConfig.Routes[domainRouteShared3]
-	// c.Assert(ok, check.Equals, false)
+	domainRouteShared1 := fmt.Sprintf("%s.%s", namespaceShared1.Name, baseDomain)
+	_, ok := dnsConfig.Routes[domainRouteShared1]
+	c.Assert(ok, check.Equals, true)
+
+	domainRouteShared2 := fmt.Sprintf("%s.%s", namespaceShared2.Name, baseDomain)
+	_, ok = dnsConfig.Routes[domainRouteShared2]
+	c.Assert(ok, check.Equals, true)
+
+	domainRouteShared3 := fmt.Sprintf("%s.%s", namespaceShared3.Name, baseDomain)
+	_, ok = dnsConfig.Routes[domainRouteShared3]
+	c.Assert(ok, check.Equals, true)
 }
 
 func (s *Suite) TestDNSConfigMapResponseWithoutMagicDNS(c *check.C) {
diff --git a/namespaces_test.go b/namespaces_test.go
index f7b345b..77c33ad 100644
--- a/namespaces_test.go
+++ b/namespaces_test.go
@@ -205,35 +205,33 @@ func (s *Suite) TestGetMapResponseUserProfiles(c *check.C) {
 	}
 	app.db.Save(machine2InShared1)
 
-	// TODO: Remove comment out when we have all nodes available to every node
-	// peersOfMachine1InShared1, err := app.getPeers(machineInShared1)
-	// c.Assert(err, check.IsNil)
+	peersOfMachine1InShared1, err := app.getPeers(machineInShared1)
+	c.Assert(err, check.IsNil)
 
-	// userProfiles := getMapResponseUserProfiles(
-	// 	*machineInShared1,
-	// 	peersOfMachine1InShared1,
-	// )
-	//
-	// log.Trace().Msgf("userProfiles %#v", userProfiles)
-	// c.Assert(len(userProfiles), check.Equals, 2)
-	//
-	// found := false
-	// for _, userProfiles := range userProfiles {
-	// 	if userProfiles.DisplayName == namespaceShared1.Name {
-	// 		found = true
-	//
-	// 		break
-	// 	}
-	// }
-	// c.Assert(found, check.Equals, true)
-	//
-	// found = false
-	// for _, userProfile := range userProfiles {
-	// 	if userProfile.DisplayName == namespaceShared2.Name {
-	// 		found = true
-	//
-	// 		break
-	// 	}
-	// }
-	// c.Assert(found, check.Equals, true)
+	userProfiles := getMapResponseUserProfiles(
+		*machineInShared1,
+		peersOfMachine1InShared1,
+	)
+
+	c.Assert(len(userProfiles), check.Equals, 3)
+
+	found := false
+	for _, userProfiles := range userProfiles {
+		if userProfiles.DisplayName == namespaceShared1.Name {
+			found = true
+
+			break
+		}
+	}
+	c.Assert(found, check.Equals, true)
+
+	found = false
+	for _, userProfile := range userProfiles {
+		if userProfile.DisplayName == namespaceShared2.Name {
+			found = true
+
+			break
+		}
+	}
+	c.Assert(found, check.Equals, true)
 }