Mercurial > hg > orthanc-tests

comparison PerfsDb/DbServer.py @ 156:f1a75985caa8

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
first Db test framework - work in progress
author am@osimis.io
date Thu, 16 Aug 2018 17:13:32 +0200
parents
children df1f9946571c
comparison
equal deleted inserted replaced
155:e0996602b306 156:f1a75985caa8
1 import typing
2 import subprocess
3 import time
4
5 from DbType import DbType
6
7 class DbServer:
8
9 class DockerDefinition:
10
11 def __init__(self, image: str, internalPort: int, envVars: typing.Dict[str, str], storagePath: str, command: typing.List[str]=None):
12 self.image = image
13 self.internalPort = internalPort
14 self.envVars = envVars
15 self.storagePath = storagePath
16 self.command = command
17
18 def __init__(self, dbType: DbType, port: int):
19
20 self.port = port
21 self.dbType = dbType
22
23 self._containerId = None
24 self._label = None
25
26 def setLabel(self, label: str):
27 self._label = label
28
29 def launch(self):
30 dockerDefinition = self.getDockerDefinition()
31
32 # check if the container is already running
33 ret = subprocess.call([
34 "docker",
35 "top",
36 self._label
37 ])
38 if ret == 0:
39 print("DbServer is already running")
40 return
41
42 # create a volume (if it already exists, it wont be modified)
43 subprocess.check_call([
44 "docker",
45 "volume",
46 "create",
47 "--name=" + self._label
48 ])
49
50 dockerRunCommand = [
51 "docker",
52 "run",
53 "-d",
54 "--name=" + self._label,
55 "-p", str(self.port) + ":" + str(dockerDefinition.internalPort),
56 "--volume=" + self._label + ":" + dockerDefinition.storagePath
57 ]
58
59 if len(dockerDefinition.envVars) > 0:
60 for k,v in dockerDefinition.envVars.items():
61 dockerRunCommand.extend(["--env", k + "=" + v])
62
63 dockerRunCommand.append(
64 dockerDefinition.image
65 )
66
67 if dockerDefinition.command is not None:
68 dockerRunCommand.extend(
69 dockerDefinition.command
70 )
71
72 print("Launching DbServer")
73 subprocess.check_call(dockerRunCommand)
74
75 print("Waiting for DbServer to be ready")
76
77 # wait until its port is open
78 retryCounter = 0
79 connected = False
80 while not connected and retryCounter < 30:
81 time.sleep(1)
82 connected = subprocess.call(["nc", "-z", "localhost", str(self.port)]) == 0
83 if retryCounter >= 30:
84 print("DbServer still not ready after 30 sec")
85 raise TimeoutError
86
87 def stop(self):
88 subprocess.check_call([
89 "docker",
90 "stop",
91 self._label
92 ])
93 subprocess.check_call([
94 "docker",
95 "rm",
96 self._label
97 ])
98
99 def clear(self):
100 # remove the volume
101 self.stop()
102 subprocess.check_call([
103 "docker",
104 "volume",
105 "rm",
106 self._label
107 ])
108
109
110 def getDockerDefinition(self):
111 if self.dbType == DbType.MySQL:
112 return DbServer.DockerDefinition(
113 image="mysql:8.0",
114 internalPort=3306,
115 envVars={
116 "MYSQL_PASSWORD": "orthanc",
117 "MYSQL_USER": "orthanc",
118 "MYSQL_DATABASE": "orthanc",
119 "MYSQL_ROOT_PASSWORD": "foo-root"
120 },
121 storagePath="/var/lib/mysql",
122 command=["mysqld", "--default-authentication-plugin=mysql_native_password", "--log-bin-trust-function-creators=1"]
123 )
124 elif self.dbType == DbType.MSSQL:
125 return DbServer.DockerDefinition(
126 image="microsoft/mssql-server-linux",
127 internalPort=1433,
128 envVars={
129 "ACCEPT_EULA": "Y",
130 "SA_PASSWORD": "MyStrOngPa55word!"
131 },
132 storagePath="/var/opt/mssql/data"
133 )
134 elif self.dbType == DbType.PG9 or self.dbType == DbType.PG10:
135 if self.dbType == DbType.PG9:
136 image = "postgres:9"
137 elif self.dbType == DbType.PG10:
138 image = "postgres:10"
139 return DbServer.DockerDefinition(
140 image=image,
141 internalPort=5432,
142 envVars={
143 },
144 storagePath="/var/lib/postgresql/data"
145 )
146