Update test values and embed location struct in peer system

netbirdio · Jan 30, 2024 · 00c7c00 · 00c7c00
1 parent 36b58d8
commit 00c7c00
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/management/server/http/posture_checks_handler_test.go b/management/server/http/posture_checks_handler_test.go
@@ -444,8 +444,8 @@ func TestPostureCheckUpdate(t *testing.T) {
 						"geo_location_check": {
 							"locations": [
 								{
-									"city_name": "test",
-									"country_code": "TE"
+									"city_name": "Los Angeles",
+									"country_code": "US"
 								}
 							],
 							"action": "allow"
@@ -462,8 +462,8 @@ func TestPostureCheckUpdate(t *testing.T) {
 					GeoLocationCheck: &api.GeoLocationCheck{
 						Locations: []api.Location{
 							{
-								CityName:    "test",
-								CountryCode: "TE",
+								CityName:    "Los Angeles",
+								CountryCode: "US",
 							},
 						},
 						Action: api.GeoLocationCheckActionAllow,

diff --git a/management/server/peer/peer.go b/management/server/peer/peer.go
@@ -71,7 +71,7 @@ type PeerSystemMeta struct {
 	Location struct {
 		CountryCode string
 		CityName    string
-	}
+	} `gorm:"embedded;embeddedPrefix:location_"`
 }
 
 func (p PeerSystemMeta) isEqual(other PeerSystemMeta) bool {