changeset 1:08dadea8f40a

start
author Sebastien Jodogne <s.jodogne@gmail.com>
date Tue, 16 Jun 2015 18:49:29 +0200
parents cc43b57242a4
children a15734e7f0af
files Database/DummyCT.dcm ExternalCommandThread.py Run.py Toolbox.py
diffstat 4 files changed, 296 insertions(+), 25 deletions(-) [+]
line wrap: on
line diff
Binary file Database/DummyCT.dcm has changed
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/ExternalCommandThread.py	Tue Jun 16 18:49:29 2015 +0200
@@ -0,0 +1,50 @@
+#!/usr/bin/python
+
+# Orthanc - A Lightweight, RESTful DICOM Store
+# Copyright (C) 2012-2015 Sebastien Jodogne, Medical Physics
+# Department, University Hospital of Liege, Belgium
+#
+# This program is free software: you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+
+import os
+import signal
+import subprocess
+import threading
+
+class ExternalCommandThread:
+    @staticmethod
+    def ExternalCommandFunction(arg, stop_event, command, env):
+        external = subprocess.Popen(command, env = env)
+
+        while (not stop_event.is_set()):
+            error = external.poll()
+            if error != None:
+                # http://stackoverflow.com/a/1489838/881731
+                os._exit(-1)
+            stop_event.wait(0.1)
+
+        print 'Stopping the external command'
+        external.terminate()
+
+    def __init__(self, command, env = None):
+        self.thread_stop = threading.Event()
+        self.thread = threading.Thread(target = self.ExternalCommandFunction, 
+                                       args = (10, self.thread_stop, command, env))
+        self.daemon = True
+        self.thread.start()
+
+    def stop(self):
+        self.thread_stop.set()
+        self.thread.join()
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Run.py	Tue Jun 16 18:49:29 2015 +0200
@@ -0,0 +1,180 @@
+#!/usr/bin/python
+
+# Orthanc - A Lightweight, RESTful DICOM Store
+# Copyright (C) 2012-2015 Sebastien Jodogne, Medical Physics
+# Department, University Hospital of Liege, Belgium
+#
+# This program is free software: you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+
+# 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
+
+
+
+import re
+import sys
+import argparse
+import subprocess
+import unittest
+
+from ExternalCommandThread import *
+from Toolbox import *
+
+
+##
+## Parse the command-line arguments
+##
+
+parser = argparse.ArgumentParser(description = 'Run the integration tests on some instance of Orthanc.')
+parser.add_argument('--server', 
+                    default = GetDockerHostAddress(),
+                    help = 'Address of the Orthanc server to test')
+parser.add_argument('--aet',
+                    default = 'ORTHANC',
+                    help = 'AET of the Orthanc instance to test')
+parser.add_argument('--dicom',
+                    type = int,
+                    default = 4242,
+                    help = 'DICOM port of the Orthanc instance to test')
+parser.add_argument('--rest',
+                    type = int,
+                    default = 8042,
+                    help = 'Port to the REST API')
+parser.add_argument('--username',
+                    default = None,
+                    help = 'Username to the REST API')
+parser.add_argument('--password',
+                    default = None,
+                    help = 'Password to the REST API')
+parser.add_argument("--force", help = "Do not warn the user",
+                    action = "store_true")
+
+args = parser.parse_args()
+
+if not args.force:
+    print("""
+WARNING: This test will remove all the content of your
+Orthanc instance running on %s!
+
+Are you sure ["yes" to go on]?""" % args.server)
+
+    if sys.stdin.readline().strip() != 'yes':
+        print('Aborting...')
+        exit(0)
+
+
+
+## 
+## Generate the configuration file for the anciliary instance of
+## Orthanc
+##
+
+CONFIG = '/tmp/Configuration.json'
+subprocess.check_call([ 'Orthanc', '--config=%s' % CONFIG ])
+
+with open(CONFIG, 'r') as f:
+    config = f.read()
+
+config = re.sub(r'("StorageDirectory"\s*:)\s*".*?"', r'\1 "/tmp/OrthancStorage"', config)
+config = re.sub(r'("IndexDirectory"\s*:)\s*".*?"', r'\1 "/tmp/OrthancStorage"', config)
+config = re.sub(r'("DicomAet"\s*:)\s*".*?"', r'\1 "ORTHANCTEST"', config)
+config = re.sub(r'("RemoteAccessAllowed"\s*:)\s*false', r'\1 true', config)
+config = re.sub(r'("AuthenticationEnabled"\s*:)\s*false', r'\1 true', config)
+config = re.sub(r'("RegisteredUsers"\s*:)\s*{', r'\1 { "alice" : [ "orthanctest" ]', config)
+config = re.sub(r'("DicomModalities"\s*:)\s*{', r'\1 { "orthanc" : [ "%s", "%s", "%s" ]' % 
+                (args.aet, args.server, args.dicom), config)
+
+localOrthanc = ExternalCommandThread([ 
+        'Orthanc', CONFIG, #'--verbose'
+        ])
+
+
+LOCAL = DefineOrthanc(aet = 'ORTHANCTEST')
+REMOTE = DefineOrthanc(url = 'http://%s:%d/' % (args.server, args.rest),
+                       username = args.username,
+                       password = args.password,
+                       aet = args.aet,
+                       dicomPort = args.dicom)
+
+
+
+class Orthanc(unittest.TestCase):
+    def setUp(self):
+        DropOrthanc(LOCAL)
+        DropOrthanc(REMOTE)
+
+    def test_system(self):
+        self.assertTrue('Version' in DoGet(REMOTE, '/system'))
+        self.assertEqual('0', DoGet(REMOTE, '/statistics')['TotalDiskSize'])
+        self.assertEqual('0', DoGet(REMOTE, '/statistics')['TotalUncompressedSize'])
+
+    def test_upload(self):
+        u = UploadInstance(REMOTE, 'DummyCT.dcm')
+        self.assertEqual('Success', u['Status'])
+        u = UploadInstance(REMOTE, 'DummyCT.dcm')
+        self.assertEqual('AlreadyStored', u['Status'])
+        self.assertEqual(1, len(DoGet(REMOTE, '/patients')))
+        self.assertEqual(1, len(DoGet(REMOTE, '/studies')))
+        self.assertEqual(1, len(DoGet(REMOTE, '/series')))
+        self.assertEqual(1, len(DoGet(REMOTE, '/instances')))
+
+        i = DoGet(REMOTE, '/instances/%s/simplified-tags' % u['ID'])
+        self.assertEqual('20070101', i['StudyDate'])
+
+
+    def test_rest_grid(self):
+        i = UploadInstance(REMOTE, 'DummyCT.dcm')['ID']
+        instance = DoGet(REMOTE, '/instances/%s' % i)
+        self.assertEqual(i, instance['ID'])
+        self.assertEqual('1.2.840.113619.2.176.2025.1499492.7040.1171286242.109',
+                         instance['MainDicomTags']['SOPInstanceUID'])
+
+        series = DoGet(REMOTE, '/series/%s' % instance['ParentSeries'])
+        self.assertEqual('1.2.840.113619.2.176.2025.1499492.7391.1171285944.394', 
+                         series['MainDicomTags']['SeriesInstanceUID'])
+
+        study = DoGet(REMOTE, '/studies/%s' % series['ParentStudy'])
+        self.assertEqual('1.2.840.113619.2.176.2025.1499492.7391.1171285944.390',
+                         study['MainDicomTags']['StudyInstanceUID'])
+
+        patient = DoGet(REMOTE, '/patients/%s' % study['ParentPatient'])
+        self.assertEqual('ozp00SjY2xG',
+                         patient['MainDicomTags']['PatientID'])
+
+        dicom = DoGet(REMOTE, '/instances/%s/file' % instance['ID'])
+        self.assertEqual(2472, len(dicom))
+        self.assertEqual('3e29b869978b6db4886355a2b1132124', ComputeMD5(dicom))
+        self.assertEqual(1, len(DoGet(REMOTE, '/instances/%s/frames' % i)))
+        self.assertEqual('TWINOW', DoGet(REMOTE, '/instances/%s/simplified-tags' % i)['StationName'])
+        self.assertEqual('TWINOW', DoGet(REMOTE, '/instances/%s/tags' % i)['0008,1010']['Value'])
+
+
+try:
+    print('Waiting for the internal Orthanc to start...')
+    while True:
+        try:
+            DoGet(LOCAL, '/instances')
+            break
+        except:
+            time.sleep(0.1)
+
+    print('Starting the tests...')
+    unittest.main(argv = [ sys.argv[0] ]) #argv = args)
+
+finally:
+    # The tests have stopped or "Ctrl-C" has been hit
+    try:
+        localOrthanc.stop()
+    except:
+        pass
--- a/Toolbox.py	Mon Jun 15 17:44:12 2015 +0200
+++ b/Toolbox.py	Tue Jun 16 18:49:29 2015 +0200
@@ -1,18 +1,38 @@
 #!/usr/bin/python
 
-# sudo docker run --rm -v `pwd`/Toolbox.py:/tmp/Toolbox.py:ro --entrypoint python jodogne/orthanc-tests /tmp/Toolbox.py
+# Orthanc - A Lightweight, RESTful DICOM Store
+# Copyright (C) 2012-2015 Sebastien Jodogne, Medical Physics
+# Department, University Hospital of Liege, Belgium
+#
+# This program is free software: you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
 
 
+
+from PIL import Image
+from urllib import urlencode
 import hashlib
 import httplib2
 import json
 import os.path
-from PIL import Image
-import zipfile
+import re
+import subprocess
 import time
-from urllib import urlencode
+import zipfile
 
 
+HERE = os.path.dirname(__file__)
+
 
 # http://stackoverflow.com/a/1313868/881731
 try:
@@ -21,19 +41,33 @@
     from StringIO import StringIO
 
 
+def DefineOrthanc(url = 'http://localhost:8042',
+                  username = None,
+                  password = None,
+                  aet = 'ORTHANC',
+                  dicomPort = 4242):
+    m = re.match(r'(http|https)://([^:]+):([^@]+)@([^@]+)', url)
+    if m != None:
+        url = m.groups()[0] + '://' + m.groups()[3]
+        username = m.groups()[1]
+        password = m.groups()[2]
 
-def CreateOrthanc(url = 'http://localhost:8042',
-                  username = None,
-                  password = None):
     if not url.endswith('/'):
         url += '/'
 
-    return [ url, username, password ]
+    return {
+        'Url' : url, 
+        'Username' : username,
+        'Password' : password,
+        'DicomAet' : aet,
+        'DicomPort' : dicomPort
+        }
 
 
 def _SetupCredentials(orthanc, http):
-    if orthanc[1] != None and orthanc[2] != None:
-        http.add_credentials(orthanc[1], orthanc[2])
+    if (orthanc['Username'] != None and 
+        orthanc['Password'] != None):
+        http.add_credentials(orthanc['Username'], orthanc['Password'])
 
 
 def DoGet(orthanc, uri, data = {}, body = None, headers = {}):
@@ -44,7 +78,7 @@
     http = httplib2.Http()
     _SetupCredentials(orthanc, http)
 
-    resp, content = http.request(orthanc[0] + uri + d, 'GET', body = body,
+    resp, content = http.request(orthanc['Url'] + uri + d, 'GET', body = body,
                                  headers = headers)
     if not (resp.status in [ 200 ]):
         raise Exception(resp.status)
@@ -68,7 +102,7 @@
     
     headers['expect'] = ''
 
-    resp, content = http.request(orthanc[0] + uri, method,
+    resp, content = http.request(orthanc['Url'] + uri, method,
                                  body = body,
                                  headers = headers)
     if not (resp.status in [ 200, 302 ]):
@@ -83,7 +117,7 @@
     http = httplib2.Http()
     _SetupCredentials(orthanc, http)
 
-    resp, content = http.request(orthanc[0] + uri, 'DELETE')
+    resp, content = http.request(orthanc['Url'] + uri, 'DELETE')
     if not (resp.status in [ 200 ]):
         raise Exception(resp.status)
     else:
@@ -99,19 +133,21 @@
     return _DoPutOrPost(orthanc, uri, 'POST', data, contentType, headers)
 
 def UploadInstance(orthanc, filename):
-    p = os.path.join(HERE, DICOM_DB, filename)
+    global HERE
+    p = os.path.join(HERE, 'Database', filename)
     f = open(p, 'rb')
     d = f.read()
     f.close()
     return DoPost(orthanc, '/instances', d, 'application/dicom')
 
 def UploadFolder(orthanc, path):
-     p = os.path.join(HERE, DICOM_DB, path)
-     for i in os.listdir(p):
-       try:
-         UploadInstance(orthanc, os.path.join(path, i))
-       except:
-         pass
+    global HERE
+    p = os.path.join(HERE, 'Database', path)
+    for i in os.listdir(p):
+        try:
+            UploadInstance(orthanc, os.path.join(path, i))
+        except:
+            pass
 
 def DropOrthanc(orthanc):
     # Reset the Lua callbacks
@@ -137,15 +173,20 @@
     s = DoGet(orthanc, uri)
     return zipfile.ZipFile(StringIO(s), "r")
 
-def IsDefinedInLua(name):
+def IsDefinedInLua(orthanc, name):
     s = DoPost(orthanc, '/tools/execute-script', 'print(type(%s))' % name, 'application/lua')
     return (s.strip() != 'nil')
 
-def WaitEmpty():
+def WaitEmpty(orthanc):
     while True:
-        if len(orthanc, DoGet('/instances')) == 0:
+        if len(DoGet(orthanc, '/instances')) == 0:
             return
         time.sleep(0.1)
 
-
-print DoGet(CreateOrthanc('http://192.168.215.82:8042'), '/system')
+def GetDockerHostAddress():
+    route = subprocess.check_output([ '/sbin/ip', 'route' ])
+    m = re.search(r'default via ([0-9.]+)', route)
+    if m == None:
+        return 'localhost'
+    else:
+        return m.groups()[0]