saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-16 09:40:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Adding tests for when the exception is raised.

Browse source
This commit is contained in:
Gareth J. Greenaway 2022-11-01 14:34:30 -07:00
parent 7c214b6426
commit 48149bc7c9
No known key found for this signature in database
GPG key ID: 10B62F8A7CAD7A41
1 changed files with 52 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
tests/pytests/unit/modules/test_mongodb.py
View file

@ -323,6 +323,32 @@ def test_user_create():
assert ret
def test_user_create_exception():
"""
Test mongodb.user_create
"""
user_create_mock = MagicMock(side_effect=pymongo.errors.PyMongoError)
pymongo_database_mock = MagicMock(autospec=True, return_value=MockPyMongoDatabase())
mongodb_client_mock = MagicMock(autospec=True, return_value=MockMongoConnect())
config_option_mock = MagicMock(
autospec=True, side_effect=["user", "password", "localhost", "27017"]
)
patch_pymongo_database = patch("pymongo.database.Database", pymongo_database_mock)
patch_mongo_client = patch("pymongo.MongoClient", mongodb_client_mock)
patch_pymongo_command = patch.object(
MockPyMongoDatabase, "command", user_create_mock
)
patch_salt_dict = patch.dict(
mongodb.__salt__, {"config.option": config_option_mock}
)
with patch_mongo_client, patch_pymongo_database, patch_salt_dict, patch_pymongo_command:
ret = mongodb.user_create("test_user", "test_password")
assert not ret
def test_user_remove():
"""
Test mongodb.user_remove
@ -349,6 +375,32 @@ def test_user_remove():
assert ret
def test_user_remove_exception():
"""
Test mongodb.user_remove
"""
user_remove_mock = MagicMock(autospec=True, side_effect=pymongo.errors.PyMongoError)
pymongo_database_mock = MagicMock(autospec=True, return_value=MockPyMongoDatabase())
mongodb_client_mock = MagicMock(autospec=True, return_value=MockMongoConnect())
config_option_mock = MagicMock(
autospec=True, side_effect=["user", "password", "localhost", "27017"]
)
patch_pymongo_database = patch("pymongo.database.Database", pymongo_database_mock)
patch_mongo_client = patch("pymongo.MongoClient", mongodb_client_mock)
patch_pymongo_command = patch.object(
MockPyMongoDatabase, "command", user_remove_mock
)
patch_salt_dict = patch.dict(
mongodb.__salt__, {"config.option": config_option_mock}
)
with patch_mongo_client, patch_pymongo_database, patch_salt_dict, patch_pymongo_command:
ret = mongodb.user_remove("test_user")
assert not ret
def test_user_roles_exists():
"""
Test mongodb.user_roles_exist