Mercurial > hg > orthanc-tests
annotate NewTests/MaxStorageReject/test_max_storage_reject.py @ 536:6546ca088276
new tests for private creator in worklists
author | Alain Mazy <am@osimis.io> |
---|---|
date | Mon, 22 May 2023 15:06:21 +0200 |
parents | d9d059c6082d |
children | 47c43fdaf91d |
rev | line source |
---|---|
490 | 1 import unittest |
2 import time | |
3 import subprocess | |
4 import pprint | |
534
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
5 import os |
490 | 6 from helpers import OrthancTestCase, Helpers |
7 | |
8 from orthanc_api_client import OrthancApiClient, generate_test_dicom_file | |
9 from orthanc_api_client import exceptions as orthanc_exceptions | |
10 | |
11 import pathlib | |
12 here = pathlib.Path(__file__).parent.resolve() | |
13 | |
14 | |
534
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
15 def count_files_in_storage(path): |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
16 all_files = [] |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
17 for root, dirs, files in os.walk(path): |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
18 for file in files: |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
19 if len(file) == 36: |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
20 all_files.append(file) |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
21 |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
22 return len(all_files) |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
23 |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
24 |
490 | 25 class TestMaxStorageReject(OrthancTestCase): |
26 | |
27 @classmethod | |
28 def prepare(cls): | |
29 test_name = "MaxStorageReject" | |
30 storage_name = "max_storage_reject" | |
31 | |
32 cls.clear_storage(storage_name=storage_name) | |
33 | |
34 config_path = cls.generate_configuration( | |
35 config_name=f"{test_name}_under_test", | |
36 storage_name=storage_name, | |
37 config={ | |
38 "MaximumPatientCount": 2, | |
39 "MaximumStorageMode": "Reject" | |
40 }, | |
41 plugins=Helpers.plugins | |
42 ) | |
43 | |
44 print(f'-------------- prepared {test_name} tests') | |
45 if Helpers.break_after_preparation: | |
46 print(f"++++ It is now time to start your Orthanc under tests with configuration file '{config_path}' +++++") | |
47 input("Press Enter to continue") | |
48 else: | |
49 print(f'-------------- launching {test_name} tests') | |
50 cls.launch_orthanc_under_tests( | |
51 config_path=config_path, | |
52 config_name=f"{test_name}_under_test", | |
53 storage_name=storage_name, | |
54 plugins=Helpers.plugins | |
55 ) | |
56 | |
57 print('-------------- waiting for orthanc-under-tests to be available') | |
58 cls.o.wait_started() | |
59 | |
60 | |
61 def test_upload_3_patients_rest_api(self): | |
62 | |
63 self.o.delete_all_content() | |
64 | |
65 # make sure the 3rd patient does not make it into the storage (through the Rest API) | |
66 self.o.upload_file(here / "../../Database/Brainix/Flair/IM-0001-0001.dcm") | |
67 self.o.upload_file(here / "../../Database/Knix/Loc/IM-0001-0001.dcm") | |
68 with self.assertRaises(orthanc_exceptions.HttpError) as ctx: | |
69 self.o.upload_file(here / "../../Database/Phenix/IM-0001-0001.dcm") | |
70 self.assertEqual(507, ctx.exception.http_status_code) | |
71 self.assertEqual(2, len(self.o.studies.get_all_ids())) | |
534
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
72 self.assertEqual(2, count_files_in_storage(self.get_storage_path("max_storage_reject"))) |
490 | 73 |
74 def upload_with_store_scu(self, path): | |
75 subprocess.check_call([Helpers.find_executable('storescu'), | |
76 "-xs", | |
77 Helpers.get_orthanc_ip(), | |
78 str(Helpers.get_orthanc_dicom_port()), | |
79 path]) | |
80 | |
81 def test_upload_3_patients_c_store(self): | |
82 | |
83 self.o.delete_all_content() | |
84 | |
85 # make sure the 3rd patient does not make it into the storage (through StoreSCU) | |
86 self.upload_with_store_scu(here / "../../Database/Brainix/Flair/IM-0001-0001.dcm") | |
87 self.upload_with_store_scu(here / "../../Database/Knix/Loc/IM-0001-0001.dcm") | |
88 with self.assertRaises(subprocess.CalledProcessError) as ctx: | |
89 self.upload_with_store_scu(here / "../../Database/Phenix/IM-0001-0001.dcm") | |
90 self.assertEqual(2, len(self.o.studies.get_all_ids())) | |
534
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
91 self.assertEqual(2, count_files_in_storage(self.get_storage_path("max_storage_reject"))) |
490 | 92 |
93 def test_upload_3_patients_dicomweb(self): | |
94 | |
95 self.o.delete_all_content() | |
96 | |
97 # make sure the 3rd patient does not make it into the storage (through DicomWeb) | |
98 self.o.upload_files_dicom_web([here / "../../Database/Brainix/Flair/IM-0001-0001.dcm"]) | |
99 self.o.upload_files_dicom_web([here / "../../Database/Knix/Loc/IM-0001-0001.dcm"]) | |
100 | |
101 with self.assertRaises(orthanc_exceptions.HttpError) as ctx: | |
102 self.o.upload_files_dicom_web([here / "../../Database/Phenix/IM-0001-0001.dcm"]) | |
103 self.assertEqual(400, ctx.exception.http_status_code) | |
104 | |
105 self.assertEqual(2, len(self.o.studies.get_all_ids())) | |
534
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
106 self.assertEqual(2, count_files_in_storage(self.get_storage_path("max_storage_reject"))) |
490 | 107 |
108 def test_upload_3_patients_dicomweb_in_one_query(self): | |
109 | |
110 self.o.delete_all_content() | |
111 | |
112 # make sure the 3rd patient does not make it into the storage (through DicomWeb) | |
113 r = self.o.upload_files_dicom_web([ | |
114 here / "../../Database/Brainix/Flair/IM-0001-0001.dcm", | |
115 here / "../../Database/Knix/Loc/IM-0001-0001.dcm", | |
116 here / "../../Database/Phenix/IM-0001-0001.dcm" | |
117 ]) | |
118 | |
119 # pprint.pprint(r) | |
120 self.assertEqual(2, len(self.o.studies.get_all_ids())) | |
121 self.assertIn('00081198', r) | |
534
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
122 self.assertEqual(0xA700, r['00081198']['Value'][0]['00081197']['Value'][0]) # one failed instance with out-of-resource status |
d9d059c6082d
Fix orphan files remaining in storage when working with MaximumStorageSize
Alain Mazy <am@osimis.io>
parents:
490
diff
changeset
|
123 self.assertEqual(2, count_files_in_storage(self.get_storage_path("max_storage_reject"))) |