Mercurial > hg > orthanc-tests
annotate Tests/Run.py @ 8:f8d781d1d267
test_changes, test_archive, test_media_archive
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Wed, 17 Jun 2015 13:45:54 +0200 |
| parents | 6d645b3011e1 |
| children | 7b69a561f4d3 |
| rev | line source |
|---|---|
| 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', | |
|
8
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
38 default = 'localhost', |
| 1 | 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') | |
|
8
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
59 parser.add_argument('--docker', help = 'These tests are run from Docker', |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
60 action = 'store_true') |
| 6 | 61 parser.add_argument('options', metavar = 'N', nargs = '*', |
| 62 help='Arguments to Python unittest') | |
| 1 | 63 |
| 64 args = parser.parse_args() | |
| 65 | |
| 66 if not args.force: | |
| 67 print(""" | |
| 68 WARNING: This test will remove all the content of your | |
| 69 Orthanc instance running on %s! | |
| 70 | |
| 71 Are you sure ["yes" to go on]?""" % args.server) | |
| 72 | |
| 73 if sys.stdin.readline().strip() != 'yes': | |
| 74 print('Aborting...') | |
| 75 exit(0) | |
| 76 | |
| 77 | |
| 78 | |
| 79 ## | |
| 80 ## Generate the configuration file for the anciliary instance of | |
| 81 ## Orthanc | |
| 82 ## | |
| 83 | |
|
8
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
84 if args.docker: |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
85 args.server = GetDockerHostAddress() |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
86 |
| 1 | 87 CONFIG = '/tmp/Configuration.json' |
| 88 subprocess.check_call([ 'Orthanc', '--config=%s' % CONFIG ]) | |
| 89 | |
|
8
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
90 with open(CONFIG, 'rt') as f: |
| 1 | 91 config = f.read() |
| 92 | |
| 93 config = re.sub(r'("StorageDirectory"\s*:)\s*".*?"', r'\1 "/tmp/OrthancStorage"', config) | |
| 94 config = re.sub(r'("IndexDirectory"\s*:)\s*".*?"', r'\1 "/tmp/OrthancStorage"', config) | |
| 95 config = re.sub(r'("DicomAet"\s*:)\s*".*?"', r'\1 "ORTHANCTEST"', config) | |
|
8
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
96 config = re.sub(r'("DicomPort"\s*:)\s*.*?,', r'\1 5001,', config) |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
97 config = re.sub(r'("HttpPort"\s*:)\s*.*?,', r'\1 5000,', config) |
| 1 | 98 config = re.sub(r'("RemoteAccessAllowed"\s*:)\s*false', r'\1 true', config) |
| 99 config = re.sub(r'("AuthenticationEnabled"\s*:)\s*false', r'\1 true', config) | |
|
8
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
100 config = re.sub(r'("RegisteredUsers"\s*:)\s*{', r'\1 { "alice" : "orthanctest"', config) |
| 1 | 101 config = re.sub(r'("DicomModalities"\s*:)\s*{', r'\1 { "orthanc" : [ "%s", "%s", "%s" ]' % |
| 102 (args.aet, args.server, args.dicom), config) | |
| 103 | |
|
8
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
104 with open(CONFIG, 'wt') as f: |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
105 f.write(config) |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
106 |
| 1 | 107 localOrthanc = ExternalCommandThread([ |
| 108 'Orthanc', CONFIG, #'--verbose' | |
| 109 ]) | |
| 110 | |
| 111 | |
|
8
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
112 LOCAL = DefineOrthanc(aet = 'ORTHANCTEST', |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
113 url = 'http://localhost:5000', |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
114 dicomPort = 5001, |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
115 username = 'alice', |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
116 password = 'orthanctest') |
|
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
117 |
| 1 | 118 REMOTE = DefineOrthanc(url = 'http://%s:%d/' % (args.server, args.rest), |
| 119 username = args.username, | |
| 120 password = args.password, | |
| 121 aet = args.aet, | |
| 122 dicomPort = args.dicom) | |
| 123 | |
| 124 | |
| 125 | |
|
8
f8d781d1d267
test_changes, test_archive, test_media_archive
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
6
diff
changeset
|
126 print('Parameters of the instance of Orthanc being tested:') |
| 3 | 127 pprint.pprint(REMOTE) |
| 2 | 128 print('') |
| 1 | 129 |
| 130 | |
| 2 | 131 print('Waiting for the internal Orthanc to start...') |
| 132 while True: | |
| 133 try: | |
| 134 DoGet(LOCAL, '/instances') | |
| 135 break | |
| 136 except: | |
| 137 time.sleep(0.1) | |
| 1 | 138 |
| 139 | |
| 140 try: | |
| 3 | 141 print('\nStarting the tests...') |
| 2 | 142 SetOrthancParameters(LOCAL, REMOTE) |
| 6 | 143 unittest.main(argv = [ sys.argv[0] ] + args.options) |
| 1 | 144 |
| 145 finally: | |
| 3 | 146 print('\nDone') |
| 147 | |
| 1 | 148 # The tests have stopped or "Ctrl-C" has been hit |
| 149 try: | |
| 150 localOrthanc.stop() | |
| 151 except: | |
| 152 pass |