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, )