Skip to content

Commit ffeb876

Browse files
committed
Replace string with MaaTokenPayload contant
1 parent 35851d1 commit ffeb876

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

src/test/java/com/uid2/shared/secure/AzureCCCoreAttestationServiceTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,8 @@ private static void attest(ICoreAttestationService provider, Handler<AsyncResult
147147

148148
static Stream<Arguments> argumentProvider() {
149149
return Stream.of(
150-
Arguments.of("azure-cc"),
151-
Arguments.of("azure-cc-aks")
150+
Arguments.of(MaaTokenPayload.AZURE_CC_AKS_PROTOCOL),
151+
Arguments.of(MaaTokenPayload.AZURE_CC_AKS_PROTOCOL)
152152
);
153153
}
154154
}

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ public void testE2E() throws AttestationException {
4646

4747
static Stream<Arguments> argumentProvider() {
4848
return Stream.of(
49-
Arguments.of("/com.uid2.shared/test/secure/azurecc/jwt_payload_aci.json", "azure-cc"),
50-
Arguments.of("/com.uid2.shared/test/secure/azurecc/jwt_payload_aks.json", "azure-cc-aks")
49+
Arguments.of("/com.uid2.shared/test/secure/azurecc/jwt_payload_aci.json", MaaTokenPayload.AZURE_CC_ACI_PROTOCOL),
50+
Arguments.of("/com.uid2.shared/test/secure/azurecc/jwt_payload_aks.json", MaaTokenPayload.AZURE_CC_AKS_PROTOCOL)
5151
);
5252
}
5353
}

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ public void testValidationSuccess_AksWithAzureSignedKataccUvm() throws Attestati
145145
var aksPayload = generateBasicPayload()
146146
.toBuilder()
147147
.complianceStatus("azure-signed-katacc-uvm")
148-
.azureProtocol("azure-cc-aks")
148+
.azureProtocol(MaaTokenPayload.AZURE_CC_AKS_PROTOCOL)
149149
.build();
150150
var enclaveId = validator.validate(aksPayload, PUBLIC_KEY);
151151
assertEquals(CCE_POLICY_DIGEST, enclaveId);
@@ -157,7 +157,7 @@ public void testValidationFailure_AksWithOtherUvm() {
157157
var aksPayload = generateBasicPayload()
158158
.toBuilder()
159159
.complianceStatus("fake-compliance")
160-
.azureProtocol("azure-cc-aks")
160+
.azureProtocol(MaaTokenPayload.AZURE_CC_AKS_PROTOCOL)
161161
.build();
162162
Throwable t = assertThrows(AttestationException.class, ()-> validator.validate(aksPayload, PUBLIC_KEY));
163163
assertEquals("Not run in Azure Compliance Utility VM", t.getMessage());

0 commit comments

Comments
 (0)