Mercurial > hg > orthanc-tests
annotate Tests/Toolbox.py @ 352:3d406c0261fc
test /tools/dicom-echo
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Thu, 03 Dec 2020 11:51:06 +0100 |
parents | bf8369ea3ff1 |
children | 79ce0f7a9714 |
rev | line source |
---|---|
0 | 1 #!/usr/bin/python |
2 | |
1 | 3 # Orthanc - A Lightweight, RESTful DICOM Store |
73 | 4 # Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics |
1 | 5 # Department, University Hospital of Liege, Belgium |
260
943166deebcb
upgrade to year 2020
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
249
diff
changeset
|
6 # Copyright (C) 2017-2020 Osimis S.A., Belgium |
1 | 7 # |
8 # This program is free software: you can redistribute it and/or | |
9 # modify it under the terms of the GNU General Public License as | |
10 # published by the Free Software Foundation, either version 3 of the | |
11 # License, or (at your option) any later version. | |
12 # | |
13 # This program is distributed in the hope that it will be useful, but | |
14 # WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
16 # General Public License for more details. | |
17 # | |
18 # You should have received a copy of the GNU General Public License | |
19 # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
0 | 20 |
21 | |
22 import hashlib | |
23 import httplib2 | |
24 import json | |
4 | 25 import os |
1 | 26 import re |
4 | 27 import signal |
1 | 28 import subprocess |
291
cfa785074c64
test_modify_transcode
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
29 import tempfile |
4 | 30 import threading |
83 | 31 import sys |
0 | 32 import time |
1 | 33 import zipfile |
0 | 34 |
337
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
35 from xml.dom import minidom |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
36 |
304 | 37 from PIL import Image, ImageChops |
38 import math | |
39 import operator | |
40 | |
83 | 41 |
42 if (sys.version_info >= (3, 0)): | |
43 from urllib.parse import urlencode | |
44 from io import StringIO | |
45 from io import BytesIO | |
341
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
46 from urllib.parse import unquote |
83 | 47 |
48 else: | |
49 from urllib import urlencode | |
341
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
50 from urlparse import unquote |
83 | 51 |
52 # http://stackoverflow.com/a/1313868/881731 | |
53 try: | |
54 from cStringIO import StringIO | |
55 except: | |
56 from StringIO import StringIO | |
3 | 57 |
341
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
58 |
83 | 59 def _DecodeJson(s): |
60 t = s | |
61 | |
62 if (sys.version_info >= (3, 0)): | |
63 try: | |
64 t = s.decode() | |
65 except: | |
66 pass | |
67 | |
68 try: | |
69 return json.loads(t) | |
70 except: | |
71 return t | |
0 | 72 |
73 | |
13 | 74 def DefineOrthanc(server = 'localhost', |
75 restPort = 8042, | |
1 | 76 username = None, |
77 password = None, | |
78 aet = 'ORTHANC', | |
79 dicomPort = 4242): | |
13 | 80 #m = re.match(r'(http|https)://([^:]+):([^@]+)@([^@]+)', url) |
81 #if m != None: | |
82 # url = m.groups()[0] + '://' + m.groups()[3] | |
83 # username = m.groups()[1] | |
84 # password = m.groups()[2] | |
0 | 85 |
13 | 86 #if not url.endswith('/'): |
87 # url += '/' | |
0 | 88 |
1 | 89 return { |
13 | 90 'Server' : server, |
91 'Url' : 'http://%s:%d/' % (server, restPort), | |
1 | 92 'Username' : username, |
93 'Password' : password, | |
94 'DicomAet' : aet, | |
95 'DicomPort' : dicomPort | |
96 } | |
0 | 97 |
98 | |
99 def _SetupCredentials(orthanc, http): | |
1 | 100 if (orthanc['Username'] != None and |
101 orthanc['Password'] != None): | |
102 http.add_credentials(orthanc['Username'], orthanc['Password']) | |
0 | 103 |
28 | 104 def DoGetRaw(orthanc, uri, data = {}, body = None, headers = {}): |
0 | 105 d = '' |
106 if len(data.keys()) > 0: | |
107 d = '?' + urlencode(data) | |
108 | |
109 http = httplib2.Http() | |
21
2a29bcff60a7
tests of image decoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
13
diff
changeset
|
110 http.follow_redirects = False |
0 | 111 _SetupCredentials(orthanc, http) |
112 | |
1 | 113 resp, content = http.request(orthanc['Url'] + uri + d, 'GET', body = body, |
0 | 114 headers = headers) |
28 | 115 return (resp, content) |
116 | |
117 | |
118 def DoGet(orthanc, uri, data = {}, body = None, headers = {}): | |
119 (resp, content) = DoGetRaw(orthanc, uri, data = data, body = body, headers = headers) | |
120 | |
0 | 121 if not (resp.status in [ 200 ]): |
239
8980bd19e31d
dicomweb: test_allowed_methods
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
122 raise Exception(resp.status, resp) |
0 | 123 else: |
83 | 124 return _DecodeJson(content) |
0 | 125 |
126 def _DoPutOrPost(orthanc, uri, method, data, contentType, headers): | |
127 http = httplib2.Http() | |
21
2a29bcff60a7
tests of image decoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
13
diff
changeset
|
128 http.follow_redirects = False |
0 | 129 _SetupCredentials(orthanc, http) |
130 | |
83 | 131 if isinstance(data, (str, bytearray, bytes)): |
0 | 132 body = data |
133 if len(contentType) != 0: | |
134 headers['content-type'] = contentType | |
135 else: | |
136 body = json.dumps(data) | |
137 headers['content-type'] = 'application/json' | |
138 | |
139 headers['expect'] = '' | |
140 | |
1 | 141 resp, content = http.request(orthanc['Url'] + uri, method, |
0 | 142 body = body, |
143 headers = headers) | |
342
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
144 if not (resp.status in [ 200, 201, 302 ]): |
239
8980bd19e31d
dicomweb: test_allowed_methods
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
145 raise Exception(resp.status, resp) |
0 | 146 else: |
83 | 147 return _DecodeJson(content) |
0 | 148 |
149 def DoDelete(orthanc, uri): | |
150 http = httplib2.Http() | |
21
2a29bcff60a7
tests of image decoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
13
diff
changeset
|
151 http.follow_redirects = False |
0 | 152 _SetupCredentials(orthanc, http) |
153 | |
1 | 154 resp, content = http.request(orthanc['Url'] + uri, 'DELETE') |
0 | 155 if not (resp.status in [ 200 ]): |
239
8980bd19e31d
dicomweb: test_allowed_methods
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
156 raise Exception(resp.status, resp) |
0 | 157 else: |
83 | 158 return _DecodeJson(content) |
0 | 159 |
160 def DoPut(orthanc, uri, data = {}, contentType = ''): | |
10 | 161 return _DoPutOrPost(orthanc, uri, 'PUT', data, contentType, {}) |
0 | 162 |
163 def DoPost(orthanc, uri, data = {}, contentType = '', headers = {}): | |
164 return _DoPutOrPost(orthanc, uri, 'POST', data, contentType, headers) | |
165 | |
13 | 166 def GetDatabasePath(filename): |
167 return os.path.join(os.path.dirname(__file__), '..', 'Database', filename) | |
168 | |
0 | 169 def UploadInstance(orthanc, filename): |
83 | 170 with open(GetDatabasePath(filename), 'rb') as f: |
171 d = f.read() | |
172 | |
0 | 173 return DoPost(orthanc, '/instances', d, 'application/dicom') |
174 | |
175 def UploadFolder(orthanc, path): | |
13 | 176 for i in os.listdir(GetDatabasePath(path)): |
1 | 177 try: |
178 UploadInstance(orthanc, os.path.join(path, i)) | |
179 except: | |
180 pass | |
0 | 181 |
182 def DropOrthanc(orthanc): | |
183 # Reset the Lua callbacks | |
184 DoPost(orthanc, '/tools/execute-script', 'function OnStoredInstance(instanceId, tags, metadata) end', 'application/lua') | |
185 | |
186 DoDelete(orthanc, '/exports') | |
187 | |
188 for s in DoGet(orthanc, '/patients'): | |
189 DoDelete(orthanc, '/patients/%s' % s) | |
190 | |
174 | 191 def InstallLuaScriptFromPath(orthanc, path): |
192 with open(GetDatabasePath(path), 'r') as f: | |
193 InstallLuaScript(orthanc, f.read()) | |
194 | |
195 def InstallLuaScript(orthanc, script): | |
196 DoPost(orthanc, '/tools/execute-script', script, 'application/lua') | |
197 | |
198 def UninstallLuaCallbacks(orthanc): | |
199 DoPost(orthanc, '/tools/execute-script', 'function OnStoredInstance() end', 'application/lua') | |
200 InstallLuaScriptFromPath(orthanc, 'Lua/TransferSyntaxEnable.lua') | |
201 | |
202 | |
0 | 203 def ComputeMD5(data): |
204 m = hashlib.md5() | |
205 m.update(data) | |
206 return m.hexdigest() | |
207 | |
249
24e5c8ca9440
DICOMweb: test_rendered
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
239
diff
changeset
|
208 def UncompressImage(data): |
83 | 209 if (sys.version_info >= (3, 0)): |
210 return Image.open(BytesIO(data)) | |
211 else: | |
212 return Image.open(StringIO(data)) | |
0 | 213 |
249
24e5c8ca9440
DICOMweb: test_rendered
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
239
diff
changeset
|
214 def GetImage(orthanc, uri, headers = {}): |
24e5c8ca9440
DICOMweb: test_rendered
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
239
diff
changeset
|
215 # http://www.pythonware.com/library/pil/handbook/introduction.htm |
24e5c8ca9440
DICOMweb: test_rendered
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
239
diff
changeset
|
216 return UncompressImage(DoGet(orthanc, uri, headers = headers)) |
24e5c8ca9440
DICOMweb: test_rendered
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
239
diff
changeset
|
217 |
0 | 218 def GetArchive(orthanc, uri): |
219 # http://stackoverflow.com/a/1313868/881731 | |
220 s = DoGet(orthanc, uri) | |
83 | 221 |
222 if (sys.version_info >= (3, 0)): | |
223 return zipfile.ZipFile(BytesIO(s), "r") | |
224 else: | |
225 return zipfile.ZipFile(StringIO(s), "r") | |
0 | 226 |
192 | 227 def PostArchive(orthanc, uri, body): |
228 # http://stackoverflow.com/a/1313868/881731 | |
229 s = DoPost(orthanc, uri, body) | |
230 | |
231 if (sys.version_info >= (3, 0)): | |
232 return zipfile.ZipFile(BytesIO(s), "r") | |
233 else: | |
234 return zipfile.ZipFile(StringIO(s), "r") | |
235 | |
1 | 236 def IsDefinedInLua(orthanc, name): |
0 | 237 s = DoPost(orthanc, '/tools/execute-script', 'print(type(%s))' % name, 'application/lua') |
238 return (s.strip() != 'nil') | |
239 | |
1 | 240 def WaitEmpty(orthanc): |
0 | 241 while True: |
1 | 242 if len(DoGet(orthanc, '/instances')) == 0: |
0 | 243 return |
342
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
244 time.sleep(0.01) |
0 | 245 |
137
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
246 def WaitJobDone(orthanc, job): |
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
247 while True: |
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
248 s = DoGet(orthanc, '/jobs/%s' % job) ['State'] |
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
249 |
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
250 if s == 'Success': |
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
251 return True |
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
252 elif s == 'Failure': |
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
253 return False |
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
254 |
342
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
255 time.sleep(0.01) |
137
412d5f70447e
testing asynchronous c-move
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
130
diff
changeset
|
256 |
138 | 257 def MonitorJob(orthanc, func): # "func" is a lambda |
258 a = set(DoGet(orthanc, '/jobs')) | |
259 func() | |
260 b = set(DoGet(orthanc, '/jobs')) | |
261 | |
262 diff = list(b - a) | |
263 if len(diff) != 1: | |
264 print('No job was created!') | |
265 return False | |
266 else: | |
267 return WaitJobDone(orthanc, diff[0]) | |
268 | |
179
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
269 def MonitorJob2(orthanc, func): # "func" is a lambda |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
270 a = set(DoGet(orthanc, '/jobs')) |
181 | 271 job = func() |
179
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
272 b = set(DoGet(orthanc, '/jobs')) |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
273 |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
274 diff = list(b - a) |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
275 if len(diff) != 1: |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
276 print('No job was created!') |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
277 return None |
181 | 278 elif (not 'ID' in job or |
279 diff[0] != job['ID']): | |
280 print('Mismatch in the job ID') | |
281 return None | |
179
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
282 elif WaitJobDone(orthanc, diff[0]): |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
283 return diff[0] |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
284 else: |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
285 print('Error while executing the job') |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
286 return None |
8a2dd77d4035
testing split/merge
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
287 |
220 | 288 def WaitAllNewJobsDone(orthanc, func): # "func" is a lambda |
289 a = set(DoGet(orthanc, '/jobs')) | |
290 func() | |
291 | |
292 first = True | |
293 | |
294 while True: | |
295 b = set(DoGet(orthanc, '/jobs')) | |
296 | |
297 diff = list(b - a) | |
298 if len(diff) == 0: | |
299 if first: | |
300 raise Exception('No job was created') | |
301 else: | |
302 return # We're done | |
303 else: | |
304 first = False | |
305 | |
306 if WaitJobDone(orthanc, diff[0]): | |
307 a.add(diff[0]) | |
308 else: | |
309 raise Exception('Error while executing the job') | |
310 | |
311 | |
1 | 312 def GetDockerHostAddress(): |
313 route = subprocess.check_output([ '/sbin/ip', 'route' ]) | |
314 m = re.search(r'default via ([0-9.]+)', route) | |
315 if m == None: | |
316 return 'localhost' | |
317 else: | |
318 return m.groups()[0] | |
4 | 319 |
44
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
320 def FindExecutable(name): |
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
321 p = os.path.join('/usr/local/bin', name) |
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
322 if os.path.isfile(p): |
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
323 return p |
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
324 |
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
325 p = os.path.join('/usr/local/sbin', name) |
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
326 if os.path.isfile(p): |
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
327 return p |
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
328 |
ffa542cce638
Toolbox.FindExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
28
diff
changeset
|
329 return name |
4 | 330 |
173 | 331 def IsOrthancVersionAbove(orthanc, major, minor, revision): |
332 v = DoGet(orthanc, '/system')['Version'] | |
333 | |
334 if v == 'mainline': | |
335 return True | |
336 else: | |
337 tmp = v.split('.') | |
338 a = int(tmp[0]) | |
339 b = int(tmp[1]) | |
340 c = int(tmp[2]) | |
341 return (a > major or | |
342 (a == major and b > minor) or | |
343 (a == major and b == minor and c >= revision)) | |
4 | 344 |
345 | |
346 class ExternalCommandThread: | |
347 @staticmethod | |
348 def ExternalCommandFunction(arg, stop_event, command, env): | |
318
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
349 with open(os.devnull, 'w') as devnull: |
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
350 external = subprocess.Popen(command, env = env, stderr = devnull) |
4 | 351 |
318
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
352 while (not stop_event.is_set()): |
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
353 error = external.poll() |
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
354 if error != None: |
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
355 # http://stackoverflow.com/a/1489838/881731 |
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
356 os._exit(-1) |
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
357 stop_event.wait(0.1) |
4 | 358 |
83 | 359 print('Stopping the external command') |
4 | 360 external.terminate() |
9 | 361 external.communicate() # Wait for the command to stop |
4 | 362 |
363 def __init__(self, command, env = None): | |
364 self.thread_stop = threading.Event() | |
365 self.thread = threading.Thread(target = self.ExternalCommandFunction, | |
366 args = (10, self.thread_stop, command, env)) | |
9 | 367 #self.daemon = True |
4 | 368 self.thread.start() |
369 | |
370 def stop(self): | |
371 self.thread_stop.set() | |
372 self.thread.join() | |
220 | 373 |
374 | |
222
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
220
diff
changeset
|
375 def AssertAlmostEqualRecursive(self, a, b, places = 7, ignoreKeys = []): |
220 | 376 if type(a) is dict: |
377 self.assertTrue(type(b) is dict) | |
378 self.assertEqual(a.keys(), b.keys()) | |
379 for key, value in a.items(): | |
222
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
220
diff
changeset
|
380 if not key in ignoreKeys: |
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
220
diff
changeset
|
381 AssertAlmostEqualRecursive(self, a[key], b[key], places) |
220 | 382 |
383 elif type(a) is list: | |
384 self.assertTrue(type(b) is list) | |
385 self.assertEqual(len(a), len(b)) | |
386 for i in range(len(a)): | |
387 AssertAlmostEqualRecursive(self, a[i], b[i], places) | |
388 | |
389 else: | |
390 self.assertAlmostEqual(a, b, places = places) | |
291
cfa785074c64
test_modify_transcode
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
391 |
cfa785074c64
test_modify_transcode
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
392 |
cfa785074c64
test_modify_transcode
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
393 def GetTransferSyntax(dicom): |
cfa785074c64
test_modify_transcode
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
394 with tempfile.NamedTemporaryFile(delete = True) as f: |
cfa785074c64
test_modify_transcode
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
395 f.write(dicom) |
cfa785074c64
test_modify_transcode
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
396 f.flush() |
318
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
397 |
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
398 with open(os.devnull, 'w') as devnull: |
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
399 data = subprocess.check_output([ FindExecutable('dcm2xml'), f.name ], |
bac7cc80f240
dicomweb: test_wado_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
304
diff
changeset
|
400 stderr = devnull) |
291
cfa785074c64
test_modify_transcode
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
401 |
cfa785074c64
test_modify_transcode
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
402 return re.search('<data-set xfer="(.*?)"', data).group(1) |
304 | 403 |
404 | |
405 def HasGdcmPlugin(orthanc): | |
406 plugins = DoGet(orthanc, '/plugins') | |
407 return ('gdcm' in plugins) | |
408 | |
409 | |
410 def _GetMaxImageDifference(im1, im2): | |
411 h = ImageChops.difference(im1, im2).histogram() | |
412 | |
413 if len(h) < 256: | |
414 raise Exception() | |
415 | |
416 i = len(h) - 1 | |
417 while h[i] == 0: | |
418 i -= 1 | |
419 | |
420 return i | |
421 | |
422 | |
423 def GetMaxImageDifference(im1, im2): | |
424 if im1.mode != im2.mode: | |
425 raise Exception('Incompatible image modes') | |
426 | |
427 if im1.mode == 'RGB': | |
428 red1, green1, blue1 = im1.split() | |
429 red2, green2, blue2 = im2.split() | |
430 return max([ _GetMaxImageDifference(red1, red2), | |
431 _GetMaxImageDifference(green1, green2), | |
432 _GetMaxImageDifference(blue1, blue2) ]) | |
433 else: | |
434 return _GetMaxImageDifference(im1, im2) | |
337
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
435 |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
436 |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
437 def DoPropFind(orthanc, uri, depth): |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
438 http = httplib2.Http() |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
439 http.follow_redirects = False |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
440 _SetupCredentials(orthanc, http) |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
441 |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
442 resp, content = http.request(orthanc['Url'] + uri, 'PROPFIND', headers = { 'Depth' : str(depth) }) |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
443 |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
444 if not (resp.status in [ 207 ]): |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
445 raise Exception(resp.status, resp) |
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
446 else: |
341
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
447 xml = minidom.parseString(content) |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
448 |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
449 if (xml.documentElement.nodeName != 'D:multistatus' or |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
450 xml.documentElement.attributes['xmlns:D'].value != 'DAV:'): |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
451 raise Exception() |
337
ec13ace43bde
trying webdav tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
318
diff
changeset
|
452 |
341
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
453 result = {} |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
454 |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
455 for i in xml.documentElement.childNodes: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
456 if i.nodeType == minidom.Node.ELEMENT_NODE: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
457 if i.nodeName != 'D:response': |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
458 raise Exception() |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
459 href = None |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
460 prop = None |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
461 for j in i.childNodes: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
462 if j.nodeType == minidom.Node.ELEMENT_NODE: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
463 if j.nodeName == 'D:href': |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
464 if href == None: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
465 href = unquote(j.firstChild.nodeValue) |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
466 else: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
467 raise Exception() |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
468 elif j.nodeName == 'D:propstat': |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
469 for k in j.childNodes: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
470 if k.nodeName == 'D:status': |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
471 if k.firstChild.nodeValue != 'HTTP/1.1 200 OK': |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
472 raise Exception() |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
473 elif k.nodeType == minidom.Node.ELEMENT_NODE: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
474 if (k.nodeName != 'D:prop' or |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
475 prop != None): |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
476 raise Exception() |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
477 prop = k |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
478 else: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
479 raise Exception() |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
480 if href == None or prop == None: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
481 raise Exception() |
342
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
482 |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
483 info = {} |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
484 |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
485 for j in prop.childNodes: |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
486 if j.nodeType == minidom.Node.ELEMENT_NODE: |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
487 if j.nodeName == 'D:displayname': |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
488 info['displayname'] = j.firstChild.nodeValue if j.firstChild != None else '' |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
489 elif j.nodeName == 'D:creationdate': |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
490 info['creationdate'] = j.firstChild.nodeValue |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
491 elif j.nodeName == 'D:getlastmodified': |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
492 info['lastmodified'] = j.firstChild.nodeValue |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
493 elif j.nodeName == 'D:resourcetype': |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
494 k = j.getElementsByTagName('D:collection') |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
495 info['folder'] = (len(k) == 1) |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
496 |
bf8369ea3ff1
more tests of webdav
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
341
diff
changeset
|
497 result[href] = info |
341
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
498 elif i.nodeType != minidom.Node.TEXT_NODE: |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
499 raise Exception() |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
500 |
66a36befb208
extending Toolbox.DoPropFind()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
337
diff
changeset
|
501 return result |