def tearDown(self):
super(SignalTestCase, self).tearDown()
class CollectionTest(Collection):
__collection__ = "collection_test"
CollectionTest.objects.truncate(callback=self.stop)
self.wait()
Database.disconnect()
def tearDown(self):
super(SignalTestCase, self).tearDown()
class CollectionTest(Collection):
__collection__ = "collection_test"
CollectionTest.objects.truncate(callback=self.stop)
self.wait()
Database.disconnect()
def tearDown(self):
super(CursorTestCase, self).tearDown()
# delete all documents
message_delete = message.delete('mongotor_test.cursor_test',
{}, True, {})
Database().send_message(message_delete, callback=self.stop)
self.wait()
Database.disconnect()
def tearDown(self):
super(CursorTestCase, self).tearDown()
# delete all documents
message_delete = message.delete('mongotor_test.cursor_test', {}, True,
{})
Database().send_message(message_delete, callback=self.stop)
self.wait()
Database.disconnect()
def test_raises_erro_when_use_collection_with_not_initialized_database(
self):
"""[CollectionTestCase] - Raises DatabaseError when use collection with a not initialized database"""
class CollectionTest(Collection):
__collection__ = 'collection_test'
Database.disconnect()
CollectionTest().save.when.called_with(callback=None) \
.throw(DatabaseError, 'you must be initialize database before perform this action')
Database.init(["localhost:27027", "localhost:27028"], dbname='test')
def test_raises_erro_when_use_collection_with_not_initialized_database(self):
"""[CollectionTestCase] - Raises DatabaseError when use collection with a not initialized database"""
class CollectionTest(Collection):
__collection__ = 'collection_test'
Database.disconnect()
CollectionTest().save.when.called_with(callback=None) \
.throw(DatabaseError, 'you must be initialize database before perform this action')
Database.init(["localhost:27027", "localhost:27028"], dbname='test')
def tearDown(self):
super(CursorTestCase, self).tearDown()
# delete all documents
message_delete = message.delete('mongotor_test.cursor_test',
{}, True, {})
node = Database().get_node(ReadPreference.PRIMARY)
node.connection(self.stop)
connection = self.wait()
connection.send_message(message_delete, with_last_error=True, callback=self.stop)
self.wait()
Database.disconnect()
def test_check_connections_when_use_cursors(self):
"""[ConnectionPoolTestCase] - check connections when use cursors"""
db = Database.init('localhost:27027', dbname='test', maxconnections=10, maxusage=29)
try:
for i in range(2):
db.cards.insert({'_id': ObjectId(), 'range': i}, callback=self.stop)
self.wait()
db._nodes[0].pool._connections.should.be.equal(0)
db.cards.find({}, callback=self.stop)
self.wait()
db._nodes[0].pool._connections.should.be.equal(0)
finally:
Database.disconnect()
def tearDown(self):
super(CursorTestCase, self).tearDown()
# delete all documents
message_delete = message.delete('mongotor_test.cursor_test', {}, True,
{})
Database().get_node(ReadPreference.PRIMARY, callback=self.stop)
node = self.wait()
node.connection(self.stop)
connection = self.wait()
connection.send_message(message_delete,
with_last_error=True,
callback=self.stop)
self.wait()
Database.disconnect()
def test_check_connections_when_use_cursors(self):
"""[ConnectionPoolTestCase] - check connections when use cursors"""
db = Database.init('localhost:27027',
dbname='test',
maxconnections=10,
maxusage=29)
try:
for i in range(2):
db.cards.insert({
'_id': ObjectId(),
'range': i
},
callback=self.stop)
self.wait()
db._nodes[0].pool._connections.should.be.equal(0)
db.cards.find({}, callback=self.stop)
self.wait()
db._nodes[0].pool._connections.should.be.equal(0)
finally:
Database.disconnect()
def tearDown(self):
super(ManagerTestCase, self).tearDown()
CollectionTest.objects.truncate(callback=self.stop)
self.wait()
Database.disconnect()
def test_disconnect_database(self):
"""[DatabaseTestCase] - Disconnect the database"""
Database.init(["localhost:27027"], dbname='test')
Database.disconnect()
Database._instance.should.be.none
def tearDown(self):
super(CollectionTestCase, self).tearDown()
Database.disconnect()
def tearDown(self):
super(ManagerTestCase, self).tearDown()
CollectionTest.objects.truncate(callback=self.stop)
self.wait()
Database.disconnect()
def tearDown(self):
super(ReplicaSetTestCase, self).tearDown()
Database.disconnect()
def tearDown(self):
super(ReplicaSetTestCase, self).tearDown()
Database.disconnect()
def tearDown(self):
super(CollectionTestCase, self).tearDown()
Database.disconnect()
def test_disconnect_database(self):
"""[DatabaseTestCase] - Disconnect the database"""
Database.connect(["localhost:27027"], dbname="test")
Database.disconnect()
Database._instance.should.be.none
