def setUp(self): super(BaseMongoTest, self).setUp() self._DB_NAME = "test_database" Connection.setup(self._DB_NAME) Connection.client.drop_database(self._DB_NAME) # Convenience attribute for integration tests self.db = Connection.client[self._DB_NAME]
# -*- coding: utf-8 -*- from tavi import Connection import logging Connection.setup("test_database") class LogCapture(logging.Handler): MSG_TYPES = ["debug", "info", "warning", "error", "critical"] def __init__(self, *args, **kwargs): self.reset() logging.Handler.__init__(self, *args, **kwargs) def __enter__(self): logging.getLogger().addHandler(self) return self def __exit__(self, type_, value, traceback): logging.getLogger().removeHandler(self) self.close() def emit(self, record): self.messages[record.levelname.lower()].append(record.getMessage()) def reset(self): self.messages = {t: [] for t in self.MSG_TYPES}
def test_setup_with_uri(self): Connection.setup("test_database", host="mongodb://localhost:27017") self.assertEqual(self.db, Connection.database)
def test_has_a_client_attribute(self): Connection.setup("test_database", host="mongodb://localhost:27017") self.assertEqual(self.client, Connection.client)
def test_setup_with_host_and_port(self): Connection.setup("test_database", host="localhost", port=27017) self.assertEqual(self.db, Connection.database)