Fix check

Author: DomAyre

src/confcom/azext_confcom/tests/latest/test_confcom_acifragmentgen.py

@@ -145,10 +145,10 @@ def test_acifragmentgen_fragment_upload_fragment(docker_image, cert_chain):
         check=True,
     ).stdout)
 
-    if oras_result.get("artifact_type") == "application/x-ms-ccepolicy-frag":
-        fragment_ref = oras_result["reference"]
-    elif "referrers" in oras_result:
+    if "referrers" in oras_result:
         fragment_ref = oras_result["referrers"][0]["reference"]
+    elif oras_result.get("manifests")[0].get("artifact_type") == "application/x-ms-ccepolicy-frag":
+        fragment_ref = oras_result["reference"]
     else:
         raise AssertionError(f"{oras_result=}")
 
@@ -235,10 +235,10 @@ def test_acifragmentgen_fragment_attach(docker_image, cert_chain, capsysbinary):
         check=True,
     ).stdout)
 
-    if oras_result.get("artifact_type") == "application/x-ms-ccepolicy-frag":
-        fragment_ref = oras_result["reference"]
-    elif "referrers" in oras_result:
+    if "referrers" in oras_result:
         fragment_ref = oras_result["referrers"][0]["reference"]
+    elif oras_result["manifests"][0].get("artifact_type") == "application/x-ms-ccepolicy-frag":
+        fragment_ref = oras_result["reference"]
     else:
         raise AssertionError(f"{oras_result=}")