Feature/add greenplum (#119)

* Add parameters to Postgress testcontainer to enable Greenplum support

* Add parameters to Postgress testcontainer to enable Greenplum support

* Shorten length of line to pass flake8 test

Co-authored-by: anna <[email protected]>

Author: annabaas

testcontainers/postgres.py

@@ -32,9 +32,13 @@ class PostgresContainer(DbContainer):
     POSTGRES_PASSWORD = os.environ.get("POSTGRES_PASSWORD", "test")
     POSTGRES_DB = os.environ.get("POSTGRES_DB", "test")
 
-    def __init__(self, image="postgres:latest"):
+    def __init__(self, image="postgres:latest", port=5432, user=None, password=None, dbname=None):
         super(PostgresContainer, self).__init__(image=image)
-        self.port_to_expose = 5432
+        self.POSTGRES_USER = user or self.POSTGRES_USER
+        self.POSTGRES_PASSWORD = password or self.POSTGRES_PASSWORD
+        self.POSTGRES_DB = dbname or self.POSTGRES_DB
+        self.port_to_expose = port
+
         self.with_exposed_ports(self.port_to_expose)
 
     def _configure(self):

tests/test_db_containers.py

@@ -31,6 +31,16 @@ def test_docker_run_postgress():
             print("server version:", row[0])
 
 
+def test_docker_run_greenplum():
+    postgres_container = PostgresContainer("datagrip/greenplum:6.8",
+                                           user="guest", password="guest", dbname="guest")
+    with postgres_container as postgres:
+        e = sqlalchemy.create_engine(postgres.get_connection_url())
+        result = e.execute("select version()")
+        for row in result:
+            print("server version:", row[0])
+
+
 def test_docker_run_mariadb():
     mariadb_container = MySqlContainer("mariadb:10.2.9")
     with mariadb_container as mariadb: