scenarios/preset/preset_grpc.py

@@ -5,6 +5,7 @@ import json
 import random
 import sys
 import tempfile
+import time
 
 from argparse import Namespace
 from concurrent.futures import ProcessPoolExecutor
@@ -32,6 +33,8 @@ parser.add_argument('--endpoint', help='Node address')
 parser.add_argument('--update', help='Save existed containers')
 parser.add_argument('--ignore-errors', help='Ignore preset errors')
 parser.add_argument('--workers', help='Count of workers in preset. Max = 50, Default = 50', default=50)
+parser.add_argument('--sleep', help='Time to sleep between container creation and object PUT (in seconds), '
+                                    'Default = 8', default=8)
 
 args: Namespace = parser.parse_args()
 print(args)
@@ -74,6 +77,10 @@ def main():
         if not ignore_errors:
             sys.exit(ERROR_WRONG_CONTAINERS_COUNT)
 
+    if args.sleep != 0:
+        print(f"Sleep for {args.sleep} seconds")
+        time.sleep(args.sleep)
+
     print(f"Upload objects to each container: {args.preload_obj} ")
     payload_file = tempfile.NamedTemporaryFile()
     random_payload(payload_file, args.size)

scenarios/preset/preset_s3.py

@@ -4,6 +4,7 @@ import argparse
 import json
 import sys
 import tempfile
+import time
 from concurrent.futures import ProcessPoolExecutor
 
 from helpers.cmd import random_payload
@@ -22,6 +23,8 @@ parser.add_argument('--location', help='AWS location. Will be empty, if has not
 parser.add_argument('--versioning', help='True/False, False by default.')
 parser.add_argument('--ignore-errors', help='Ignore preset errors')
 parser.add_argument('--workers', help='Count of workers in preset. Max = 50, Default = 50', default=50)
+parser.add_argument('--sleep', help='Time to sleep between container creation and object PUT (in seconds), '
+                                    'Default = 8', default=8)
 
 args = parser.parse_args()
 print(args)
@@ -65,6 +68,10 @@ def main():
         if not ignore_errors:
             sys.exit(ERROR_WRONG_CONTAINERS_COUNT)
 
+    if args.sleep != 0:
+        print(f"Sleep for {args.sleep} seconds")
+        time.sleep(args.sleep)
+
     print(f"Upload objects to each bucket: {objects_per_bucket} ")
     payload_file = tempfile.NamedTemporaryFile()
     random_payload(payload_file, args.size)