|
1
|
1 #!/usr/bin/python
|
|
|
2
|
|
|
3 # Orthanc - A Lightweight, RESTful DICOM Store
|
|
|
4 # Copyright (C) 2012-2015 Sebastien Jodogne, Medical Physics
|
|
|
5 # Department, University Hospital of Liege, Belgium
|
|
|
6 #
|
|
|
7 # This program is free software: you can redistribute it and/or
|
|
|
8 # modify it under the terms of the GNU General Public License as
|
|
|
9 # published by the Free Software Foundation, either version 3 of the
|
|
|
10 # License, or (at your option) any later version.
|
|
|
11 #
|
|
|
12 # This program is distributed in the hope that it will be useful, but
|
|
|
13 # WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
15 # General Public License for more details.
|
|
|
16 #
|
|
|
17 # You should have received a copy of the GNU General Public License
|
|
|
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
19
|
|
|
20
|
|
|
21 import re
|
|
|
22 import sys
|
|
|
23 import argparse
|
|
|
24 import subprocess
|
|
|
25 import unittest
|
|
3
|
26 import pprint
|
|
1
|
27
|
|
2
|
28 from Tests import *
|
|
5
|
29 from Toolbox import DoGet
|
|
1
|
30
|
|
|
31
|
|
|
32 ##
|
|
|
33 ## Parse the command-line arguments
|
|
|
34 ##
|
|
|
35
|
|
|
36 parser = argparse.ArgumentParser(description = 'Run the integration tests on some instance of Orthanc.')
|
|
|
37 parser.add_argument('--server',
|
|
|
38 default = GetDockerHostAddress(),
|
|
|
39 help = 'Address of the Orthanc server to test')
|
|
|
40 parser.add_argument('--aet',
|
|
|
41 default = 'ORTHANC',
|
|
|
42 help = 'AET of the Orthanc instance to test')
|
|
|
43 parser.add_argument('--dicom',
|
|
|
44 type = int,
|
|
|
45 default = 4242,
|
|
|
46 help = 'DICOM port of the Orthanc instance to test')
|
|
|
47 parser.add_argument('--rest',
|
|
|
48 type = int,
|
|
|
49 default = 8042,
|
|
|
50 help = 'Port to the REST API')
|
|
|
51 parser.add_argument('--username',
|
|
2
|
52 default = 'alice',
|
|
1
|
53 help = 'Username to the REST API')
|
|
|
54 parser.add_argument('--password',
|
|
2
|
55 default = 'orthanctest',
|
|
1
|
56 help = 'Password to the REST API')
|
|
2
|
57 parser.add_argument('--force', help = 'Do not warn the user',
|
|
|
58 action = 'store_true')
|
|
1
|
59
|
|
|
60 args = parser.parse_args()
|
|
|
61
|
|
|
62 if not args.force:
|
|
|
63 print("""
|
|
|
64 WARNING: This test will remove all the content of your
|
|
|
65 Orthanc instance running on %s!
|
|
|
66
|
|
|
67 Are you sure ["yes" to go on]?""" % args.server)
|
|
|
68
|
|
|
69 if sys.stdin.readline().strip() != 'yes':
|
|
|
70 print('Aborting...')
|
|
|
71 exit(0)
|
|
|
72
|
|
|
73
|
|
|
74
|
|
|
75 ##
|
|
|
76 ## Generate the configuration file for the anciliary instance of
|
|
|
77 ## Orthanc
|
|
|
78 ##
|
|
|
79
|
|
|
80 CONFIG = '/tmp/Configuration.json'
|
|
|
81 subprocess.check_call([ 'Orthanc', '--config=%s' % CONFIG ])
|
|
|
82
|
|
|
83 with open(CONFIG, 'r') as f:
|
|
|
84 config = f.read()
|
|
|
85
|
|
|
86 config = re.sub(r'("StorageDirectory"\s*:)\s*".*?"', r'\1 "/tmp/OrthancStorage"', config)
|
|
|
87 config = re.sub(r'("IndexDirectory"\s*:)\s*".*?"', r'\1 "/tmp/OrthancStorage"', config)
|
|
|
88 config = re.sub(r'("DicomAet"\s*:)\s*".*?"', r'\1 "ORTHANCTEST"', config)
|
|
|
89 config = re.sub(r'("RemoteAccessAllowed"\s*:)\s*false', r'\1 true', config)
|
|
|
90 config = re.sub(r'("AuthenticationEnabled"\s*:)\s*false', r'\1 true', config)
|
|
|
91 config = re.sub(r'("RegisteredUsers"\s*:)\s*{', r'\1 { "alice" : [ "orthanctest" ]', config)
|
|
|
92 config = re.sub(r'("DicomModalities"\s*:)\s*{', r'\1 { "orthanc" : [ "%s", "%s", "%s" ]' %
|
|
|
93 (args.aet, args.server, args.dicom), config)
|
|
|
94
|
|
|
95 localOrthanc = ExternalCommandThread([
|
|
|
96 'Orthanc', CONFIG, #'--verbose'
|
|
|
97 ])
|
|
|
98
|
|
|
99
|
|
|
100 LOCAL = DefineOrthanc(aet = 'ORTHANCTEST')
|
|
|
101 REMOTE = DefineOrthanc(url = 'http://%s:%d/' % (args.server, args.rest),
|
|
|
102 username = args.username,
|
|
|
103 password = args.password,
|
|
|
104 aet = args.aet,
|
|
|
105 dicomPort = args.dicom)
|
|
|
106
|
|
|
107
|
|
|
108
|
|
2
|
109 print('Parameters of the instance of Orthanc to test:')
|
|
3
|
110 pprint.pprint(REMOTE)
|
|
2
|
111 print('')
|
|
1
|
112
|
|
|
113
|
|
2
|
114 print('Waiting for the internal Orthanc to start...')
|
|
|
115 while True:
|
|
|
116 try:
|
|
|
117 DoGet(LOCAL, '/instances')
|
|
|
118 break
|
|
|
119 except:
|
|
|
120 time.sleep(0.1)
|
|
1
|
121
|
|
|
122
|
|
|
123 try:
|
|
3
|
124 print('\nStarting the tests...')
|
|
2
|
125 SetOrthancParameters(LOCAL, REMOTE)
|
|
5
|
126 unittest.main(argv = [ sys.argv[0] ])
|
|
1
|
127
|
|
|
128 finally:
|
|
3
|
129 print('\nDone')
|
|
|
130
|
|
1
|
131 # The tests have stopped or "Ctrl-C" has been hit
|
|
|
132 try:
|
|
|
133 localOrthanc.stop()
|
|
|
134 except:
|
|
|
135 pass
|
