From adc3f0b82b370ef5e18326bb65ad392be57d101a Mon Sep 17 00:00:00 2001
From: Weston Steimel <weston.steimel@anchore.com>
Date: Fri, 17 Mar 2023 11:37:44 +0000
Subject: [PATCH] fix: fix profile load and add test (#59)

Signed-off-by: Weston Steimel <weston.steimel@anchore.com>
---
 src/yardstick/cli/config.py   |  2 +-
 tests/unit/cli/test_config.py | 22 ++++++++++++++++++++++
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/yardstick/cli/config.py b/src/yardstick/cli/config.py
index b00cea0..4c6875c 100644
--- a/src/yardstick/cli/config.py
+++ b/src/yardstick/cli/config.py
@@ -103,7 +103,7 @@ def clean_dict_keys(d):
 
 
 def yaml_decoder(data) -> Dict[Any, Any]:
-    return clean_dict_keys(yaml.load(data, yaml.CSafeLoader))
+    return clean_dict_keys(yaml.safe_load(data))
 
 
 def load(path: str = ".yardstick.yaml") -> Application:
diff --git a/tests/unit/cli/test_config.py b/tests/unit/cli/test_config.py
index 70878d3..776fc0c 100644
--- a/tests/unit/cli/test_config.py
+++ b/tests/unit/cli/test_config.py
@@ -38,6 +38,16 @@ def test_config(tmp_path):
     file = tmp_path / "config.yaml"
     file.write_text(subject)
 
+    profile_text = """
+test_profile:
+  something:
+    name: jim
+    config_path: .abc/xyx.conf
+    refresh: false
+"""
+    profile_file = tmp_path / ".yardstick.profiles.yaml"
+    profile_file.write_text(profile_text)
+
     cfg = config.load(str(file))
 
     assert cfg.result_sets["sboms"].matrix.images == [
@@ -47,3 +57,15 @@ def test_config(tmp_path):
         "docker.io/anchore/test_images:grype-quality-node-d89207b@sha256:f56164678054e5eb59ab838367373a49df723b324617b1ba6de775749d7f91d4",
         "docker.io/vulhub/cve-2017-1000353:latest@sha256:da2a59314b9ccfb428a313a7f163adcef77a74a393b8ebadeca8223b8cea9797",
     ]
+
+    assert cfg.profiles == config.Profiles(
+        {
+            "test_profile": {
+                "something": {
+                    "name": "jim",
+                    "config_path": ".abc/xyx.conf",
+                    "refresh": False,
+                },
+            },
+        }
+    )