class MongoAdapterTestCase(TestCase):
def setUp(self):
"""
Instantiate the adapter.
"""
from pymongo.errors import ServerSelectionTimeoutError
from pymongo import MongoClient
database_name = "test_db"
# Skip these tests if a mongo client is not running
try:
client = MongoClient(
serverSelectionTimeoutMS=0.1
)
client.server_info()
self.adapter = MongoDatabaseAdapter(database=database_name)
except ServerSelectionTimeoutError:
raise SkipTest("Unable to connect to mongo database.")
def tearDown(self):
"""
Remove the test database.
"""
self.adapter.drop()
class MongoAdapterTestCase(TestCase):
def setUp(self):
"""
Instantiate the adapter.
"""
from pymongo.errors import ServerSelectionTimeoutError
from pymongo import MongoClient
database_name = "test_db"
# Skip these tests if a mongo client is not running.
try:
client = MongoClient(serverSelectionTimeoutMS=0.2)
client.server_info()
self.adapter = MongoDatabaseAdapter(database=database_name)
except ServerSelectionTimeoutError:
raise SkipTest("Unable to connect to mongo database.")
def tearDown(self):
"""
Remove the test database.
"""
self.adapter.drop()
class BaseMongoDatabaseAdapterTestCase(TestCase):
def setUp(self):
"""
Instantiate the adapter.
"""
database_name = "test_db"
self.adapter = MongoDatabaseAdapter(database=database_name)
def tearDown(self):
"""
Remove the test database.
"""
self.adapter.drop()
def setUp(self):
"""
Instantiate the adapter.
"""
database_name = "test_db"
self.adapter = MongoDatabaseAdapter(database=database_name)
def setUp(self):
"""
Instantiate the adapter.
"""
from pymongo.errors import ServerSelectionTimeoutError
from pymongo import MongoClient
database_name = "test_db"
# Skip these tests if a mongo client is not running.
try:
client = MongoClient(serverSelectionTimeoutMS=1)
client.server_info()
self.adapter = MongoDatabaseAdapter(None, database=database_name)
except ServerSelectionTimeoutError:
raise SkipTest("Unable to connect to mongo database.")
def setUp(self):
super(MongoStorageIntegrationTests, self).setUp()
from pymongo.errors import ServerSelectionTimeoutError
from pymongo import MongoClient
# Skip these tests if a mongo client is not running
try:
client = MongoClient(serverSelectionTimeoutMS=0.1)
client.server_info()
except ServerSelectionTimeoutError:
raise SkipTest("Unable to connect to mongo database.")
self.chatbot.storage_adapters = []
self.chatbot.storage = MongoDatabaseAdapter()
