annotate Run.py @ 2:a15734e7f0af

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