@@ -87,7 +87,7 @@ func DeleteDocs(client *req.Client, deleteEndpoint string) (string, error) {
8787 return result , err
8888}
8989
90- func RunRequests (client * req.Client , dbReq string , hostsEndpoint string ) (string , error ) {
90+ func RunRequests (t * testing. T , client * req.Client , dbReq string , hostsEndpoint string ) (string , error ) {
9191 var err error
9292 var body []byte
9393 headerMap := map [string ]string {
@@ -99,11 +99,11 @@ func RunRequests(client *req.Client, dbReq string, hostsEndpoint string) (string
9999 operation := (gjson .Get (dbReq , `operation` )).Str
100100 var retryFn = (func (resp * req.Response , err error ) bool {
101101 if err != nil {
102- fmt . Println (err .Error ())
102+ t . Fatalf (err .Error ())
103103 }
104104 body , err = io .ReadAll (resp .Body )
105105 if err != nil {
106- fmt . Println (err .Error ())
106+ t . Fatalf (err .Error ())
107107 }
108108 result = (string (body ))
109109 return true
@@ -112,7 +112,7 @@ func RunRequests(client *req.Client, dbReq string, hostsEndpoint string) (string
112112 if operation == "backup-status" {
113113 retryFn = (func (resp * req.Response , err error ) bool {
114114 if err != nil {
115- fmt . Printf ( "error: %s" , err .Error ())
115+ t . Fatalf ( err .Error ())
116116 }
117117 body , _ := io .ReadAll (resp .Body )
118118 status = (gjson .Get (string (body ), `status` )).Str
@@ -130,7 +130,7 @@ func RunRequests(client *req.Client, dbReq string, hostsEndpoint string) (string
130130 SetBodyString (dbReq ).
131131 Post (hostsEndpoint )
132132 if err != nil {
133- return "" , err
133+ t . Fatalf ( err . Error ())
134134 }
135135 defer resp .Body .Close ()
136136
@@ -158,7 +158,7 @@ func TestMlDbBackupRestore(t *testing.T) {
158158 }
159159
160160 if ! tagPres {
161- imageTag = "11.1 .0-centos-1.1.2 "
161+ imageTag = "11.3 .0-ubi-rootless "
162162 t .Logf ("No imageTag variable present, setting to default value: " + imageTag )
163163 }
164164
@@ -167,18 +167,20 @@ func TestMlDbBackupRestore(t *testing.T) {
167167
168168 namespaceName := "ml-" + strings .ToLower (random .UniqueId ())
169169 kubectlOptions := k8s .NewKubectlOptions ("" , "" , namespaceName )
170+ valuesMap := map [string ]string {
171+ "persistence.enabled" : "true" ,
172+ "replicaCount" : "1" ,
173+ "image.repository" : imageRepo ,
174+ "image.tag" : imageTag ,
175+ "auth.adminUsername" : username ,
176+ "auth.adminPassword" : password ,
177+ "logCollection.enabled" : "false" ,
178+ }
179+
170180 options := & helm.Options {
171181 KubectlOptions : kubectlOptions ,
172- SetValues : map [string ]string {
173- "persistence.enabled" : "true" ,
174- "replicaCount" : "1" ,
175- "image.repository" : imageRepo ,
176- "image.tag" : imageTag ,
177- "auth.adminUsername" : username ,
178- "auth.adminPassword" : password ,
179- "logCollection.enabled" : "false" ,
180- },
181- Version : initialChartVersion ,
182+ SetValues : valuesMap ,
183+ Version : initialChartVersion ,
182184 }
183185
184186 t .Logf ("====Installing Helm Chart" )
@@ -198,6 +200,8 @@ func TestMlDbBackupRestore(t *testing.T) {
198200 //add the helm chart repo and install the last helm chart release from repository
199201 //to test and upgrade this chart to the latest one to be released
200202 if runUpgradeTest {
203+ delete (valuesMap , "image.repository" )
204+ delete (valuesMap , "image.tag" )
201205 helm .AddRepo (t , options , "marklogic" , "https://marklogic.github.io/marklogic-kubernetes/" )
202206 defer helm .RemoveRepo (t , options , "marklogic" )
203207 helmChartPath = "marklogic/marklogic"
@@ -236,6 +240,15 @@ func TestMlDbBackupRestore(t *testing.T) {
236240 testUtil .HelmUpgrade (t , helmUpgradeOptions , releaseName , kubectlOptions , []string {podName }, initialChartVersion )
237241 }
238242
243+ // wait until the pod is in Running status
244+ output , err := testUtil .WaitUntilPodRunning (t , kubectlOptions , podName , 10 , 15 * time .Second )
245+ if err != nil {
246+ t .Error (err .Error ())
247+ }
248+ if output != "Running" {
249+ t .Error (output )
250+ }
251+
239252 //create backup directories and setup permissions
240253 k8s .RunKubectl (t , kubectlOptions , "exec" , podName , "--" , "/bin/bash" , "-c" , "cd /tmp && mkdir backup && chmod 777 backup && mkdir backup/incrBackup && chmod 777 backup/incrBackup" )
241254
@@ -287,8 +300,9 @@ func TestMlDbBackupRestore(t *testing.T) {
287300 IncludeReplicas : "true" }
288301 bkupReqRes , _ := json .Marshal (bkupReq )
289302
303+ t .Log ("====Full backup for Documents DB" )
290304 //full backup for Documents DB
291- result , err = RunRequests (client , string (bkupReqRes ), manageEndpoint )
305+ result , err = RunRequests (t , client , string (bkupReqRes ), manageEndpoint )
292306 if err != nil {
293307 t .Fatalf (err .Error ())
294308 }
@@ -302,7 +316,7 @@ func TestMlDbBackupRestore(t *testing.T) {
302316 bkupStatusReqRes , _ := json .Marshal (bkupStatusReq )
303317
304318 //get status of full backup job
305- result , err = RunRequests (client , string (bkupStatusReqRes ), manageEndpoint )
319+ result , err = RunRequests (t , client , string (bkupStatusReqRes ), manageEndpoint )
306320 if err != nil {
307321 t .Fatalf (err .Error ())
308322 }
@@ -311,10 +325,12 @@ func TestMlDbBackupRestore(t *testing.T) {
311325 //verify full backup is completed
312326 assert .Equal (t , "completed" , bkupStatus )
313327
328+ t .Log ("====Delete a document from Documents DB" )
314329 deleteEndpoint := fmt .Sprintf ("http://%s/v1/documents?database=Documents&uri=%s" , tunnel8000 .Endpoint (), docs [1 ])
315330 fmt .Println (deleteEndpoint )
316331
317332 //incremental backup
333+ t .Log ("====Incremental backup for Documents DB" )
318334 incrBkupReq := & BackupRestoreReq {
319335 Operation : "backup-database" ,
320336 BackupDir : "/tmp/backup" ,
@@ -324,7 +340,7 @@ func TestMlDbBackupRestore(t *testing.T) {
324340 incrBkupReqRes , _ := json .Marshal (incrBkupReq )
325341
326342 //incremnetal backup for Documents DB
327- result , err = RunRequests (client , string (incrBkupReqRes ), manageEndpoint )
343+ result , err = RunRequests (t , client , string (incrBkupReqRes ), manageEndpoint )
328344 if err != nil {
329345 t .Fatalf (err .Error ())
330346 }
@@ -338,7 +354,7 @@ func TestMlDbBackupRestore(t *testing.T) {
338354 incrBkupStatusReqRes , _ := json .Marshal (incrBkupStatusReq )
339355
340356 //get status of backup job
341- result , err = RunRequests (client , string (incrBkupStatusReqRes ), manageEndpoint )
357+ result , err = RunRequests (t , client , string (incrBkupStatusReqRes ), manageEndpoint )
342358 if err != nil {
343359 t .Fatalf (err .Error ())
344360 }
@@ -361,7 +377,7 @@ func TestMlDbBackupRestore(t *testing.T) {
361377 brstrReqRes , _ := json .Marshal (rstrReq )
362378
363379 //restore Documents DB from incremental backup
364- result , err = RunRequests (client , string (brstrReqRes ), manageEndpoint )
380+ result , err = RunRequests (t , client , string (brstrReqRes ), manageEndpoint )
365381 if err != nil {
366382 t .Fatalf (err .Error ())
367383 }
0 commit comments