diff --git a/test/test_black_white.py b/test/test_black_white.py index be50007..1ede818 100644 --- a/test/test_black_white.py +++ b/test/test_black_white.py @@ -18,12 +18,12 @@ from src.black_white import setup_black_white_lists def test_setup_black_white_lists(): # Simple - blacklist_library = "library1, library2" - whitelist_library = "library1, library2" - blacklist_library_type = "library_type1, library_type2" - whitelist_library_type = "library_type1, library_type2" - blacklist_users = "user1, user2" - whitelist_users = "user1, user2" + blacklist_library = ["library1", "library2"] + whitelist_library = ["library1", "library2"] + blacklist_library_type = ["library_type1", "library_type2"] + whitelist_library_type = ["library_type1", "library_type2"] + blacklist_users = ["user1", "user2"] + whitelist_users = ["user1", "user2"] ( results_blacklist_library, @@ -48,6 +48,14 @@ def test_setup_black_white_lists(): assert return_blacklist_users == ["user1", "user2"] assert return_whitelist_users == ["user1", "user2"] +def test_library_mapping_black_white_list(): + blacklist_library = ["library1", "library2"] + whitelist_library = ["library1", "library2"] + blacklist_library_type = ["library_type1", "library_type2"] + whitelist_library_type = ["library_type1", "library_type2"] + blacklist_users = ["user1", "user2"] + whitelist_users = ["user1", "user2"] + # Library Mapping and user mapping library_mapping = {"library1": "library3"} user_mapping = {"user1": "user3"}