88
99from mesh_client import INT_ENDPOINT , MeshClient
1010
11- CONTAINER_NAME = "notifications-reports"
12- REPORT_FILENAME = "SM0K3-reconciliation-report.csv"
1311WORK_DIR = os .path .dirname (os .path .realpath (__file__ ))
1412STARTUP_SCRIPT_PATH = f"{ WORK_DIR } /../../bash/run_container_app_job.sh"
1513
1614
1715def test_notifications ():
18- try :
19- logging .info ("Running notifications smoke test" )
16+ logging .info ("Running notifications smoke test" )
2017
21- environment , storage_account , resource_group_name = configure ()
18+ environment , storage_account , resource_group_name = configure ()
2219
23- if environment == "prod" :
24- return
20+ if environment == "prod" :
21+ return
2522
26- setup_mesh_inbox_test_data (environment , resource_group_name )
23+ setup_mesh_inbox_test_data (environment , resource_group_name )
2724
28- for job in ["smm" , "cap" , "smb" , "smk" ]:
29- job_result = run_subprocess (
30- f"Starting notifications container app job manbrs-{ job } -{ environment } " ,
31- [STARTUP_SCRIPT_PATH , environment , job ],
32- )
33- assert job_result .returncode == 0
34-
35- download_result = run_subprocess (
36- "Downloading generated smoke test report from blob storage" ,
37- azure_storage_blob_download_reports_command (storage_account ),
25+ for job in ["smm" , "cap" , "smb" , "smk" ]:
26+ logging .info (
27+ "Starting notifications container app job manbrs-%s-%s" , job , environment
3828 )
39- assert download_result .returncode == 0
40- assert REPORT_FILENAME in download_result .stdout
41-
42- report_contents = open (f"{ WORK_DIR } /{ REPORT_FILENAME } " ).read ()
43- assert "SM0K3" in report_contents
4429
45- logging . info ( "Finished notifications smoke test" )
46- finally :
47- run_subprocess (
48- "Deleting generated smoke test report from blob storage" ,
49- azure_storage_blob_delete_reports_command ( storage_account ) ,
30+ job_result = subprocess . run (
31+ [ STARTUP_SCRIPT_PATH , environment , job ],
32+ check = True ,
33+ capture_output = True ,
34+ text = True ,
5035 )
36+ assert job_result .returncode == 0
37+
38+ logging .info ("Finished notifications smoke test" )
5139
5240
5341def configure ():
@@ -64,47 +52,6 @@ def configure():
6452 return (environment , storage_account , resource_group_name )
6553
6654
67- def run_subprocess (description : str , command : list [str ]):
68- logging .info (description )
69- return subprocess .run (command , capture_output = True , text = True )
70-
71-
72- def azure_storage_blob_download_reports_command (storage_account : str ) -> list [str ]:
73- return [
74- "az" ,
75- "storage" ,
76- "blob" ,
77- "download-batch" ,
78- "--destination" ,
79- WORK_DIR ,
80- "--source" ,
81- CONTAINER_NAME ,
82- "--pattern" ,
83- REPORT_FILENAME ,
84- "--account-name" ,
85- storage_account ,
86- "--auth-mode" ,
87- "login" ,
88- ]
89-
90-
91- def azure_storage_blob_delete_reports_command (storage_account : str ) -> list [str ]:
92- return [
93- "az" ,
94- "storage" ,
95- "blob" ,
96- "delete-batch" ,
97- "--source" ,
98- CONTAINER_NAME ,
99- "--pattern" ,
100- REPORT_FILENAME ,
101- "--account-name" ,
102- storage_account ,
103- "--auth-mode" ,
104- "login" ,
105- ]
106-
107-
10855def setup_mesh_inbox_test_data (environment : str , resource_group_name : str ):
10956 populate_mesh_env_vars (environment , resource_group_name )
11057 mesh_client ().send_message (
0 commit comments