Add MS Sql Server

Author: annabaas

testcontainers/mssql.py

@@ -0,0 +1,37 @@
+from os import environ
+
+from testcontainers.core.generic import DbContainer
+
+
+class SqlServerContainer(DbContainer):
+    SQLSERVER_PASSWORD = environ.get("SQLSERVER_PASSWORD", "1Secure*Password1")
+    SQLSERVER_USER = "SA"
+    SQLSERVER_DBNAME = "tempdb"
+
+    def __init__(self, image="mcr.microsoft.com/mssql/server:2019-latest", **kwargs):
+        super(SqlServerContainer, self).__init__(image)
+        self.port_to_expose = 1433
+        self.with_exposed_ports(self.port_to_expose)
+        self.ACCEPT_EULA = 'Y'
+        self.MSSQL_PID = 'Developer'
+
+        if 'SQLSERVER_PASSWORD' in kwargs:
+            self.SQLSERVER_PASSWORD = kwargs['SQLSERVER_PASSWORD']
+        if 'SQLSERVER_USER' in kwargs:
+            self.SQLSERVER_USER = kwargs['SQLSERVER_USER']
+        if 'SQLSERVER_DBNAME' in kwargs:
+            self.SQLSERVER_DBNAME = kwargs['SQLSERVER_DBNAME']
+
+    def _configure(self):
+        self.with_env("SA_PASSWORD", self.SQLSERVER_PASSWORD)
+        self.with_env("SQLSERVER_USER", self.SQLSERVER_USER)
+        self.with_env("SQLSERVER_DBNAME", self.SQLSERVER_DBNAME)
+        self.with_env("ACCEPT_EULA", self.ACCEPT_EULA)
+        self.with_env("MSSQL_PID", self.MSSQL_PID)
+
+    def get_connection_url(self):
+        return super()._create_connection_url(dialect="mssql+pymssql",
+                                              username=self.SQLSERVER_USER,
+                                              password=self.SQLSERVER_PASSWORD,
+                                              db_name=self.SQLSERVER_DBNAME,
+                                              port=self.port_to_expose)

tests/test_db_containers.py

@@ -4,6 +4,7 @@
 
 from testcontainers.core.generic import GenericContainer
 from testcontainers.core.waiting_utils import wait_for
+from testcontainers.mssql import SqlServerContainer
 from testcontainers.mysql import MySqlContainer, MariaDbContainer
 from testcontainers.oracle import OracleDbContainer
 from testcontainers.postgres import PostgresContainer
@@ -100,3 +101,13 @@ def connect():
         cursor = db.restaurants.find({"borough": "Manhattan"})
         for document in cursor:
             print(document)
+
+
+def test_docker_run_mssql():
+    config = SqlServerContainer()
+    with config as mssql:
+        e = sqlalchemy.create_engine(mssql.get_connection_url())
+        result = e.execute('select @@servicename')
+        for row in result:
+            print(row[0])
+            assert row[0] == 'MSSQLSERVER'