Skip to content

Commit aa6b7e6

Browse files
committed
replace assert's with raise's
Signed-off-by: Bruno Vavala <bruno.vavala@intel.com>
1 parent 3999658 commit aa6b7e6

File tree

1 file changed

+31
-12
lines changed

1 file changed

+31
-12
lines changed

python/pdo/test/storage.py

+31-12
Original file line numberDiff line numberDiff line change
@@ -75,15 +75,23 @@ def verify_store_signature(store_response, duration, verifying_key) :
7575
try :
7676
block_data = os.urandom(1000)
7777
result = client.store_blocks([block_data], duration=default_duration)
78-
assert result
7978

80-
assert (1 == verify_store_signature(result, default_duration, client.verifying_key))
79+
r = verify_store_signature(result, default_duration, client.verifying_key)
80+
if 1 != r:
81+
raise ValueError("verify_store_signature error: {}".format(r))
82+
83+
if 'block_ids' not in result:
84+
raise ValueError("block_ids not in result")
8185

8286
block_ids = result['block_ids']
83-
assert block_ids and len(block_ids) == 1
87+
if len(block_ids) != 1:
88+
raise ValueError("len of block_ids is not 1: {}".format(len(block_ids)))
89+
if list != type(result['block_ids']):
90+
raise ValueError("block_ids not a list")
91+
if len(result['block_ids']) < 1:
92+
raise ValueError("no block in block_ids")
8493

8594
block_id = result['block_ids'][0]
86-
assert block_id
8795

8896
except Exception as e :
8997
logger.error('put test failed; %s', str(e))
@@ -94,7 +102,8 @@ def verify_store_signature(store_response, duration, verifying_key) :
94102
# -----------------------------------------------------------------
95103
try :
96104
verify_block_data = client.get_block(block_id)
97-
assert block_data == verify_block_data
105+
if block_data != verify_block_data:
106+
raise ValueError("retrieved block data different than expected")
98107
except Exception as e :
99108
logger.error('verify put test failed; %s', str(e))
100109
sys.exit(-1)
@@ -108,13 +117,19 @@ def verify_store_signature(store_response, duration, verifying_key) :
108117
block_data.append(os.urandom(10))
109118
block_data.append(os.urandom(10))
110119
result = client.store_blocks(block_data, duration=default_duration)
111-
assert result
112120

113-
assert (1 == verify_store_signature(result, default_duration, client.verifying_key))
121+
r = verify_store_signature(result, default_duration, client.verifying_key)
122+
if 1 != r:
123+
raise ValueError("verify_store_signature error: {}".format(r))
124+
125+
if 'block_ids' not in result:
126+
raise ValueError("block_ids not in result")
114127

115128
block_ids = result['block_ids']
116129
logger.info('RESULT: %s', result)
117-
assert block_ids and len(block_ids) == 3
130+
131+
if len(block_ids) != 3:
132+
raise ValueError("block_ids should contain 3 blocks: {}".format(len(block_ids)))
118133

119134
except Exception as e :
120135
logger.error('bulk upload test failed; %s', str(e))
@@ -126,7 +141,8 @@ def verify_store_signature(store_response, duration, verifying_key) :
126141
try :
127142
for i in range(len(block_ids)) :
128143
verify_block_data = client.get_block(block_ids[i])
129-
assert block_data[i] == verify_block_data
144+
if block_data[i] != verify_block_data:
145+
raise ValueError("retrieved block data different than expected: index {}".format(i))
130146
except Exception as e :
131147
logger.error('failed to verify bulk upload; %s', str(e))
132148
sys.exit(-1)
@@ -137,7 +153,8 @@ def verify_store_signature(store_response, duration, verifying_key) :
137153
try :
138154
verify_block_data_list = client.get_blocks(block_ids)
139155
for i in range(len(block_ids)) :
140-
assert block_data[i] == verify_block_data_list[i]
156+
if block_data[i] != verify_block_data_list[i]:
157+
raise ValueError("retrieved block data different than expected in list: index {}".format(i))
141158
except Exception as e :
142159
logger.error('failed to verify bulk upload; %s', str(e))
143160
sys.exit(-1)
@@ -149,8 +166,10 @@ def verify_store_signature(store_response, duration, verifying_key) :
149166
status = client.check_blocks(block_ids)
150167
assert status and len(status) == 3
151168
for s in status :
152-
assert s['size'] == 10
153-
assert 0 < s['duration'] and s['duration'] <= default_duration
169+
if s['size'] != 10:
170+
raise ValueError("status size not 10: {}".format(s['size']))
171+
if 0 >= s['duration'] and s['duration'] > default_duration:
172+
raise ValueError("block status duration not within range: {}".format(s['duration']))
154173

155174
except Exception as e :
156175
logger.exception('bulk status failed; %s', str(e))

0 commit comments

Comments
 (0)