Mercurial > hg > orthanc-tests
annotate NewTests/MaxStorage/test_max_storage_reject.py @ 624:543e372d2265
added a PG max storage test
| author | Alain Mazy <am@osimis.io> |
|---|---|
| date | Wed, 14 Feb 2024 11:25:40 +0100 |
| parents | NewTests/MaxStorageReject/test_max_storage_reject.py@47c43fdaf91d |
| children |
| 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 | |
| 544 | 63 if self.o.get_system()["ApiVersion"] > 20: # from Orthanc 1.12.1 |
| 64 self.o.delete_all_content() | |
| 490 | 65 |
| 544 | 66 # make sure the 3rd patient does not make it into the storage (through the Rest API) |
| 67 self.o.upload_file(here / "../../Database/Brainix/Flair/IM-0001-0001.dcm") | |
| 68 self.o.upload_file(here / "../../Database/Knix/Loc/IM-0001-0001.dcm") | |
| 69 with self.assertRaises(orthanc_exceptions.HttpError) as ctx: | |
| 70 self.o.upload_file(here / "../../Database/Phenix/IM-0001-0001.dcm") | |
| 71 self.assertEqual(507, ctx.exception.http_status_code) | |
| 72 self.assertEqual(2, len(self.o.studies.get_all_ids())) | |
| 73 self.assertEqual(2, count_files_in_storage(self.get_storage_path("max_storage_reject"))) | |
| 490 | 74 |
| 75 def upload_with_store_scu(self, path): | |
| 76 subprocess.check_call([Helpers.find_executable('storescu'), | |
| 77 "-xs", | |
| 78 Helpers.get_orthanc_ip(), | |
| 79 str(Helpers.get_orthanc_dicom_port()), | |
| 80 path]) | |
| 81 | |
| 82 def test_upload_3_patients_c_store(self): | |
| 83 | |
| 544 | 84 if self.o.get_system()["ApiVersion"] > 20: # from Orthanc 1.12.1 |
| 85 self.o.delete_all_content() | |
| 86 | |
| 87 # make sure the 3rd patient does not make it into the storage (through StoreSCU) | |
| 88 self.upload_with_store_scu(here / "../../Database/Brainix/Flair/IM-0001-0001.dcm") | |
| 89 self.upload_with_store_scu(here / "../../Database/Knix/Loc/IM-0001-0001.dcm") | |
| 90 with self.assertRaises(subprocess.CalledProcessError) as ctx: | |
| 91 self.upload_with_store_scu(here / "../../Database/Phenix/IM-0001-0001.dcm") | |
| 92 self.assertEqual(2, len(self.o.studies.get_all_ids())) | |
| 93 self.assertEqual(2, count_files_in_storage(self.get_storage_path("max_storage_reject"))) | |
| 490 | 94 |
| 95 def test_upload_3_patients_dicomweb(self): | |
| 96 | |
| 544 | 97 if self.o.get_system()["ApiVersion"] > 20: # from Orthanc 1.12.1 |
| 98 self.o.delete_all_content() | |
| 99 | |
| 100 # make sure the 3rd patient does not make it into the storage (through DicomWeb) | |
| 101 self.o.upload_files_dicom_web([here / "../../Database/Brainix/Flair/IM-0001-0001.dcm"]) | |
| 102 self.o.upload_files_dicom_web([here / "../../Database/Knix/Loc/IM-0001-0001.dcm"]) | |
| 490 | 103 |
| 544 | 104 with self.assertRaises(orthanc_exceptions.HttpError) as ctx: |
| 105 self.o.upload_files_dicom_web([here / "../../Database/Phenix/IM-0001-0001.dcm"]) | |
| 106 self.assertEqual(400, ctx.exception.http_status_code) | |
| 490 | 107 |
| 544 | 108 self.assertEqual(2, len(self.o.studies.get_all_ids())) |
| 109 self.assertEqual(2, count_files_in_storage(self.get_storage_path("max_storage_reject"))) | |
| 490 | 110 |
| 111 def test_upload_3_patients_dicomweb_in_one_query(self): | |
| 112 | |
| 544 | 113 if self.o.get_system()["ApiVersion"] > 20: # from Orthanc 1.12.1 |
| 114 self.o.delete_all_content() | |
| 115 | |
| 116 # make sure the 3rd patient does not make it into the storage (through DicomWeb) | |
| 117 r = self.o.upload_files_dicom_web([ | |
| 118 here / "../../Database/Brainix/Flair/IM-0001-0001.dcm", | |
| 119 here / "../../Database/Knix/Loc/IM-0001-0001.dcm", | |
| 120 here / "../../Database/Phenix/IM-0001-0001.dcm" | |
| 121 ]) | |
| 490 | 122 |
| 544 | 123 # pprint.pprint(r) |
| 124 self.assertEqual(2, len(self.o.studies.get_all_ids())) | |
| 125 self.assertIn('00081198', r) | |
| 126 self.assertEqual(0xA700, r['00081198']['Value'][0]['00081197']['Value'][0]) # one failed instance with out-of-resource status | |
| 127 self.assertEqual(2, count_files_in_storage(self.get_storage_path("max_storage_reject"))) |