ale64bit/xk6-frostfs
1
0
Fork 0
forked from TrueCloudLab/xk6-frostfs
Code Issues Pull requests Projects Releases Packages Wiki Activity

[#30] preset: Create missing files

Browse source 
Also, stop if none of the container creation commands succeeded.

Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
Pavel Karpy 2022-10-25 18:01:08 +03:00 committed by fyrchik
parent 0dda9f17fa
commit af840ed376
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
scenarios/preset/preset_grpc.py
View file

@ -68,7 +68,7 @@ def main():
data = {'containers': container_list, 'objects': objects_struct, 'obj_size': args.size + " Kb"} data = {'containers': container_list, 'objects': objects_struct, 'obj_size': args.size + " Kb"}
with open(args.out, 'w') as f: with open(args.out, 'w+') as f:
json.dump(data, f, ensure_ascii=False) json.dump(data, f, ensure_ascii=False)
print(f"Result:") print(f"Result:")
@ -77,7 +77,7 @@ def main():
def random_payload(payload_filepath): def random_payload(payload_filepath):
with open('%s' % payload_filepath, 'wb') as fout: with open('%s' % payload_filepath, 'w+b') as fout:
fout.write(os.urandom(1024 * int(args.size))) fout.write(os.urandom(1024 * int(args.size)))
@ -100,7 +100,7 @@ def create_container():
output, success = execute_cmd(cmd_line) output, success = execute_cmd(cmd_line)
if not success: if not success:
print(f" > Container has not been created.") print(f" > Container has not been created:\n{output}")
else: else:
try: try:
fst_str = output.split('\n')[0] fst_str = output.split('\n')[0]
@ -119,7 +119,7 @@ def upload_object(container, payload_filepath):
out, success = execute_cmd(cmd_line) out, success = execute_cmd(cmd_line)
if not success: if not success:
print(f" > Object {object_name} has not been uploaded.") print(f" > Object {object_name} has not been uploaded:\n{out}")
else: else:
try: try:
# taking second string from command output # taking second string from command output

4
scenarios/preset/preset_s3.py
View file

@ -66,7 +66,7 @@ def main():
data = {'buckets': bucket_list, 'objects': objects_struct, 'obj_size': args.size + " Kb"} data = {'buckets': bucket_list, 'objects': objects_struct, 'obj_size': args.size + " Kb"}
with open(args.out, 'w') as f: with open(args.out, 'w+') as f:
json.dump(data, f, ensure_ascii=False) json.dump(data, f, ensure_ascii=False)
print(f"Result:") print(f"Result:")
@ -75,7 +75,7 @@ def main():
def random_payload(payload_filepath): def random_payload(payload_filepath):
with open('%s' % payload_filepath, 'wb') as fout: with open('%s' % payload_filepath, 'w+b') as fout:
fout.write(os.urandom(1024 * int(args.size))) fout.write(os.urandom(1024 * int(args.size)))