@@ -67,8 +67,11 @@ def test_enrichment_create_text_format
6767 create_ws_cmd = "bundle exec exe/exa-ai webset-create " \
6868 "--search '{\" query\" :\" Tech companies\" ,\" count\" :1}' " \
6969 "--output-format json"
70- ws_stdout , _stderr , _status = run_command ( create_ws_cmd )
70+ ws_stdout , ws_stderr , ws_status = run_command ( create_ws_cmd )
71+ assert ws_status . success? , "webset-create failed: #{ ws_stderr } "
72+
7173 webset = parse_json_output ( ws_stdout )
74+ refute_nil webset [ "id" ] , "webset-create returned no ID"
7275 webset_id = track_webset ( webset [ "id" ] )
7376
7477 # Create enrichment
@@ -97,8 +100,11 @@ def test_enrichment_create_options_format
97100 create_ws_cmd = "bundle exec exe/exa-ai webset-create " \
98101 "--search '{\" query\" :\" Startups\" ,\" count\" :1}' " \
99102 "--output-format json"
100- ws_stdout , _stderr , _status = run_command ( create_ws_cmd )
103+ ws_stdout , ws_stderr , ws_status = run_command ( create_ws_cmd )
104+ assert ws_status . success? , "webset-create failed: #{ ws_stderr } "
105+
101106 webset = parse_json_output ( ws_stdout )
107+ refute_nil webset [ "id" ] , "webset-create returned no ID"
102108 webset_id = track_webset ( webset [ "id" ] )
103109
104110 # Create enrichment with options format
@@ -133,8 +139,11 @@ def test_enrichment_create_options_from_file
133139 create_ws_cmd = "bundle exec exe/exa-ai webset-create " \
134140 "--search '{\" query\" :\" E-commerce companies\" ,\" count\" :1}' " \
135141 "--output-format json"
136- ws_stdout , _stderr , _status = run_command ( create_ws_cmd )
142+ ws_stdout , ws_stderr , ws_status = run_command ( create_ws_cmd )
143+ assert ws_status . success? , "webset-create failed: #{ ws_stderr } "
144+
137145 webset = parse_json_output ( ws_stdout )
146+ refute_nil webset [ "id" ] , "webset-create returned no ID"
138147 webset_id = track_webset ( webset [ "id" ] )
139148
140149 # Create temporary options file
@@ -175,16 +184,22 @@ def test_enrichment_get
175184 create_ws_cmd = "bundle exec exe/exa-ai webset-create " \
176185 "--search '{\" query\" :\" Healthcare companies\" ,\" count\" :1}' " \
177186 "--output-format json"
178- ws_stdout , _stderr , _status = run_command ( create_ws_cmd )
187+ ws_stdout , ws_stderr , ws_status = run_command ( create_ws_cmd )
188+ assert ws_status . success? , "webset-create failed: #{ ws_stderr } "
189+
179190 webset = parse_json_output ( ws_stdout )
191+ refute_nil webset [ "id" ] , "webset-create returned no ID"
180192 webset_id = track_webset ( webset [ "id" ] )
181193
182194 create_enrich_cmd = "bundle exec exe/exa-ai enrichment-create #{ webset_id } " \
183195 "--description 'Find phone number' " \
184196 "--format text " \
185197 "--output-format json"
186- enrich_stdout , _stderr , _status = run_command ( create_enrich_cmd )
198+ enrich_stdout , enrich_stderr , enrich_status = run_command ( create_enrich_cmd )
199+ assert enrich_status . success? , "enrichment-create failed: #{ enrich_stderr } "
200+
187201 enrichment = parse_json_output ( enrich_stdout )
202+ refute_nil enrichment [ "id" ] , "enrichment-create returned no ID"
188203 enrichment_id = track_enrichment ( webset_id , enrichment [ "id" ] )
189204
190205 # Get the enrichment
@@ -208,17 +223,23 @@ def test_enrichment_list
208223 create_ws_cmd = "bundle exec exe/exa-ai webset-create " \
209224 "--search '{\" query\" :\" Fintech companies\" ,\" count\" :1}' " \
210225 "--output-format json"
211- ws_stdout , _stderr , _status = run_command ( create_ws_cmd )
226+ ws_stdout , ws_stderr , ws_status = run_command ( create_ws_cmd )
227+ assert ws_status . success? , "webset-create failed: #{ ws_stderr } "
228+
212229 webset = parse_json_output ( ws_stdout )
230+ refute_nil webset [ "id" ] , "webset-create returned no ID"
213231 webset_id = track_webset ( webset [ "id" ] )
214232
215233 # Create an enrichment
216234 create_enrich_cmd = "bundle exec exe/exa-ai enrichment-create #{ webset_id } " \
217235 "--description 'Find headquarters location' " \
218236 "--format text " \
219237 "--output-format json"
220- enrich_stdout , _stderr , _status = run_command ( create_enrich_cmd )
238+ enrich_stdout , enrich_stderr , enrich_status = run_command ( create_enrich_cmd )
239+ assert enrich_status . success? , "enrichment-create failed: #{ enrich_stderr } "
240+
221241 enrichment = parse_json_output ( enrich_stdout )
242+ refute_nil enrichment [ "id" ] , "enrichment-create returned no ID"
222243 track_enrichment ( webset_id , enrichment [ "id" ] )
223244
224245 # List enrichments
@@ -243,16 +264,22 @@ def test_enrichment_update
243264 create_ws_cmd = "bundle exec exe/exa-ai webset-create " \
244265 "--search '{\" query\" :\" Retail companies\" ,\" count\" :1}' " \
245266 "--output-format json"
246- ws_stdout , _stderr , _status = run_command ( create_ws_cmd )
267+ ws_stdout , ws_stderr , ws_status = run_command ( create_ws_cmd )
268+ assert ws_status . success? , "webset-create failed: #{ ws_stderr } "
269+
247270 webset = parse_json_output ( ws_stdout )
271+ refute_nil webset [ "id" ] , "webset-create returned no ID"
248272 webset_id = track_webset ( webset [ "id" ] )
249273
250274 create_enrich_cmd = "bundle exec exe/exa-ai enrichment-create #{ webset_id } " \
251275 "--description 'Initial description' " \
252276 "--format text " \
253277 "--output-format json"
254- enrich_stdout , _stderr , _status = run_command ( create_enrich_cmd )
278+ enrich_stdout , enrich_stderr , enrich_status = run_command ( create_enrich_cmd )
279+ assert enrich_status . success? , "enrichment-create failed: #{ enrich_stderr } "
280+
255281 enrichment = parse_json_output ( enrich_stdout )
282+ refute_nil enrichment [ "id" ] , "enrichment-create returned no ID"
256283 enrichment_id = track_enrichment ( webset_id , enrichment [ "id" ] )
257284
258285 # Update the enrichment
@@ -278,16 +305,22 @@ def test_enrichment_delete
278305 create_ws_cmd = "bundle exec exe/exa-ai webset-create " \
279306 "--search '{\" query\" :\" Manufacturing companies\" ,\" count\" :1}' " \
280307 "--output-format json"
281- ws_stdout , _stderr , _status = run_command ( create_ws_cmd )
308+ ws_stdout , ws_stderr , ws_status = run_command ( create_ws_cmd )
309+ assert ws_status . success? , "webset-create failed: #{ ws_stderr } "
310+
282311 webset = parse_json_output ( ws_stdout )
312+ refute_nil webset [ "id" ] , "webset-create returned no ID"
283313 webset_id = track_webset ( webset [ "id" ] )
284314
285315 create_enrich_cmd = "bundle exec exe/exa-ai enrichment-create #{ webset_id } " \
286316 "--description 'To be deleted' " \
287317 "--format text " \
288318 "--output-format json"
289- enrich_stdout , _stderr , _status = run_command ( create_enrich_cmd )
319+ enrich_stdout , enrich_stderr , enrich_status = run_command ( create_enrich_cmd )
320+ assert enrich_status . success? , "enrichment-create failed: #{ enrich_stderr } "
321+
290322 enrichment = parse_json_output ( enrich_stdout )
323+ refute_nil enrichment [ "id" ] , "enrichment-create returned no ID"
291324 enrichment_id = track_enrichment ( webset_id , enrichment [ "id" ] )
292325
293326 # Delete the enrichment (with --force to skip confirmation)
@@ -313,16 +346,22 @@ def test_enrichment_cancel
313346 create_ws_cmd = "bundle exec exe/exa-ai webset-create " \
314347 "--search '{\" query\" :\" Education companies\" ,\" count\" :1}' " \
315348 "--output-format json"
316- ws_stdout , _stderr , _status = run_command ( create_ws_cmd )
349+ ws_stdout , ws_stderr , ws_status = run_command ( create_ws_cmd )
350+ assert ws_status . success? , "webset-create failed: #{ ws_stderr } "
351+
317352 webset = parse_json_output ( ws_stdout )
353+ refute_nil webset [ "id" ] , "webset-create returned no ID"
318354 webset_id = track_webset ( webset [ "id" ] )
319355
320356 create_enrich_cmd = "bundle exec exe/exa-ai enrichment-create #{ webset_id } " \
321357 "--description 'To be cancelled' " \
322358 "--format text " \
323359 "--output-format json"
324- enrich_stdout , _stderr , _status = run_command ( create_enrich_cmd )
360+ enrich_stdout , enrich_stderr , enrich_status = run_command ( create_enrich_cmd )
361+ assert enrich_status . success? , "enrichment-create failed: #{ enrich_stderr } "
362+
325363 enrichment = parse_json_output ( enrich_stdout )
364+ refute_nil enrichment [ "id" ] , "enrichment-create returned no ID"
326365 enrichment_id = track_enrichment ( webset_id , enrichment [ "id" ] )
327366
328367 # Cancel the enrichment
@@ -346,8 +385,11 @@ def test_enrichment_create_pretty_format
346385 create_ws_cmd = "bundle exec exe/exa-ai webset-create " \
347386 "--search '{\" query\" :\" AI companies\" ,\" count\" :1}' " \
348387 "--output-format json"
349- ws_stdout , _stderr , _status = run_command ( create_ws_cmd )
388+ ws_stdout , ws_stderr , ws_status = run_command ( create_ws_cmd )
389+ assert ws_status . success? , "webset-create failed: #{ ws_stderr } "
390+
350391 webset = parse_json_output ( ws_stdout )
392+ refute_nil webset [ "id" ] , "webset-create returned no ID"
351393 webset_id = track_webset ( webset [ "id" ] )
352394
353395 # Create enrichment with pretty format
0 commit comments