/*
 * Copyright 2024 Red Hat, Inc. and/or its affiliates
 * and other contributors as indicated by the @author tags.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.keycloak.sdjwt;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import org.junit.Test;

import static org.junit.Assert.assertEquals;

/**
 * @author <a href="mailto:francis.pouatcha@adorsys.com">Francis Pouatcha</a>
 */
public class ArrayElementDisclosureTest {

        @Test
        public void testSdJwtWithUndiclosedArrayElements6_1() {
                ObjectNode claimSet = TestUtils.readClaimSet(getClass(), "sdjwt/s6.1-holder-claims.json");

                DisclosureSpec disclosureSpec = DisclosureSpec.builder()
                                .withUndisclosedClaim("email", "JnwGqRFZjMprsoZobherdQ")
                                .withUndisclosedClaim("phone_number", "ffZ03jm_zeHyG4-yoNt6vg")
                                .withUndisclosedClaim("address", "INhOGJnu82BAtsOwiCJc_A")
                                .withUndisclosedClaim("birthdate", "d0l3jsh5sBzj2oEhZxrJGw")
                                .withUndisclosedArrayElt("nationalities", 1, "nPuoQnkRFq3BIeAm7AnXFA")
                                .build();

                IssuerSignedJWT issuerSignedJWT = IssuerSignedJWT.builder()
                        .withClaims(claimSet, disclosureSpec)
                        .build();
                SdJwt sdJwt = SdJwt.builder()
                                .withIssuerSignedJwt(issuerSignedJWT)
                                .withUseDefaultDecoys(false)
                                .build();

                IssuerSignedJWT jwt = sdJwt.getIssuerSignedJWT();

                JsonNode expected = TestUtils.readClaimSet(getClass(),
                                "sdjwt/s6.1-issuer-payload-udisclosed-array-ellement.json");
                assertEquals(expected, jwt.getPayload());
        }

        @Test
        public void testSdJwtWithUndiclosedAndDecoyArrayElements6_1() {
                ObjectNode claimSet = TestUtils.readClaimSet(getClass(), "sdjwt/s6.1-holder-claims.json");

                DisclosureSpec disclosureSpec = DisclosureSpec.builder()
                                .withUndisclosedClaim("email", "JnwGqRFZjMprsoZobherdQ")
                                .withUndisclosedClaim("phone_number", "ffZ03jm_zeHyG4-yoNt6vg")
                                .withUndisclosedClaim("address", "INhOGJnu82BAtsOwiCJc_A")
                                .withUndisclosedClaim("birthdate", "d0l3jsh5sBzj2oEhZxrJGw")
                                .withUndisclosedArrayElt("nationalities", 0, "Qg_O64zqAxe412a108iroA")
                                .withUndisclosedArrayElt("nationalities", 1, "nPuoQnkRFq3BIeAm7AnXFA")
                                .withDecoyArrayElt("nationalities", 1, "5bPs1IquZNa0hkaFzzzZNw")
                                .build();

            IssuerSignedJWT issuerSignedJWT = IssuerSignedJWT.builder()
                    .withClaims(claimSet, disclosureSpec)
                    .build();
                SdJwt sdJwt = SdJwt.builder()
                                .withIssuerSignedJwt(issuerSignedJWT)
                                .withUseDefaultDecoys(false)
                                .build();
                IssuerSignedJWT jwt = sdJwt.getIssuerSignedJWT();

                JsonNode expected = TestUtils.readClaimSet(getClass(),
                                "sdjwt/s6.1-issuer-payload-decoy-array-ellement.json");
                assertEquals(expected, jwt.getPayload());
        }
}
