test print_profiles function

Author: cadenmyers13

tests/test_profilesmanager.py

@@ -0,0 +1,47 @@
+import pytest
+import yaml
+
+from diffpy.cmi.profilesmanager import ProfilesManager
+
+install_params = [
+    (  # input: profiles to install
+        # expected: print_profile output showing profileA
+        # installed but not profileB
+        ("profileA",),
+        """Installed Profiles:
+-------------------
+  profileA
+
+Available Profiles:
+-------------------
+  profileB
+""",
+    ),
+]
+
+
+@pytest.mark.parametrize("profiles_to_install,expected", install_params)
+def test_print_profiles(
+    profiles_to_install, expected, example_cases, capsys, mocker
+):
+    case5dir = example_cases / "case5"
+    req_dir = case5dir / "requirements" / "profiles"
+
+    installed_reqs = []
+    for profile in profiles_to_install:
+        req_file = req_dir / f"{profile}.yml"
+        content = yaml.safe_load(req_file.read_text())
+        for pack in content.get("packs", []):
+            installed_reqs.append(pack)
+
+    def fake_check_profile(identifier):
+        return identifier == "profileA"
+
+    mocker.patch.object(
+        ProfilesManager, "check_profile", side_effect=fake_check_profile
+    )
+    pfm = ProfilesManager(root_path=case5dir)
+    pfm.print_profiles()
+    captured = capsys.readouterr()
+    actual = captured.out
+    assert actual.strip() == expected.strip()