feat: validate the cc when using city params

speedtest/server.go

@@ -51,6 +51,7 @@ type Server struct {
 	DLSpeed      float64       `json:"dl_speed"`
 	ULSpeed      float64       `json:"ul_speed"`
 	TestDuration TestDuration  `json:"test_duration"`
+	CC           string        `json:"cc"`
 
 	Context *Speedtest `json:"-"`
 }
@@ -254,6 +255,15 @@ func (s *Speedtest) FetchServerListContext(ctx context.Context) (Servers, error)
 		if err = decoder.Decode(&servers); err != nil {
 			return servers, err
 		}
+		if s.config.Location != nil {
+			var tmpServers Servers
+			for _, server := range servers {
+				if server.CC == strings.ToUpper(s.config.Location.CC) {
+					tmpServers = append(tmpServers, server)
+				}
+			}
+			servers = tmpServers
+		}
 	case typeXMLPayload:
 		var list ServerList
 		// Decode xml

speedtest/server_test.go

@@ -3,6 +3,7 @@ package speedtest
 import (
 	"math"
 	"math/rand"
+	"strings"
 	"testing"
 	"time"
 )
@@ -195,3 +196,19 @@ func TestTotalDurationCount(t *testing.T) {
 		t.Error("addition in testDurationTotalCount didn't work")
 	}
 }
+
+func TestCityFlag(t *testing.T) {
+	client := New(WithUserConfig(&UserConfig{CityFlag: "yishun"}))
+	servers, err := client.FetchServers()
+	if err != nil {
+		t.Errorf(err.Error())
+	}
+	row := 1
+	for _, server := range servers {
+
+		if server.CC != strings.ToUpper(client.config.Location.CC) {
+			t.Error("server country code does not match client country code")
+		}
+		row++
+	}
+}