def setUp(self):
options.db = 'test'
super(DBTestCase, self).setUp()
Connection.set_io_loop(self.io_loop)
self.db = Connection.get_connection()
self.sync_db = self.db.connection.sync_client()[self.db.name]
self.addCleanup(self.clean)
def setUp(self):
options.db = 'test'
super(DBTestCase, self).setUp()
Connection.set_io_loop(self.io_loop)
self.db = Connection.get_connection()
self.sync_db = self.db.connection.sync_client()[self.db.name]
self.addCleanup(self.clean)
def test_should_compose_uri_with_environ(self):
options.mongo_uri = 'mycrazy://{GLB_FEED_MONGO_USR}:{GLB_FEED_MONGO_PWD}@localhost'
os.environ['GLB_FEED_MONGO_USR'] = '******'
os.environ['GLB_FEED_MONGO_PWD'] = 'fakepass'
Connection.load_config()
assert Connection._config[
'mongo_uri'] == 'mycrazy://*****:*****@localhost'
def test_should_use_mongo_uri(self):
options.mongo_uri = 'mycrazy://uri'
Connection.load_config()
with mock.patch('motor.MotorClient') as MockMotorClient:
MockMotorClient.return_value.open_sync.return_value = 'my'
assert Connection.create_client() == 'my'
MockMotorClient.assert_called_with(host='mycrazy://uri')
def test_should_use_mongo_uri(self):
options.mongo_uri = 'mycrazy://uri'
Connection.load_config()
with mock.patch('motor.MotorClient') as MockMotorClient:
MockMotorClient.return_value.open_sync.return_value = 'my'
assert Connection.create_client() == 'my'
MockMotorClient.assert_called_with(host='mycrazy://uri')
def test_should_drop(self):
db = Connection.get_connection()
collection = db.test_collection
result = yield collection.insert({'name': 'test'})
result = yield collection.count()
assert result == 1
#TODO: Isso deveria funcionar assim: Connection.drop()
yield db.test_collection.drop()
collection = db.test_collection
result2 = yield collection.count()
assert result2 == 0
def test_should_drop(self):
db = Connection.get_connection()
collection = db.test_collection
result = yield collection.insert({'name': 'test'})
result = yield collection.count()
assert result == 1
#TODO: Isso deveria funcionar assim: Connection.drop()
yield db.test_collection.drop()
collection = db.test_collection
result2 = yield collection.count()
assert result2 == 0
def clean(self):
Connection.disconnect()
self._reset_config()
self._drop_test_collection()
Connection.disconnect()
def test_should_compose_uri_with_environ(self):
options.mongo_uri = 'mycrazy://{GLB_FEED_MONGO_USR}:{GLB_FEED_MONGO_PWD}@localhost'
os.environ['GLB_FEED_MONGO_USR'] = '******'
os.environ['GLB_FEED_MONGO_PWD'] = 'fakepass'
Connection.load_config()
assert Connection._config['mongo_uri'] == 'mycrazy://*****:*****@localhost'
def test_should_connect(self):
db = Connection.get_connection()
assert db.name == 'test'
def tearDown(self):
super(TestConnection, self).tearDown()
Connection.disconnect()
def clean(self):
Connection.drop()
Connection.disconnect()
def _drop_test_collection(self):
db = Connection.get_connection()
sync_db = db.connection.sync_client()[db.name] # pega client sync
sync_db.test_collection.drop()
def _reset_config(self):
options.db = 'test'
options.mongo_uri = 'mongodb://localhost'
Connection.load_config()
def clean(self):
Connection.disconnect()
self._reset_config()
self._drop_test_collection()
Connection.disconnect()
def test_should_connect(self):
db = Connection.get_connection()
assert db.name == 'test'
def _reset_config(self):
options.db = 'test'
options.mongo_uri = 'mongodb://localhost'
Connection.load_config()
def clean(self):
Connection.drop()
Connection.disconnect()
def _drop_test_collection(self):
db = Connection.get_connection()
sync_db = db.connection.sync_client()[db.name] # pega client sync
sync_db.test_collection.drop()
def tearDown(self):
super(TestConnection, self).tearDown()
Connection.disconnect()
def setUp(self):
super(TestConnection, self).setUp()
self._reset_config()
Connection.set_io_loop(self.io_loop)
self.addCleanup(self.clean)
def setUp(self):
super(TestConnection, self).setUp()
self._reset_config()
Connection.set_io_loop(self.io_loop)
self.addCleanup(self.clean)