bd version parameter added to constructor

Author: sepi

testcontainers/config.py

@@ -38,7 +38,6 @@
 }
 
 postgres_container = {
-    "image": "postgres:latest",
     "env": {
         "POSTGRES_USER": postgres_db["user"],
         "POSTGRES_PASSWORD": postgres_db["passwd"],

testcontainers/mysql.py

@@ -6,9 +6,9 @@
 
 
 class MySqlDockerContainer(DockerContainer):
-    def __init__(self, image='mysql:latest'):
-        super(DockerContainer, self).__init__()
-        self.image = image
+    def __init__(self, version='latest', image='mysql'):
+        super(self.__class__, self).__init__()
+        self.image = "{}:{}".format(image, version)
 
     def start(self):
         """

testcontainers/postgres.py

@@ -12,8 +12,8 @@ def start(self):
 
     @wait_container_is_ready()
     def connection(self):
-        return psycopg2.connect(**config.postgres_db)
+        return psycopg2.connect(self.image, **config.postgres_db)
 
-    def __init__(self, image="postgres:latest"):
+    def __init__(self, version='latest', image="postgres"):
         super(DockerContainer, self).__init__()
-        config.postgres_container["image"] = image
+        self.image = "{}:{}".format(image, version)

tests/test_docker_manager.py

@@ -15,7 +15,7 @@ def test_docker_run_selenium():
 
 
 def test_docker_run_mysql():
-    with MySqlDockerContainer() as mysql:
+    with MySqlDockerContainer('5.5') as mysql:
         conn = mysql.connection()
         cur = conn.cursor()