@@ -346,131 +346,6 @@ async def _launch_scan(
346
346
},
347
347
},
348
348
}
349
- # TODO
350
-
351
- assert 0
352
-
353
-
354
- def foo ():
355
- # check args (avoid whitespace headaches...)
356
- assert resp ["scanner_server_args" ].split () == scanner_server_args .split ()
357
- for got_args , exp_args in zip (resp ["ewms_task" ]["tms_args" ], tms_args ):
358
- print (got_args , exp_args )
359
- for got , exp in zip (got_args .split (), exp_args .split ()):
360
- print (got , exp )
361
- if exp == CLUSTER_ID_PLACEHOLDER :
362
- assert RE_UUID4HEX .fullmatch (got )
363
- else :
364
- assert got == exp
365
- assert len (got_args .split ()) == len (exp_args .split ())
366
- assert len (resp ["ewms_task" ]["tms_args" ]) == len (tms_args )
367
-
368
- # check env vars
369
- print (resp ["ewms_task" ]["env_vars" ])
370
- assert set (resp ["ewms_task" ]["env_vars" ].keys ()) == {
371
- "scanner_server" ,
372
- "tms_starters" ,
373
- }
374
-
375
- # check env vars, more closely
376
- # "scanner_server"
377
- assert set ( # these have `value`s
378
- e ["name" ]
379
- for e in resp ["ewms_task" ]["env_vars" ]["scanner_server" ]
380
- if e ["value" ] is not None and e ["value_from" ] is None
381
- ) == {
382
- "SKYSCAN_BROKER_ADDRESS" ,
383
- "SKYSCAN_BROKER_AUTH" ,
384
- "SKYSCAN_SKYDRIVER_ADDRESS" ,
385
- "SKYSCAN_SKYDRIVER_AUTH" ,
386
- "SKYSCAN_SKYDRIVER_SCAN_ID" ,
387
- "SKYSCAN_EWMS_PILOT_LOG" ,
388
- "SKYSCAN_MQ_CLIENT_LOG" ,
389
- * post_scan_body ["scanner_server_env" ].keys (), # type: ignore[attr-defined]
390
- }
391
- assert (
392
- set ( # these have `value_from`s
393
- e
394
- for e in resp ["ewms_task" ]["env_vars" ]["scanner_server" ]
395
- if e ["value_from" ] is not None and e ["value" ] is None
396
- )
397
- == set ()
398
- )
399
- # "tms_starters"
400
- for env_dicts in resp ["ewms_task" ]["env_vars" ]["tms_starters" ]:
401
- assert set ( # these have `value`s
402
- e ["name" ]
403
- for e in env_dicts
404
- if e ["value" ] is not None and e ["value_from" ] is None
405
- ) == {
406
- "EWMS_PILOT_TASK_TIMEOUT" , # set by CI runner
407
- "EWMS_TMS_S3_BUCKET" ,
408
- "EWMS_TMS_S3_URL" ,
409
- "SKYSCAN_BROKER_ADDRESS" ,
410
- "SKYSCAN_BROKER_AUTH" ,
411
- "SKYSCAN_SKYDRIVER_ADDRESS" ,
412
- "SKYSCAN_SKYDRIVER_AUTH" ,
413
- "SKYSCAN_SKYDRIVER_SCAN_ID" ,
414
- "SKYSCAN_EWMS_PILOT_LOG" ,
415
- "SKYSCAN_MQ_CLIENT_LOG" ,
416
- "WORKER_K8S_LOCAL_APPLICATION_NAME" ,
417
- "EWMS_PILOT_DUMP_TASK_OUTPUT" ,
418
- }
419
- assert (
420
- next (
421
- x ["value" ]
422
- for x in env_dicts
423
- if x ["name" ] == "EWMS_PILOT_DUMP_TASK_OUTPUT"
424
- )
425
- == "True"
426
- )
427
- assert set ( # these have `value_from`s
428
- e ["name" ]
429
- for e in env_dicts
430
- if e ["value_from" ] is not None and e ["value" ] is None
431
- ) == {
432
- "CONDOR_TOKEN" ,
433
- "EWMS_TMS_S3_ACCESS_KEY_ID" ,
434
- "EWMS_TMS_S3_SECRET_KEY" ,
435
- } or set ( # these have `value_from`s
436
- e ["name" ]
437
- for e in env_dicts
438
- if e ["value_from" ] is not None and e ["value" ] is None
439
- ) == {
440
- "WORKER_K8S_CONFIG_FILE_BASE64" ,
441
- "EWMS_TMS_S3_ACCESS_KEY_ID" ,
442
- "EWMS_TMS_S3_SECRET_KEY" ,
443
- }
444
-
445
- # check env vars, even MORE closely
446
- for env_dicts in [resp ["ewms_task" ]["env_vars" ]["scanner_server" ]] + resp [
447
- "ewms_task"
448
- ]["env_vars" ]["tms_starters" ]:
449
- assert (
450
- next (x ["value" ] for x in env_dicts if x ["name" ] == "SKYSCAN_BROKER_ADDRESS" )
451
- == "localhost"
452
- )
453
- assert re .match (
454
- r"http://localhost:[0-9]+" ,
455
- next (
456
- x ["value" ]
457
- for x in env_dicts
458
- if x ["name" ] == "SKYSCAN_SKYDRIVER_ADDRESS"
459
- ),
460
- )
461
- assert (
462
- len (
463
- next (
464
- x ["value" ]
465
- for x in env_dicts
466
- if x ["name" ] == "SKYSCAN_SKYDRIVER_SCAN_ID"
467
- )
468
- )
469
- == 32
470
- )
471
-
472
- # get scan_id
473
- assert resp ["scan_id" ]
474
349
475
350
return resp # type: ignore[no-any-return]
476
351
0 commit comments