Skip to content

Commit 915b721

Browse files
swapneilsSwapneil Singh
andauthored
Update EKS and Kinesis load tests to match new validation format. (#886)
* Update validate.go API for EKS tests * Use S3 validation for kinesis tests --------- Co-authored-by: Swapneil Singh <[email protected]>
1 parent bb70814 commit 915b721

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

load_tests/load_test.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -517,9 +517,16 @@ def run_eks_tests():
517517
expect_time = log_stream['lastEventTimestamp']
518518
actual_time = log_stream['lastIngestionTime']
519519
log_delay = get_log_delay(actual_time/1000-expect_time/1000)
520-
os.environ['LOG_PREFIX'] = log_stream['logStreamName']
521-
os.environ['DESTINATION'] = 'cloudwatch'
522-
processes.add(subprocess.Popen(['go', 'run', './load_tests/validation/validate.go', input_record, log_delay]))
520+
log_prefix = resource_resolver.get_destination_cloudwatch_prefix(test_configuration["input_configuration"])
521+
exec_args = ['go', 'run', './load_tests/validation/validate.go',
522+
'-input-record', input_record,
523+
'-log-delay', log_delay,
524+
'-region', AWS_REGION,
525+
'-bucket', S3_BUCKET_NAME,
526+
'-log-group', LOG_GROUP_NAME,
527+
'-prefix', log_prefix,
528+
'-destination', OUTPUT_PLUGIN]
529+
processes.add(subprocess.Popen(exec_args))
523530

524531
# Wait until all subprocesses for validation completed
525532
for p in processes:

load_tests/validation/validate.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ func main() {
6868
inputMap = make(map[uint32]struct{}, *inputRecord)
6969

7070
totalRecordFound := 0
71-
if *destination == "s3" {
71+
if *destination == "s3" || *destination == "kinesis" || *destination == "firehose" {
7272
s3Client, err := getS3Client(*region)
7373
if err != nil {
7474
exitErrorf("[TEST FAILURE] Unable to create new S3 client: %v", err)

0 commit comments

Comments
 (0)