2023-01-06 16:01:09 +00:00
|
|
|
# A simple script that creates a file of inputted size saved at the inputted destination.
|
2023-01-06 18:48:40 +00:00
|
|
|
# aimed at testing storage is functioning on Longhorn running in a kubernetes cluster.
|
|
|
|
|
|
|
|
|
|
import os
|
|
|
|
|
import time
|
|
|
|
|
|
|
|
|
|
# Get env vars:
|
|
|
|
|
print('Fetching configuration from environment')
|
|
|
|
|
size = os.getenv('STOR_SIZE') # In bytes (x1000)
|
|
|
|
|
path = os.getenv('STOR_PATH')
|
|
|
|
|
|
|
|
|
|
print('Path: '+ path + ', Size: ' + size + 'B')
|
|
|
|
|
|
|
|
|
|
def generate(number, path, size):
|
|
|
|
|
filepath = path + '/' + str(number) + 'stor.bin'
|
|
|
|
|
with open(filepath, mode='wb') as out:
|
|
|
|
|
out.write(os.urandom(int(size)))
|
|
|
|
|
|
|
|
|
|
print('File of ' + size + 'B created at ' + filepath)
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
passes = 0
|
|
|
|
|
while passes < 1000:
|
|
|
|
|
generate(passes,path,size)
|
|
|
|
|
passes += 1
|
|
|
|
|
|
|
|
|
|
while True:
|
|
|
|
|
print("Program running...")
|
|
|
|
|
time.sleep(10) # Is this s or ms?
|
