如何在Python中对使用mongoengine连接的函数进行测试？

huangapple
go评论113阅读模式
英文:

How to realize tests on functions which are using mongoengine connection in Python?

问题

以下是您要翻译的内容：

  1. 我有一个使用Mongo数据库的FastAPI
  2. 我尝试为一些需要Mongo连接的函数编写测试
  3. 这是我的用户集合和其字段:
  5. from mongoengine import Document, EmailField, StringField, EnumField 
  6. class User(Document):  
  7.     meta = {"collection": "users"}
  8.     email = EmailField(unique=True)
  9.     hashed_password = StringField()
  10.     role = EnumField(UserRole)
  12.     def username(self):
  13.         return self.email.split('@')[0]
  15.     def __str__(self):
  16.         return f'{self.email} ({self.role})'
  18. 这是我的函数:
  20. class UserService:
  22.     @staticmethod
  23.     def existing_user(email: str) -> Optional[User]:
  26.         try:
  27.              user = User.objects(email=email).first()
  28.         except ServerSelectionTimeoutError as e:
  29.             raise e
  31.         return user
  33. 请问如何最好地测试这个函数（使用pytest）？我想要模拟我的MongoDB，但我无法使用mongomock或直接使用monkeypatch来实现
  35. 非常感谢
  37. 我的测试文件
  39.  from domain.entities.user import UserRole, User
  40.  from mongomock import MongoClient
  41.  class TestUserServiceMethods:
  43.     email="john@doe.fr"
  45.     # 用户示例
  46.     user = User(email=email,
  47.                   role=UserRole.admin,
  48.                   hashed_password="hashed_pwd")
  50.     add_user = {
  51.       "email": email,
  52.       "role": UserRole.admin,
  53.       "hashed_password": "hashed_pwd"
  54.     }
  56.     
  58.     user_service = UserService()
  60.     client = MongoClient()
  62.     mock_db = client['ml-db']
  63.     user_collection = mock_db['users']
  64.     user_collection.insert_one(add_user)
  68.     def test_existing_user(self, monkeypatch):
  69.         with monkeypatch.context() as m:
  70.             m.setattr('pymongo.MongoClient', lambda *args, **kwargs: self.client)
  72.             result = self.user_service.existing_user(self.email)
  74.              assert result == self.user
  76. 似乎模拟没有成功，因为我得到了以下错误：
  78. mongoengine.connection.ConnectionFailure: 未定义默认连接
  80. 我应该直接使用FastAPI的test_client吗？
英文:

I have Fast API with a Mongo Database
I try to realize my tests for some functions which need mongo connection
Here my collection users and its fields:

  1. from mongoengine import Document, EmailField, StringField, EnumField 
  2. class User(Document):  
  3.     meta = {"collection": "users"}
  4.     email = EmailField(unique=True)
  5.     hashed_password = StringField()
  6.     role = EnumField(UserRole)
  8.     def username(self):
  9.         return self.email.split('@')[0]
  11.     def __str__(self):
  12.         return f'{self.email} ({self.role})

Here my function

  1. class UserService:
  3.     @staticmethod
  4.     def existing_user(email: str) -> Optional[User]:
  7.         try:
  8.              user = User.objects(email=email).first()
  9.         except ServerSelectionTimeoutError as e:
  10.             raise e
  12.         return user

What is the best way to test this function please (using pytest)? I would like to mock my MongoDB but I didn't manage with mongomock or directly with monkeypatch

Thank you very much

My test file

  1.  from domain.entities.user import UserRole, User
  2.  from mongomock import MongoClient
  3.  class TestUserServiceMethods:
  5.     email="john@doe.fr"
  7.     # User example
  8.     user = User(email=email,
  9.                   role=UserRole.admin,
  10.                   hashed_password="hashed_pwd")
  12.     add_user = {
  13.       "email": email,
  14.       "role": UserRole.admin,
  15.       "hashed_password": "hashed_pwd"
  16.     }
  18.     
  20.     user_service = UserService()
  22.     client = MongoClient()
  24.     mock_db = client['ml-db']
  25.     user_collection = mock_db['users']
  26.     user_collection.insert_one(add_user)
  30.     def test_existing_user(self, monkeypatch):
  31.         with monkeypatch.context() as m:
  32.             m.setattr('pymongo.MongoClient', lambda *args, **kwargs: self.client)
  34.             result = self.user_service.existing_user(self.email)
  36.              assert result == self.user

It seems the mock didn't sucess because I got as error :

  1. mongoengine.connection.ConnectionFailure: You have not defined a default connection

Should I use directly Fast app with test_client?

答案1

得分: 0

你需要在执行任何操作之前，从mogoengine运行connect方法，使用db_name（默认别名为"default"）和connection_string。

例如：

  1. from mongoengine import connect
  3. connect("db_name", host="connection_string", mongo_client_class=client)

代码中的client将来自monfoengine，而在测试中将来自mongomock。

英文:

You need to run connect method from mogoengine with db_name (default alias is "default") and connection_string before any action you do.

for example :

  1. from mongoengine import connect
  3. connect("db_name", host="connection_string", mongo_client_class=client)

the client in the code would be from monfoengine and in tests it would be from mongomock

答案2

得分: 0

Here is the translated code:

  1. **更新** 正如Ozs所说，最好的方法是创建一个虚拟数据库，在最后将其删除。下面是解决我的问题的示例：
  3. ```python
  4. from unittest import TestCase
  5. from mongoengine import connect
  6. import UserService, password_context 
  7. import UserRole, User
  9. class TestUserServiceMethods(TestCase):
  11.     @classmethod
  12.     def setUpClass(cls):
  13.         cls.connection = connect('mongoenginetest', host='mongomock://localhost/ml-db')
  15.     def setUp(cls):
  16.         cls.data = {
  17.             "id": "6499594c58ebb74dd4985e59",
  18.             "email": "mail@test.fr",
  19.             "pwd": "test",
  20.             "hashed_pwd": password_context.hash("test"),
  21.             "role": UserRole.admin
  22.         }
  24.         cls.doc = {
  25.             "email": cls.data['email'],
  26.             "role": cls.data['role'],
  27.             "hashed_password": cls.data['hashed_pwd']
  28.         }
  30.         cls.user = User(
  31.             email=cls.data['email'],
  32.             role=cls.data['role'],
  33.             hashed_password=cls.data['hashed_pwd'])
  35.         cls.user_service = UserService()
  37.         cls.user.save()
  39.     @classmethod
  40.     def tearDownClass(cls):
  41.         cls.connection.drop_database('mongoenginetest')
  43.     def test_existing_user(self):
  44.         user = self.user_service.existing_user(self.data['email'])
  45.         self.assertIsNotNone(user)
  47.         user_ko = self.user_service.existing_user("error@mail.fr")
  48.         self.assertIsNone(user_ko)

Please note that I've translated the code, but I haven't included the comments as they appear to be in English and don't require translation.

英文:

UPDATE As Ozs said, the best way is to create a Fake database and to drop it at the end. Here an example which resolves my problem

  1. from unittest import TestCase
  3. from mongoengine import connect
  5. import UserService, password_context 
  6. import UserRole, User
  9. class TestUserServiceMethods(TestCase):
  11.     @classmethod
  12.     def setUpClass(cls):
  13.         cls.connection = connect('mongoenginetest', host='mongomock://localhost/ml-db')
  15.     def setUp(cls):
  16.         cls.data = {
  17.             "id": "6499594c58ebb74dd4985e59",
  18.             "email": "mail@test.fr",
  19.             "pwd": "test",
  20.             "hashed_pwd": password_context.hash("test"),
  21.             "role": UserRole.admin
  22.         }
  24.         cls.doc = {
  25.             "email": cls.data['email'],
  26.             "role": cls.data['role'],
  27.             "hashed_password": cls.data['hashed_pwd']
  28.         }
  29.       
  31.         cls.user = User(
  32.             email=cls.data['email'],
  33.             role=cls.data['role'],
  34.             hashed_password=cls.data['hashed_pwd'])
  36.         cls.user_service = UserService()
  38.         cls.user.save()
  40.     @classmethod
  41.     def tearDownClass(cls):
  42.         cls.connection.drop_database('mongoenginetest')
  44.     def test_existing_user(self):
  45.         user = self.user_service.existing_user(self.data['email'])
  46.         self.assertIsNotNone(user)
  48.         user_ko = self.user_service.existing_user("error@mail.fr")
  49.         self.assertIsNone(user_ko)

huangapple
  • 本文由 发表于 2023年6月27日 20:24:02
  • 转载请务必保留本文链接：https://go.coder-hub.com/76564826.html
  • mongodb
  • mongoengine
  • mongomock
  • pytest
  • python
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定