def _insert_document(self, document):
message_insert = message.insert('mongotor_test.cursor_test', [document],
True, True, {})
node = Database().get_node(ReadPreference.PRIMARY)
node.connection(self.stop)
connection = self.wait()
connection.send_message(message_insert, with_last_error=True, callback=self.stop)
self.wait()
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()
