Use ParameterizedTest to test payload

Author: cYKatherine

src/test/java/com/uid2/shared/secure/azurecc/MaaTokenSignatureValidatorTest.java

@@ -3,17 +3,18 @@
 import com.uid2.shared.secure.AttestationException;
 import com.uid2.shared.secure.TestClock;
 import org.junit.jupiter.api.Disabled;
-import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.ValueSource;
 
 import static com.uid2.shared.secure.TestUtils.loadFromJson;
 import static com.uid2.shared.secure.azurecc.MaaTokenUtils.validateAndParseToken;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
 public class MaaTokenSignatureValidatorTest {
-    @Test
-    public void testAciPayload() throws Exception {
+    @ParameterizedTest
+    @ValueSource(strings = {"/com.uid2.shared/test/secure/azurecc/jwt_payload_aci.json", "/com.uid2.shared/test/secure/azurecc/jwt_payload_aks.json"})
+    public void testPayload(String payloadPath) throws Exception {
         // expire at 1695313895
-        var payloadPath = "/com.uid2.shared/test/secure/azurecc/jwt_payload_aci.json";
         var payload = loadFromJson(payloadPath);
         var clock = new TestClock();
         clock.setCurrentTimeMs(1695313893000L);
@@ -31,27 +32,6 @@ public void testAciPayload() throws Exception {
         assertEquals(expectedPublicKey, tokenPayload.getRuntimeData().getPublicKey());
     }
 
-    @Test
-    public void testAksPayload() throws Exception {
-        // expire at 1695313895
-        var payloadPath = "/com.uid2.shared/test/secure/azurecc/jwt_payload_aks.json";
-        var payload = loadFromJson(payloadPath);
-        var clock = new TestClock();
-        clock.setCurrentTimeMs(1695313893000L);
-
-        var expectedCcePolicy = "fef932e0103f6132437e8a1223f32efc4bea63342f893b5124645224ef29ba73";
-        var expectedLocation = "East US";
-        var expectedPublicKey = "abc";
-
-        var tokenPayload = validateAndParseToken(payload, clock, "azure-cc-aks");
-        assertEquals(true, tokenPayload.isSevSnpVM());
-        assertEquals(true, tokenPayload.isUtilityVMCompliant());
-        assertEquals(false, tokenPayload.isVmDebuggable());
-        assertEquals(expectedCcePolicy, tokenPayload.getCcePolicyDigest());
-        assertEquals(expectedLocation, tokenPayload.getRuntimeData().getLocation());
-        assertEquals(expectedPublicKey, tokenPayload.getRuntimeData().getPublicKey());
-    }
-
     @Disabled
     // replace below Placeholder with real MAA token to run E2E verification.
     public void testE2E() throws AttestationException {