def test_mongo_connection(self):
from pyramid_mongokit import mongo_connection
request = mock.Mock()
self.assertEqual(
request.registry.getUtility.return_value,
mongo_connection(request)
)
def setUp(self):
self.config = testing.setUp()
self.config.include("pyramid_rest")
self.config.include("velo")
self.request = testing.DummyRequest()
self.request.context = testing.DummyResource()
self.request.rest_resource_url = partial(rest_resource_url, self.request)
self.request.rest_resource_path = partial(rest_resource_path, self.request)
self.request.POST = MultiDict()
self.request.mongo_connection = mongo_connection(self.request)
self.request.mongo_db = mongo_db(self.request)
def test_mongo_connection(self, m_get_mongo_connection, m_end_request):
from pyramid_mongokit import mongo_connection
request = mock.Mock()
self.assertEqual(
mongo_connection(request),
m_get_mongo_connection.return_value,
)
# 1. Mongo client was retrived from request's registry
m_get_mongo_connection.assert_called_once_with(request.registry)
# 2. Method start_request of mongo client was called
m_get_mongo_connection.return_value.start_request.assert_called_once_with()
# 3. Method end_request of mongo client was registered as a callback for
# end of request processing
request.add_finished_callback.assert_called_once_with(
m_end_request,
)
