@@ -374,33 +374,58 @@ func TestSmokeLoad_CustomPartition_WithK6Stream(t *testing.T) {
374
374
// DeleteStream(t, NewGlob.QueryClient, custom_partition_stream)
375
375
// }
376
376
377
+ func TestSmokeSetTarget (t * testing.T ) {
378
+ CreateStream (t , NewGlob .QueryClient , NewGlob .Stream )
379
+ if NewGlob .IngestorUrl .String () == "" {
380
+ // RunFlog(t, NewGlob.QueryClient, NewGlob.Stream)
381
+ body := getTargetBody ()
382
+ req , _ := NewGlob .QueryClient .NewRequest ("POST" , "/targets" , strings .NewReader (body ))
383
+ response , err := NewGlob .QueryClient .Do (req )
384
+ require .NoErrorf (t , err , "Request failed: %s" , err )
385
+ require .Equalf (t , 200 , response .StatusCode , "Server returned http code: %s and response: %s" , response .Status , readAsString (response .Body ))
386
+ }
387
+ }
388
+
377
389
func TestSmokeSetAlert (t * testing.T ) {
378
390
CreateStream (t , NewGlob .QueryClient , NewGlob .Stream )
379
391
if NewGlob .IngestorUrl .String () == "" {
380
392
RunFlog (t , NewGlob .QueryClient , NewGlob .Stream )
381
- body := getAlertBody (NewGlob .Stream )
382
- req , _ := NewGlob .QueryClient .NewRequest ("POST" , "/alerts" , strings .NewReader (body ))
393
+ req , _ := NewGlob .QueryClient .NewRequest ("GET" , "/targets" , nil )
383
394
response , err := NewGlob .QueryClient .Do (req )
384
395
require .NoErrorf (t , err , "Request failed: %s" , err )
396
+ bodyTargets , _ := io .ReadAll (response .Body )
397
+ reader1 := bytes .NewReader (bodyTargets )
398
+ targetId := getIdFromTargetResponse (reader1 )
399
+ body := getAlertBody (NewGlob .Stream , targetId )
400
+ req , _ = NewGlob .QueryClient .NewRequest ("POST" , "/alerts" , strings .NewReader (body ))
401
+ response , err = NewGlob .QueryClient .Do (req )
402
+ require .NoErrorf (t , err , "Request failed: %s" , err )
385
403
require .Equalf (t , 200 , response .StatusCode , "Server returned http code: %s and response: %s" , response .Status , readAsString (response .Body ))
386
404
387
405
}
388
406
}
389
407
390
408
func TestSmokeGetAlert (t * testing.T ) {
391
409
if NewGlob .IngestorUrl .String () == "" {
392
- req , _ := NewGlob .QueryClient .NewRequest ("GET" , "/alerts " , nil )
410
+ req , _ := NewGlob .QueryClient .NewRequest ("GET" , "/targets " , nil )
393
411
response , err := NewGlob .QueryClient .Do (req )
394
412
require .NoErrorf (t , err , "Request failed: %s" , err )
413
+ bodyTargets , _ := io .ReadAll (response .Body )
414
+ reader1 := bytes .NewReader (bodyTargets )
415
+ targetId := getIdFromTargetResponse (reader1 )
416
+ req , _ = NewGlob .QueryClient .NewRequest ("GET" , "/alerts" , nil )
417
+ response , err = NewGlob .QueryClient .Do (req )
418
+ require .NoErrorf (t , err , "Request failed: %s" , err )
395
419
body , _ := io .ReadAll (response .Body )
396
- reader1 : = bytes .NewReader (body )
420
+ reader1 = bytes .NewReader (body )
397
421
reader2 := bytes .NewReader (body )
398
422
expected := readAsString (reader1 )
399
423
id , state := getIdStateFromAlertResponse (reader2 )
400
424
require .Equalf (t , 200 , response .StatusCode , "Server returned http code: %s and response: %s" , response .Status , body )
401
- res := createAlertResponse (id , state , NewGlob .Stream )
425
+ res := createAlertResponse (id , state , NewGlob .Stream , targetId )
402
426
require .JSONEq (t , expected , res , "Get alert response doesn't match with Alert config returned" )
403
427
DeleteAlert (t , NewGlob .QueryClient , id )
428
+ DeleteTarget (t , NewGlob .QueryClient , targetId )
404
429
}
405
430
DeleteStream (t , NewGlob .QueryClient , NewGlob .Stream )
406
431
}
0 commit comments