|
5 | 5 |
|
6 | 6 | class SqlServerContainer(DbContainer):
|
7 | 7 | SQLSERVER_PASSWORD = environ.get("SQLSERVER_PASSWORD", "1Secure*Password1")
|
8 |
| - SQLSERVER_USER = "SA" |
9 |
| - SQLSERVER_DBNAME = "tempdb" |
10 |
| - SQLSERVER_DRIVER = "ODBC Driver 17 for SQL Server" |
11 | 8 |
|
12 |
| - def __init__(self, image="mcr.microsoft.com/mssql/server:2019-latest", **kwargs): |
| 9 | + def __init__(self, image="mcr.microsoft.com/mssql/server:2019-latest", user="SA", password=None, |
| 10 | + port=1433, dbname="tempdb", driver="ODBC Driver 17 for SQL Server"): |
13 | 11 | super(SqlServerContainer, self).__init__(image)
|
14 |
| - self.port_to_expose = 1433 |
| 12 | + |
| 13 | + self.SQLSERVER_PASSWORD = password or self.SQLSERVER_PASSWORD |
| 14 | + self.port_to_expose = port |
| 15 | + self.SQLSERVER_USER = user |
| 16 | + self.SQLSERVER_DBNAME = dbname |
| 17 | + self.SQLSERVER_DRIVER = driver |
| 18 | + |
15 | 19 | self.with_exposed_ports(self.port_to_expose)
|
16 | 20 | self.ACCEPT_EULA = 'Y'
|
17 | 21 | self.MSSQL_PID = 'Developer'
|
18 | 22 |
|
19 |
| - if 'SQLSERVER_PASSWORD' in kwargs: |
20 |
| - self.SQLSERVER_PASSWORD = kwargs['SQLSERVER_PASSWORD'] |
21 |
| - if 'SQLSERVER_USER' in kwargs: |
22 |
| - self.SQLSERVER_USER = kwargs['SQLSERVER_USER'] |
23 |
| - if 'SQLSERVER_DBNAME' in kwargs: |
24 |
| - self.SQLSERVER_DBNAME = kwargs['SQLSERVER_DBNAME'] |
25 |
| - if 'SQLSERVER_DRIVER' in kwargs: |
26 |
| - self.SQLSERVER_DRIVER = kwargs['SQLSERVER_DRIVER'] |
27 |
| - |
28 | 23 | def _configure(self):
|
29 | 24 | self.with_env("SA_PASSWORD", self.SQLSERVER_PASSWORD)
|
30 | 25 | self.with_env("SQLSERVER_USER", self.SQLSERVER_USER)
|
|
0 commit comments