Method test_db_setup

st2common/tests/unit/test_db.py:360–390

(self, mock_mongoengine)

    @mock.patch("st2common.models.db.mongoengine")
    def test_db_setup(self, mock_mongoengine):
        db_setup(
            db_name="name",
            db_host="host",
            db_port=12345,
            username="username",
            password="password",
            authentication_mechanism="MONGODB-X509",
            ensure_indexes=False,
        )

        call_args = mock_mongoengine.connection.connect.call_args_list[0][0]
        call_kwargs = mock_mongoengine.connection.connect.call_args_list[0][1]

        self.assertEqual(call_args, ("name",))
        self.assertEqual(
            call_kwargs,
            {
                "host": "host",
                "port": 12345,
                "username": "username",
                "password": "password",
                "tz_aware": True,
                "authentication_mechanism": "MONGODB-X509",
                "tls": True,
                "tlsAllowInvalidHostnames": False,
                "connectTimeoutMS": 3000,
                "serverSelectionTimeoutMS": 3000,
                "uuidRepresentation": "pythonLegacy",
            },
        )

    @mock.patch("st2common.models.db.mongoengine")
    @mock.patch("st2common.models.db.LOG")