Skip to content

Commit 4c066b5

Browse files
committed
Review comments: add explicit parameters to __init__
1 parent 3cf948e commit 4c066b5

File tree

1 file changed

+9
-14
lines changed

1 file changed

+9
-14
lines changed

testcontainers/mssql.py

Lines changed: 9 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -5,26 +5,21 @@
55

66
class SqlServerContainer(DbContainer):
77
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"
118

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"):
1311
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+
1519
self.with_exposed_ports(self.port_to_expose)
1620
self.ACCEPT_EULA = 'Y'
1721
self.MSSQL_PID = 'Developer'
1822

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-
2823
def _configure(self):
2924
self.with_env("SA_PASSWORD", self.SQLSERVER_PASSWORD)
3025
self.with_env("SQLSERVER_USER", self.SQLSERVER_USER)

0 commit comments

Comments
 (0)