Added MongoDb container (#47)

* Added MongoDb container

* remove comments from test

* remove literal for port exposed

* added assertion on mongodb test

* Removed unnecessary print

Author: dabrign

docs/database.rst

@@ -1,7 +1,7 @@
 Database containers
 ===================
 
-Allows to spin up docker database images such as MySQL, PostgreSQL, MariaDB and Oracle XE.
+Allows to spin up docker database images such as MySQL, PostgreSQL, MariaDB, Oracle XE and MongoDb.
 
 MySQL example
 -------------
@@ -81,3 +81,35 @@ Elasticsearch
     with es:
         es.get_url()  # gives you the http URL to connect to Elasticsearch
 
+MongoDb
+-----------
+
+Example of MongoDb database usage:
+
+::
+
+    def test_docker_run_mongodb():
+        mongo_container = MongoDbContainer("mongo:latest")
+        with mongo_container as mongo:
+            db = mongo.get_connection_client().test
+            result = db.restaurants.insert_one(
+                {
+                    "address": {
+                        "street": "2 Avenue",
+                        "zipcode": "10075",
+                        "building": "1480",
+                        "coord": [-73.9557413, 40.7720266]
+                    },
+                    "borough": "Manhattan",
+                    "cuisine": "Italian",
+                    "name": "Vella",
+                    "restaurant_id": "41704620"
+                }
+            )
+            print(result.inserted_id)
+            cursor = db.restaurants.find({"borough": "Manhattan"})
+            for document in cursor:
+                print(document)
+
+Connection is made using pymongo package and MongoClient class.
+Alternatively, you can use get_connection_url method to use the driver that better fits for your use case.

docs/index.rst

@@ -16,6 +16,7 @@ Currently available features:
 - PostgreSQL db container
 - Google Cloud PubSub emulator container
 - Elasticsearch container
+- MongoDb container
 - Generic Docker containers
 
 Installation
@@ -33,6 +34,7 @@ and can be installed using pip, depending on which containers you need:
    pip install testcontainers[postgresql]
    pip install testcontainers[selenium]
    pip install testcontainers[google-cloud-pubsub]
+   pip install testcontainers[mongodb]
    # or with multiple
    pip install testcontainers[mysql,postgresql,selenium,google-cloud-pubsub]
 

setup.py

@@ -53,6 +53,7 @@
         'postgresql': ['sqlalchemy', 'psycopg2-binary'],
         'selenium': ['selenium==2.53.1'],
         'google-cloud-pubsub': ['google-cloud-pubsub'],
+        'mongo': ['pymongo']
     },
     long_description_content_type="text/markdown",
     long_description=long_description,

testcontainers/mongodb.py

@@ -0,0 +1,43 @@
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+import os
+
+from pymongo import MongoClient
+
+from testcontainers.core.generic import DockerContainer
+
+
+class MongoDbContainer(DockerContainer):
+    MONGO_INITDB_ROOT_USERNAME = os.environ.get("MONGO_INITDB_ROOT_USERNAME", "test")
+    MONGO_INITDB_ROOT_PASSWORD = os.environ.get("MONGO_INITDB_ROOT_PASSWORD", "test")
+    MONGO_DB = os.environ.get("MONGO_DB", "test")
+
+    def __init__(self, image="mongodb:latest"):
+        super(MongoDbContainer, self).__init__(image=image)
+        self.command="mongo"
+        self.port_to_expose = 27017
+        self.with_exposed_ports(self.port_to_expose)
+
+    def _configure(self):
+
+        self.with_env("MONGO_INITDB_ROOT_USERNAME", self.MONGO_INITDB_ROOT_USERNAME)
+        self.with_env("MONGO_INITDB_ROOT_PASSWORD", self.MONGO_INITDB_ROOT_PASSWORD)
+        self.with_env("MONGO_DB", self.MONGO_DB)
+
+    def get_connection_url(self):
+        return "mongodb://{}:{}".format(self.get_container_host_ip(), self.get_exposed_port(self.port_to_expose))
+
+    def get_connection_client(self):
+        return MongoClient("mongodb://{}:{}".format(self.get_container_host_ip(),
+                                                    self.get_exposed_port(self.port_to_expose)))
+

tests/test_db_containers.py

@@ -7,6 +7,7 @@
 from testcontainers.mysql import MySqlContainer, MariaDbContainer
 from testcontainers.oracle import OracleDbContainer
 from testcontainers.postgres import PostgresContainer
+from testcontainers.mongodb import MongoDbContainer
 
 
 def test_docker_run_mysql():
@@ -50,6 +51,27 @@ def test_docker_run_oracle():
         assert {row[0] for row in result} == versions
 
 
+def test_docker_run_mongodb():
+    mongo_container = MongoDbContainer("mongo:latest")
+    with mongo_container as mongo:
+        db = mongo.get_connection_client().test
+        doc = {
+            "address": {
+                "street": "2 Avenue",
+                "zipcode": "10075",
+                "building": "1480",
+                "coord": [-73.9557413, 40.7720266]
+            },
+            "borough": "Manhattan",
+            "cuisine": "Italian",
+            "name": "Vella",
+            "restaurant_id": "41704620"
+        }
+        result = db.restaurants.insert_one(doc)
+        cursor = db.restaurants.find({"borough": "Manhattan"})
+        assert cursor.next()['restaurant_id'] == doc['restaurant_id']
+
+
 def test_docker_generic_db():
     mongo_container = GenericContainer("mongo:latest")
     mongo_container.with_bind_ports(27017, 27017)