From ef7f44e4bb7713b18e6c0ab1b6e3510075b6623b Mon Sep 17 00:00:00 2001 From: jwijenbergh Date: Wed, 22 May 2024 14:32:38 +0200 Subject: Discovery: return country_code to the client --- internal/discovery/discovery_test.go | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'internal/discovery/discovery_test.go') diff --git a/internal/discovery/discovery_test.go b/internal/discovery/discovery_test.go index c2bbf44..03ea109 100644 --- a/internal/discovery/discovery_test.go +++ b/internal/discovery/discovery_test.go @@ -90,12 +90,12 @@ func TestSecureLocationList(t *testing.T) { Version: 1, List: []Server{ // institute access server, this should not be found - {Server: discotypes.Server{Type: "institute_access"}, CountryCode: ""}, + {Server: discotypes.Server{Type: "institute_access"}}, // secure internet servers, these should be found - {Server: discotypes.Server{Type: "secure_internet"}, CountryCode: "b"}, - {Server: discotypes.Server{Type: "secure_internet"}, CountryCode: "c"}, + {Server: discotypes.Server{Type: "secure_internet", CountryCode: "b"}}, + {Server: discotypes.Server{Type: "secure_internet", CountryCode: "c"}}, // Unexpected type, this should not be found - {Server: discotypes.Server{Type: "test"}, CountryCode: "d"}, + {Server: discotypes.Server{Type: "test", CountryCode: "d"}}, }, }, } @@ -150,7 +150,7 @@ func TestServerByURL(t *testing.T) { // TestServerByCountryCode tests the function for getting a server by the country code func TestServerByCountryCode(t *testing.T) { - s1 := Server{Server: discotypes.Server{Type: "secure_internet"}, CountryCode: "a"} + s1 := Server{Server: discotypes.Server{Type: "secure_internet", CountryCode: "a"}} d := Discovery{ ServerList: Servers{ Version: 1, @@ -158,8 +158,8 @@ func TestServerByCountryCode(t *testing.T) { // secure internet server s1, // Unexpected types, these should not be found - {Server: discotypes.Server{Type: "institute_access"}, CountryCode: "b"}, - {Server: discotypes.Server{Type: "test"}, CountryCode: "c"}, + {Server: discotypes.Server{Type: "institute_access", CountryCode: "b"}}, + {Server: discotypes.Server{Type: "test", CountryCode: "c"}}, }, }, } -- cgit v1.2.3