Update some tests

cyralinc · Jan 30, 2024 · 26ce92e · 26ce92e
1 parent 982edc7
commit 26ce92e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cyral/internal/repository/resource_cyral_repository_test.go b/cyral/internal/repository/resource_cyral_repository_test.go
@@ -115,6 +115,7 @@ var (
 		MongoDBSettings: &repository.MongoDBSettings{
 			ReplicaSetName: "some-replica-set",
 			ServerType:     repository.ReplicaSet,
+			Flavor:         "mongodb",
 		},
 	}
 
@@ -136,6 +137,7 @@ var (
 			ReplicaSetName: "myReplicaSet",
 			ServerType:     "replicaset",
 			SRVRecordName:  "mySRVRecord",
+			Flavor:         "documentdb",
 		},
 	}
 )
@@ -236,6 +238,13 @@ func repoCheckFuctions(repo repository.RepoInfo, resName string) resource.TestCh
 					repo.MongoDBSettings.ReplicaSetName),
 			}...)
 		}
+		if repo.MongoDBSettings.Flavor != "" {
+			checkFuncs = append(checkFuncs, []resource.TestCheckFunc{
+				resource.TestCheckResourceAttr(resourceFullName,
+					"mongodb_settings.0.flavor",
+					repo.MongoDBSettings.Flavor),
+			}...)
+		}
 		checkFuncs = append(checkFuncs, []resource.TestCheckFunc{
 			resource.TestCheckResourceAttr(resourceFullName,
 				"mongodb_settings.0.server_type",