@@ -392,6 +392,7 @@ headers:
392
392
key: test` ), 2 )
393
393
394
394
time .Sleep (1 * time .Second )
395
+
395
396
rawEvt := `{"test": "test"}`
396
397
397
398
errChan := make (chan error )
@@ -435,6 +436,7 @@ custom_headers:
435
436
436
437
rawEvt := `{"test": "test"}`
437
438
errChan := make (chan error )
439
+
438
440
go assertEvents (out , []string {rawEvt }, errChan )
439
441
440
442
req , err := http .NewRequestWithContext (ctx , http .MethodPost , fmt .Sprintf ("%s/test" , testHTTPServerAddr ), strings .NewReader (rawEvt ))
@@ -468,9 +470,11 @@ func (sr *slowReader) Read(p []byte) (int, error) {
468
470
if sr .index >= len (sr .body ) {
469
471
return 0 , io .EOF
470
472
}
473
+
471
474
time .Sleep (sr .delay ) // Simulate a delay in reading
472
475
n := copy (p , sr .body [sr .index :])
473
476
sr .index += n
477
+
474
478
return n , nil
475
479
}
476
480
@@ -497,10 +501,12 @@ func assertEvents(out chan types.Event, expected []string, errChan chan error) {
497
501
errChan <- fmt .Errorf (`expected %s, got '%+v'` , expected , evt .Line .Raw )
498
502
return
499
503
}
504
+
500
505
if evt .Line .Src != "127.0.0.1" {
501
506
errChan <- fmt .Errorf ("expected '127.0.0.1', got '%s'" , evt .Line .Src )
502
507
return
503
508
}
509
+
504
510
if evt .Line .Module != "http" {
505
511
errChan <- fmt .Errorf ("expected 'http', got '%s'" , evt .Line .Module )
506
512
return
606
612
607
613
rawEvt := `{"test": "test"}`
608
614
errChan := make (chan error )
615
+
609
616
go assertEvents (out , []string {rawEvt }, errChan )
610
617
611
618
req , err := http .NewRequestWithContext (ctx , http .MethodPost , fmt .Sprintf ("%s/test" , testHTTPServerAddrTLS ), strings .NewReader (rawEvt ))
666
673
667
674
rawEvt := `{"test": "test"}`
668
675
errChan := make (chan error )
676
+
669
677
go assertEvents (out , []string {rawEvt }, errChan )
670
678
671
679
req , err := http .NewRequestWithContext (ctx , http .MethodPost , fmt .Sprintf ("%s/test" , testHTTPServerAddrTLS ), strings .NewReader (rawEvt ))
@@ -702,6 +710,7 @@ headers:
702
710
703
711
rawEvt := `{"test": "test"}`
704
712
errChan := make (chan error )
713
+
705
714
go assertEvents (out , []string {rawEvt , rawEvt }, errChan )
706
715
707
716
var b strings.Builder
@@ -753,9 +762,10 @@ headers:
753
762
key: test` ), 2 )
754
763
755
764
time .Sleep (1 * time .Second )
756
- rawEvt := `{"test": "test"}`
757
765
766
+ rawEvt := `{"test": "test"}`
758
767
errChan := make (chan error )
768
+
759
769
go assertEvents (out , []string {rawEvt , rawEvt }, errChan )
760
770
761
771
client := & http.Client {}
@@ -786,10 +796,13 @@ func assertMetrics(t *testing.T, reg *prometheus.Registry, metrics []prometheus.
786
796
require .NoError (t , err )
787
797
788
798
isExist := false
799
+
789
800
for _ , metricFamily := range promMetrics {
790
801
if metricFamily .GetName () == "cs_httpsource_hits_total" {
791
802
isExist = true
803
+
792
804
assert .Len (t , metricFamily .GetMetric (), 1 )
805
+
793
806
for _ , metric := range metricFamily .GetMetric () {
794
807
assert .InDelta (t , float64 (expected ), metric .GetCounter ().GetValue (), 0.000001 )
795
808
labels := metric .GetLabel ()
@@ -801,6 +814,7 @@ func assertMetrics(t *testing.T, reg *prometheus.Registry, metrics []prometheus.
801
814
}
802
815
}
803
816
}
817
+
804
818
if ! isExist && expected > 0 {
805
819
t .Fatalf ("expected metric cs_httpsource_hits_total not found" )
806
820
}
0 commit comments