From 94881a2555b8874a3fb4bacd01a66acd74f15f4b Mon Sep 17 00:00:00 2001
From: Andrea Angiolillo <andrea.angiolillo@mongodb.com>
Date: Mon, 15 May 2023 15:40:22 +0100
Subject: [PATCH] INTMDB-801: Add Missing fields in the data federation store
 struct (#478)

---
 mongodbatlas/data_federation.go      |  7 ++++++-
 mongodbatlas/data_federation_test.go | 14 +++++++++++++-
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/mongodbatlas/data_federation.go b/mongodbatlas/data_federation.go
index 0fb145f23..537611b96 100644
--- a/mongodbatlas/data_federation.go
+++ b/mongodbatlas/data_federation.go
@@ -95,16 +95,21 @@ type DataFederationDatabaseView struct {
 // DataFederationStore represents data stores for the data federation.
 type DataFederationStore struct {
 	ReadPreference           *ReadPreference `json:"readPreference,omitempty"`
-	IncludeTags              *bool           `json:"includeTags,omitempty"`
 	AdditionalStorageClasses []*string       `json:"additionalStorageClasses,omitempty"`
+	Urls                     []*string       `json:"urls,omitempty"`
 	Name                     string          `json:"name,omitempty"`
 	Provider                 string          `json:"provider,omitempty"`
 	ClusterName              string          `json:"clusterName,omitempty"`
+	ClusterID                string          `json:"clusterId,omitempty"`
 	Region                   string          `json:"region,omitempty"`
 	Bucket                   string          `json:"bucket,omitempty"`
 	Prefix                   string          `json:"prefix,omitempty"`
 	Delimiter                string          `json:"delimiter,omitempty"`
 	ProjectID                string          `json:"projectId,omitempty"`
+	DefaultFormat            string          `json:"defaultFormat,omitempty"`
+	IncludeTags              *bool           `json:"includeTags,omitempty"`
+	Public                   *bool           `json:"public,omitempty"`
+	AllowInsecure            *bool           `json:"allowInsecure,omitempty"`
 }
 
 // ReadPreference describes how to route read requests to the cluster.
diff --git a/mongodbatlas/data_federation_test.go b/mongodbatlas/data_federation_test.go
index dc5aa0917..3f6325bdb 100644
--- a/mongodbatlas/data_federation_test.go
+++ b/mongodbatlas/data_federation_test.go
@@ -239,7 +239,13 @@ func TestDataFederation_Get(t *testing.T) {
 						  	"prefix": "/metrics",
 						  	"delimiter": "/",
 						  	"includeTags": false,
-							"additionalStorageClasses" : ["STANDARD_IA"]
+							"additionalStorageClasses" : ["STANDARD_IA"],
+							"clusterName": "test",
+							"clusterId": "test",
+							"public": false,
+							"allowInsecure": false,
+							"defaultFormat": "test",
+							"urls": ["test"]
 						}
 					]
 				}
@@ -308,6 +314,12 @@ func TestDataFederation_Get(t *testing.T) {
 					Delimiter:                "/",
 					IncludeTags:              pointer(false),
 					AdditionalStorageClasses: []*string{pointer("STANDARD_IA")},
+					ClusterName:              "test",
+					ClusterID:                "test",
+					DefaultFormat:            "test",
+					Public:                   pointer(false),
+					AllowInsecure:            pointer(false),
+					Urls:                     []*string{pointer("test")},
 				},
 			},
 		},