/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2021 Red Hat, Inc., and individual 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.wildfly.extension.elytron;


import static org.junit.Assert.assertEquals;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.jboss.as.controller.descriptions.ModelDescriptionConstants;
import org.jboss.as.subsystem.test.AbstractSubsystemBaseTest;
import org.jboss.as.subsystem.test.AdditionalInitialization;
import org.jboss.as.subsystem.test.KernelServices;
import org.jboss.as.subsystem.test.KernelServicesBuilder;
import org.jboss.dmr.ModelNode;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

/**
 * Verifies that attributes that allow expressions resolve them correctly.
 *
 * @author <a href="mailto:szaldana@redhat.com">Sonia Zaldana</a>
 */
public class ResolveExpressionAttributesTestCase extends AbstractSubsystemBaseTest {

    private ModelNode serverModel;

    public ResolveExpressionAttributesTestCase() {
        super(ElytronExtension.SUBSYSTEM_NAME, new ElytronExtension());
    }

    @Before
    public void init() throws Exception {
        KernelServicesBuilder builder = createKernelServicesBuilder(AdditionalInitialization.MANAGEMENT)
                .setSubsystemXml(getSubsystemXml());
        KernelServices kernelServices = builder.build();
        Assert.assertTrue("Subsystem boot failed!", kernelServices.isSuccessfulBoot());
        ModelNode rootModel = kernelServices.readWholeModel();
        serverModel =  rootModel.require(ModelDescriptionConstants.SUBSYSTEM).require(ElytronExtension.SUBSYSTEM_NAME);
    }

    @Override
    protected String getSubsystemXml() throws IOException {
        return readResource("elytron-expressions.xml");
    }

    @Test
    public void testExpressionAttributesResolved() {
        testAggregateRealm();
        testCertificateAuthorityAccount();
        testCertificateAuthority();
        testCredentialStore();
        testCustomComponent();
        testElytronDefinition();
        testEvidenceDecoder();
        testFailoverRealm();
        testFileSystemRealm();
        testFilteringKeyStoreDefinition();
        testIdentityRealm();
        testJaspiConfiguration();
        testJdbcRealm();
        testKerberosSecurityFactory();
        testKeyStore();
        testLdapKeyStore();
        testLdapRealm();
        testPermissionMappers();
        testPrincipalDecoders();
        testPrincipalTransformers();
        testPropertiesRealm();
        testProvider();
        testRealmMappers();
        testRoleDecoders();
        testRoleMappers();
        testSaslServer();
        testSSLComponents();
        testTokenRealm();
    }

    private void testAggregateRealm() {
        ModelNode aggRealm = serverModel.get(ElytronDescriptionConstants.AGGREGATE_REALM).get("AggregateOne");
        assertEquals("NameRewriterXY", getValue(aggRealm, ElytronDescriptionConstants.PRINCIPAL_TRANSFORMER));
    }

    private void testCertificateAuthorityAccount() {
        ModelNode caAccount = serverModel.get(ElytronDescriptionConstants.CERTIFICATE_AUTHORITY_ACCOUNT).get("MyCA");
        assertEquals(Arrays.asList("https://www.test.com"), getValue(caAccount, ElytronDescriptionConstants.CONTACT_URLS, true));
        assertEquals("LetsEncrypt", getValue(caAccount, ElytronDescriptionConstants.CERTIFICATE_AUTHORITY));
        assertEquals("server", getValue(caAccount, ElytronDescriptionConstants.ALIAS));
    }

    private void testCertificateAuthority() {
        ModelNode ca = serverModel.get(ElytronDescriptionConstants.CERTIFICATE_AUTHORITY).get("testCA");
        assertEquals("https://www.test.com", getValue(ca, ElytronDescriptionConstants.STAGING_URL));
        assertEquals("https://www.test.com", getValue(ca, ElytronDescriptionConstants.URL));
    }

    private void testCredentialStore() {
        // Credential Stores
        ModelNode cs = serverModel.get(ElytronDescriptionConstants.CREDENTIAL_STORE).get("test1");
        assertEquals("test1.store", getValue(cs, ElytronDescriptionConstants.LOCATION));
        assertEquals("JCEKS", getValue(cs, ElytronDescriptionConstants.TYPE));
        assertEquals("provider", getValue(cs, ElytronDescriptionConstants.PROVIDER_NAME));
        cs = cs.get(ElytronDescriptionConstants.IMPLEMENTATION_PROPERTIES);
        assertEquals("JCEKS", getValue(cs, "keyStoreType"));
        cs = serverModel.get(ElytronDescriptionConstants.CREDENTIAL_STORE).get("test4");
        assertEquals("test1.store", getValue(cs, ElytronDescriptionConstants.PATH));

        // Secret Credential Store
        cs = serverModel.get(ElytronDescriptionConstants.SECRET_KEY_CREDENTIAL_STORE).get("test3");
        assertEquals("false", getValue(cs, ElytronDescriptionConstants.CREATE));
        assertEquals("false", getValue(cs, ElytronDescriptionConstants.POPULATE));
        assertEquals("192", getValue(cs, ElytronDescriptionConstants.KEY_SIZE));
        assertEquals("test3", getValue(cs, ElytronDescriptionConstants.DEFAULT_ALIAS));
    }

    private void testCustomComponent() {
        // Using custom permission mapper as example
        ModelNode mapper = serverModel.get(ElytronDescriptionConstants.CUSTOM_PERMISSION_MAPPER).get("MyPermissionMapper");
        assertEquals("value", getValue(mapper.get(ElytronDescriptionConstants.CONFIGURATION), "test"));
    }

    private void testElytronDefinition() {
        assertEquals(Arrays.asList("test"), getValue(serverModel, ElytronDescriptionConstants.DISALLOWED_PROVIDERS, true));
        assertEquals("false", getValue(serverModel, ElytronDescriptionConstants.REGISTER_JASPI_FACTORY));

    }

    private void testEvidenceDecoder() {
        ModelNode decoder = serverModel.get(ElytronDescriptionConstants.X509_SUBJECT_ALT_NAME_EVIDENCE_DECODER).get("rfc822Decoder");
        assertEquals("rfc822Name", getValue(decoder, ElytronDescriptionConstants.ALT_NAME_TYPE));
        assertEquals("1", getValue(decoder, ElytronDescriptionConstants.SEGMENT));
    }

    private void testFailoverRealm() {
        ModelNode failoverRealm = serverModel.get(ElytronDescriptionConstants.FAILOVER_REALM).get("FailoverRealm");
        assertEquals("true", getValue(failoverRealm, ElytronDescriptionConstants.EMIT_EVENTS));
    }

    private void testFileSystemRealm()  {
        ModelNode fileRealm = serverModel.get(ElytronDescriptionConstants.FILESYSTEM_REALM).get("FileRealm");
        assertEquals("2", getValue(fileRealm, ElytronDescriptionConstants.LEVELS));
        assertEquals("false", getValue(fileRealm, ElytronDescriptionConstants.ENCODED));
    }

    private void testFilteringKeyStoreDefinition() {
        ModelNode keystore = serverModel.get(ElytronDescriptionConstants.FILTERING_KEY_STORE).get("FilteringKeyStore");
        assertEquals("NONE:+firefly", getValue(keystore, ElytronDescriptionConstants.ALIAS_FILTER));
    }

    private void testIdentityRealm() {
        ModelNode realm = serverModel.get(ElytronDescriptionConstants.IDENTITY_REALM).get("local");
        assertEquals("$local", getValue(realm, ElytronDescriptionConstants.IDENTITY));
        assertEquals("groups", getValue(realm, ElytronDescriptionConstants.ATTRIBUTE_NAME));
        assertEquals(Arrays.asList("SuperUser"), getValue(realm, ElytronDescriptionConstants.ATTRIBUTE_VALUES, true));
    }

    private void testJaspiConfiguration() {
        ModelNode jaspi = serverModel.get(ElytronDescriptionConstants.JASPI_CONFIGURATION).get("test");
        assertEquals("HttpServlet", getValue(jaspi, ElytronDescriptionConstants.LAYER));
        assertEquals("default /test", getValue(jaspi, ElytronDescriptionConstants.APPLICATION_CONTEXT));
        assertEquals("Test Definition", getValue(jaspi, ElytronDescriptionConstants.DESCRIPTION));

        ModelNode testModule = jaspi.get(ElytronDescriptionConstants.SERVER_AUTH_MODULES).get(0);
        assertEquals("REQUISITE", getValue(testModule, ElytronDescriptionConstants.FLAG));

        ModelNode options = testModule.get(ElytronDescriptionConstants.OPTIONS);
        assertEquals("b", getValue(options, "a"));
    }

    private void testJdbcRealm() {
        ModelNode jdbcRealm = serverModel.get(ElytronDescriptionConstants.JDBC_REALM).get("JdbcRealm").get(ElytronDescriptionConstants.PRINCIPAL_QUERY).get(0);

        // Bcrypt password mapper
        ModelNode mapper = jdbcRealm.get(ElytronDescriptionConstants.BCRYPT_MAPPER);
        assertEquals("2", getValue(mapper, ElytronDescriptionConstants.PASSWORD_INDEX));
        assertEquals("3", getValue(mapper, ElytronDescriptionConstants.SALT_INDEX));
        assertEquals("4", getValue(mapper, ElytronDescriptionConstants.ITERATION_COUNT_INDEX));
        assertEquals("hex", getValue(mapper, ElytronDescriptionConstants.HASH_ENCODING));
        assertEquals("hex", getValue(mapper, ElytronDescriptionConstants.SALT_ENCODING));

        // Clear password mapper
        mapper = jdbcRealm.get(ElytronDescriptionConstants.CLEAR_PASSWORD_MAPPER);
        assertEquals("2", getValue(mapper, ElytronDescriptionConstants.PASSWORD_INDEX));

        // Simple digest password mapper
        mapper = jdbcRealm.get(ElytronDescriptionConstants.SIMPLE_DIGEST_MAPPER);
        assertEquals("2", getValue(mapper, ElytronDescriptionConstants.PASSWORD_INDEX));
        assertEquals("hex", getValue(mapper, ElytronDescriptionConstants.HASH_ENCODING));

        // Salted simple digest password mapper
        mapper = jdbcRealm.get(ElytronDescriptionConstants.SALTED_SIMPLE_DIGEST_MAPPER);
        assertEquals("2", getValue(mapper, ElytronDescriptionConstants.PASSWORD_INDEX));
        assertEquals("3", getValue(mapper, ElytronDescriptionConstants.SALT_INDEX));
        assertEquals("password-salt-digest-sha-1", getValue(mapper, ElytronDescriptionConstants.ALGORITHM));
        assertEquals("hex", getValue(mapper, ElytronDescriptionConstants.HASH_ENCODING));
        assertEquals("hex", getValue(mapper, ElytronDescriptionConstants.SALT_ENCODING));

        // Scram password mapper
        mapper = jdbcRealm.get(ElytronDescriptionConstants.SCRAM_MAPPER);
        assertEquals("2", getValue(mapper, ElytronDescriptionConstants.PASSWORD_INDEX));
        assertEquals("3", getValue(mapper, ElytronDescriptionConstants.SALT_INDEX));
        assertEquals("4", getValue(mapper, ElytronDescriptionConstants.ITERATION_COUNT_INDEX));
        assertEquals("hex", getValue(mapper, ElytronDescriptionConstants.HASH_ENCODING));
        assertEquals("hex", getValue(mapper, ElytronDescriptionConstants.SALT_ENCODING));

        // Modular crypt mapper
        mapper = jdbcRealm.get(ElytronDescriptionConstants.MODULAR_CRYPT_MAPPER);
        assertEquals("2", getValue(mapper, ElytronDescriptionConstants.PASSWORD_INDEX));
    }

    private void testKerberosSecurityFactory() {
        ModelNode kerberos = serverModel.get(ElytronDescriptionConstants.KERBEROS_SECURITY_FACTORY).get("KerberosFactory");
        assertEquals("bob@Elytron.org", getValue(kerberos, ElytronDescriptionConstants.PRINCIPAL));
        assertEquals("10", getValue(kerberos, ElytronDescriptionConstants.MINIMUM_REMAINING_LIFETIME));
        assertEquals("120", getValue(kerberos, ElytronDescriptionConstants.REQUEST_LIFETIME));
        assertEquals("100", getValue(kerberos, ElytronDescriptionConstants.FAIL_CACHE));
        assertEquals("false", getValue(kerberos, ElytronDescriptionConstants.SERVER));
        assertEquals("true", getValue(kerberos, ElytronDescriptionConstants.OBTAIN_KERBEROS_TICKET));
        assertEquals("true", getValue(kerberos, ElytronDescriptionConstants.DEBUG));
        assertEquals("true", getValue(kerberos, ElytronDescriptionConstants.WRAP_GSS_CREDENTIAL));
        assertEquals("true", getValue(kerberos, ElytronDescriptionConstants.REQUIRED));
        assertEquals(Arrays.asList("KRB5", "KRB5LEGACY"), getValue(kerberos, ElytronDescriptionConstants.MECHANISM_NAMES, true));
        assertEquals(Arrays.asList("1.2.840.113554.1.2.2", "1.3.6.1.5.5.2"), getValue(kerberos, ElytronDescriptionConstants.MECHANISM_OIDS, true));
    }

    private void testKeyStore() {
        ModelNode keystore = serverModel.get(ElytronDescriptionConstants.KEY_STORE).get("jks_store");
        assertEquals("jks", getValue(keystore, ElytronDescriptionConstants.TYPE));
        assertEquals("SunJSSE", getValue(keystore, ElytronDescriptionConstants.PROVIDER_NAME));
        assertEquals("one,two,three", getValue(keystore, ElytronDescriptionConstants.ALIAS_FILTER));
        assertEquals("true", getValue(keystore, ElytronDescriptionConstants.REQUIRED));
    }

    private void testLdapKeyStore() {
        ModelNode keystore = serverModel.get(ElytronDescriptionConstants.LDAP_KEY_STORE).get("LdapKeyStore");

        // search
        assertEquals("dc=elytron,dc=wildfly,dc=org", getValue(keystore, ElytronDescriptionConstants.SEARCH_PATH));
        assertEquals("true", getValue(keystore, ElytronDescriptionConstants.SEARCH_RECURSIVE));
        assertEquals("1000", getValue(keystore, ElytronDescriptionConstants.SEARCH_TIME_LIMIT));
        assertEquals("(&(objectClass=inetOrgPerson)(sn={0}))", getValue(keystore, ElytronDescriptionConstants.FILTER_ALIAS));
        assertEquals("(&(objectClass=inetOrgPerson)(usercertificate={0}))", getValue(keystore, ElytronDescriptionConstants.FILTER_CERTIFICATE));
        assertEquals("(sn=serenity*)", getValue(keystore, ElytronDescriptionConstants.FILTER_ITERATE));

        // attribute mapping
        assertEquals("sn", getValue(keystore, ElytronDescriptionConstants.ALIAS_ATTRIBUTE));
        assertEquals("usercertificate", getValue(keystore, ElytronDescriptionConstants.CERTIFICATE_ATTRIBUTE));
        assertEquals("X.509", getValue(keystore, ElytronDescriptionConstants.CERTIFICATE_TYPE));
        assertEquals("userSMIMECertificate", getValue(keystore, ElytronDescriptionConstants.CERTIFICATE_CHAIN_ATTRIBUTE));
        assertEquals("PKCS7", getValue(keystore, ElytronDescriptionConstants.CERTIFICATE_CHAIN_ENCODING));
        assertEquals("userPKCS12", getValue(keystore, ElytronDescriptionConstants.KEY_ATTRIBUTE));
        assertEquals("PKCS12", getValue(keystore, ElytronDescriptionConstants.KEY_TYPE));

        // new item template
        ModelNode template = keystore.get(ElytronDescriptionConstants.NEW_ITEM_TEMPLATE);
        assertEquals("ou=keystore,dc=elytron,dc=wildfly,dc=org", getValue(template, ElytronDescriptionConstants.NEW_ITEM_PATH));
        assertEquals("cn", getValue(template, ElytronDescriptionConstants.NEW_ITEM_RDN));
        assertEquals("objectClass", getValue(template.get(ElytronDescriptionConstants.NEW_ITEM_ATTRIBUTES).get(0), ElytronDescriptionConstants.NAME));
        assertEquals(Arrays.asList("top", "inetOrgPerson"), getValue(template.get(ElytronDescriptionConstants.NEW_ITEM_ATTRIBUTES).get(0), ElytronDescriptionConstants.VALUE, true));
    }

    private void testLdapRealm() {
        ModelNode realm = serverModel.get(ElytronDescriptionConstants.LDAP_REALM).get("LdapRealmWithAttributeMapping");
        assertEquals("false", getValue(realm, ElytronDescriptionConstants.DIRECT_VERIFICATION));
        assertEquals("true", getValue(realm, ElytronDescriptionConstants.ALLOW_BLANK_PASSWORD));

        // Identity mapping
        ModelNode identityMapping = realm.get(ElytronDescriptionConstants.IDENTITY_MAPPING);
        assertEquals("uid", getValue(identityMapping, ElytronDescriptionConstants.RDN_IDENTIFIER));
        assertEquals("true", getValue(identityMapping, ElytronDescriptionConstants.USE_RECURSIVE_SEARCH));
        assertEquals("dc=elytron,dc=wildfly,dc=org", getValue(identityMapping, ElytronDescriptionConstants.SEARCH_BASE_DN));
        assertEquals("(rdn_identifier={0})", getValue(identityMapping, ElytronDescriptionConstants.FILTER_NAME));
        assertEquals("(uid=*)", getValue(identityMapping, ElytronDescriptionConstants.ITERATOR_FILTER));
        assertEquals("dc=elytron,dc=wildfly,dc=org", getValue(identityMapping, ElytronDescriptionConstants.NEW_IDENTITY_PARENT_DN));

        // Attribute mapping
        ModelNode attributeMapping = identityMapping.get(ElytronDescriptionConstants.ATTRIBUTE_MAPPING);
        assertEquals("CN", getValue(attributeMapping.get(0), ElytronDescriptionConstants.FROM));
        assertEquals("businessUnit", getValue(attributeMapping.get(0), ElytronDescriptionConstants.TO));
        assertEquals("ref", getValue(attributeMapping.get(0), ElytronDescriptionConstants.REFERENCE));
        assertEquals("(&(objectClass=groupOfNames)(member={0}))", getValue(attributeMapping.get(0), ElytronDescriptionConstants.FILTER));
        assertEquals("ou=Finance,dc=elytron,dc=wildfly,dc=org", getValue(attributeMapping.get(0), ElytronDescriptionConstants.FILTER_BASE_DN));
        assertEquals("true", getValue(attributeMapping.get(0), ElytronDescriptionConstants.SEARCH_RECURSIVE));
        assertEquals("0", getValue(attributeMapping.get(0), ElytronDescriptionConstants.ROLE_RECURSION));
        assertEquals("cn", getValue(attributeMapping.get(0), ElytronDescriptionConstants.ROLE_RECURSION_NAME));
        assertEquals("OU", getValue(attributeMapping.get(1), ElytronDescriptionConstants.EXTRACT_RDN));

        // User password credential mapping
        ModelNode userPass = identityMapping.get(ElytronDescriptionConstants.USER_PASSWORD_MAPPER);
        assertEquals("userPassword", getValue(userPass, ElytronDescriptionConstants.FROM));
        assertEquals("true", getValue(userPass, ElytronDescriptionConstants.WRITABLE));
        assertEquals("true", getValue(userPass, ElytronDescriptionConstants.VERIFIABLE));

        // Otp credential mapping
        ModelNode otp = identityMapping.get(ElytronDescriptionConstants.OTP_CREDENTIAL_MAPPER);
        assertEquals("otpAlgorithm", getValue(otp, ElytronDescriptionConstants.ALGORITHM_FROM));
        assertEquals("otpHash", getValue(otp, ElytronDescriptionConstants.HASH_FROM));
        assertEquals("otpSeed", getValue(otp, ElytronDescriptionConstants.SEED_FROM));
        assertEquals("otpSequence", getValue(otp, ElytronDescriptionConstants.SEQUENCE_FROM));

        // X509 Credential mapping
        ModelNode x509 = identityMapping.get(ElytronDescriptionConstants.X509_CREDENTIAL_MAPPER);
        assertEquals("x509digest", getValue(x509, ElytronDescriptionConstants.DIGEST_FROM));
        assertEquals("usercertificate", getValue(x509, ElytronDescriptionConstants.CERTIFICATE_FROM));
        assertEquals("SHA-1", getValue(x509, ElytronDescriptionConstants.DIGEST_ALGORITHM));
        assertEquals("x509serialNumber", getValue(x509, ElytronDescriptionConstants.SERIAL_NUMBER_FROM));
        assertEquals("x509subject", getValue(x509, ElytronDescriptionConstants.SUBJECT_DN_FROM));

        // New identity attribute
        ModelNode newIdentity = identityMapping.get(ElytronDescriptionConstants.NEW_IDENTITY_ATTRIBUTES).get(0);
        assertEquals("sn", getValue(newIdentity, ElytronDescriptionConstants.NAME));
        assertEquals(Arrays.asList("BlankSurname"), getValue(newIdentity, ElytronDescriptionConstants.VALUE, true));
    }

    private void testPermissionMappers() {
        ModelNode mapper = serverModel.get(ElytronDescriptionConstants.SIMPLE_PERMISSION_MAPPER).get("SimplePermissionMapperLegacy");
        assertEquals("and", getValue(mapper, ElytronDescriptionConstants.MAPPING_MODE));

        mapper = mapper.get(ElytronDescriptionConstants.PERMISSION_MAPPINGS).get(0);
        assertEquals(Arrays.asList("John", "Joe"), getValue(mapper, ElytronDescriptionConstants.PRINCIPALS, true));
        assertEquals(Arrays.asList("User", "Administrator"), getValue(mapper, ElytronDescriptionConstants.ROLES, true));

        mapper = mapper.get(ElytronDescriptionConstants.PERMISSIONS).get(1);
        assertEquals("../c", getValue(mapper, ElytronDescriptionConstants.TARGET_NAME));
        assertEquals("delete", getValue(mapper, ElytronDescriptionConstants.ACTION));
    }

    private void testPrincipalDecoders() {
        ModelNode decoder = serverModel.get(ElytronDescriptionConstants.X500_ATTRIBUTE_PRINCIPAL_DECODER).get("MyX500PrincipalDecoderTwo");
        assertEquals("2.5.4.3", getValue(decoder, ElytronDescriptionConstants.OID));
        assertEquals(",", getValue(decoder, ElytronDescriptionConstants.JOINER));
        assertEquals("2", getValue(decoder, ElytronDescriptionConstants.START_SEGMENT));
        assertEquals("1", getValue(decoder, ElytronDescriptionConstants.MAXIMUM_SEGMENTS));
        assertEquals("true", getValue(decoder, ElytronDescriptionConstants.REVERSE));
        assertEquals("true", getValue(decoder, ElytronDescriptionConstants.CONVERT));
        assertEquals(Arrays.asList("2.5.4.3", "2.5.4.11"), getValue(decoder, ElytronDescriptionConstants.REQUIRED_OIDS, true));

        decoder = serverModel.get(ElytronDescriptionConstants.X500_ATTRIBUTE_PRINCIPAL_DECODER).get("MyCnDecoder");
        assertEquals("Cn", getValue(decoder, ElytronDescriptionConstants.ATTRIBUTE_NAME));

        decoder = serverModel.get(ElytronDescriptionConstants.CONSTANT_PRINCIPAL_DECODER).get("ConstantDecoder");
        assertEquals("test", getValue(decoder, ElytronDescriptionConstants.CONSTANT));
    }

    private void testPrincipalTransformers() {
        ModelNode pt = serverModel.get(ElytronDescriptionConstants.REGEX_PRINCIPAL_TRANSFORMER).get("NameRewriterXY");
        assertEquals("y$1", getValue(pt, ElytronDescriptionConstants.REPLACEMENT));
        assertEquals("x(.*)", getValue(pt, ElytronDescriptionConstants.PATTERN));
        assertEquals("false", getValue(pt, ElytronDescriptionConstants.REPLACE_ALL));

        pt = serverModel.get(ElytronDescriptionConstants.CONSTANT_PRINCIPAL_TRANSFORMER).get("ConstantNameRewriter");
        assertEquals("test", getValue(pt, ElytronDescriptionConstants.CONSTANT));

        pt = serverModel.get(ElytronDescriptionConstants.CASE_PRINCIPAL_TRANSFORMER).get("CaseNameRewriter");
        assertEquals("true", getValue(pt, ElytronDescriptionConstants.UPPER_CASE));

        pt = serverModel.get(ElytronDescriptionConstants.REGEX_VALIDATING_PRINCIPAL_TRANSFORMER).get("RegexValidateNameRewriter");
        assertEquals("true", getValue(pt, ElytronDescriptionConstants.MATCH));
    }

    private void testPropertiesRealm() {
        ModelNode propRealm = serverModel.get(ElytronDescriptionConstants.PROPERTIES_REALM).get("PropRealm");
        assertEquals("FileRealm", getValue(propRealm.get(ElytronDescriptionConstants.USERS_PROPERTIES), ElytronDescriptionConstants.DIGEST_REALM_NAME));
        assertEquals("false", getValue(propRealm.get(ElytronDescriptionConstants.USERS_PROPERTIES), ElytronDescriptionConstants.PLAIN_TEXT));
        assertEquals("groups", getValue(propRealm, ElytronDescriptionConstants.GROUPS_ATTRIBUTE));
    }

    private void testProvider() {
        ModelNode provider = serverModel.get(ElytronDescriptionConstants.PROVIDER_LOADER).get("openssl");
        assertEquals("val", getValue(provider.get(ElytronDescriptionConstants.CONFIGURATION), "prop"));
        provider = serverModel.get(ElytronDescriptionConstants.PROVIDER_LOADER).get("elytron");
        assertEquals("arg", getValue(provider, ElytronDescriptionConstants.ARGUMENT));
    }

    private void testRealmMappers() {
        ModelNode realmMapper = serverModel.get(ElytronDescriptionConstants.MAPPED_REGEX_REALM_MAPPER).get("MappedRealmMapper");
        assertEquals(".*@(.*)", getValue(realmMapper, ElytronDescriptionConstants.PATTERN));
        assertEquals("test", getValue(realmMapper.get(ElytronDescriptionConstants.REALM_MAP), "test"));
    }

    private void testRoleDecoders() {
        ModelNode roleDecoder = serverModel.get(ElytronDescriptionConstants.SOURCE_ADDRESS_ROLE_DECODER).get("ipRoleDecoder");
        assertEquals(Arrays.asList("admin", "user"), getValue(roleDecoder, ElytronDescriptionConstants.ROLES, true));
        assertEquals("10.12.14.16", getValue(roleDecoder, ElytronDescriptionConstants.SOURCE_ADDRESS));
        roleDecoder = serverModel.get(ElytronDescriptionConstants.SOURCE_ADDRESS_ROLE_DECODER).get("regexRoleDecoder");
        assertEquals("10\\.12\\.14\\.\\d+$", getValue(roleDecoder, ElytronDescriptionConstants.PATTERN));
    }

    private void testRoleMappers() {
        ModelNode roleMapper = serverModel.get(ElytronDescriptionConstants.ADD_PREFIX_ROLE_MAPPER).get("RolePrefixer");
        assertEquals("prefix", getValue(roleMapper, ElytronDescriptionConstants.PREFIX));
        roleMapper = serverModel.get(ElytronDescriptionConstants.ADD_SUFFIX_ROLE_MAPPER).get("RoleSuffixer");
        assertEquals("suffix", getValue(roleMapper, ElytronDescriptionConstants.SUFFIX));

        roleMapper = serverModel.get(ElytronDescriptionConstants.MAPPED_ROLE_MAPPER).get("MappedRoleMapper");
        assertEquals("false", getValue(roleMapper, ElytronDescriptionConstants.KEEP_MAPPED));
        assertEquals("true", getValue(roleMapper, ElytronDescriptionConstants.KEEP_NON_MAPPED));

        roleMapper = serverModel.get(ElytronDescriptionConstants.REGEX_ROLE_MAPPER).get("RegexRoleMapper");
        assertEquals("false", getValue(roleMapper, ElytronDescriptionConstants.REPLACEMENT));
        assertEquals("false", getValue(roleMapper, ElytronDescriptionConstants.REPLACE_ALL));
        assertEquals("*(x.)", getValue(roleMapper, ElytronDescriptionConstants.PATTERN));

        roleMapper = serverModel.get(ElytronDescriptionConstants.CONSTANT_ROLE_MAPPER).get("ConstantRoleMapper");
        assertEquals(Arrays.asList("role"), getValue(roleMapper, ElytronDescriptionConstants.ROLES, true));

        roleMapper = serverModel.get(ElytronDescriptionConstants.LOGICAL_ROLE_MAPPER).get("LogicalRoleMapper");
        assertEquals("and", getValue(roleMapper, ElytronDescriptionConstants.LOGICAL_OPERATION));
    }

    private void testSaslServer() {
        ModelNode factory = serverModel.get(ElytronDescriptionConstants.SASL_AUTHENTICATION_FACTORY).get("SaslAuthenticationDefinition").get(ElytronDescriptionConstants.MECHANISM_CONFIGURATIONS).get(0);
        assertEquals("PLAIN", getValue(factory, ElytronDescriptionConstants.MECHANISM_NAME));
        assertEquals("host", getValue(factory, ElytronDescriptionConstants.HOST_NAME));
        assertEquals("protocol", getValue(factory, ElytronDescriptionConstants.PROTOCOL));
        assertEquals("Test Realm", getValue(factory.get(ElytronDescriptionConstants.MECHANISM_REALM_CONFIGURATIONS).get(0), ElytronDescriptionConstants.REALM_NAME));
    }

    private void testSSLComponents() {
        // SSL Context
        ModelNode context = serverModel.get(ElytronDescriptionConstants.SERVER_SSL_CONTEXT).get("server");
        assertEquals(Arrays.asList("TLSv1.2"), getValue(context, ElytronDescriptionConstants.PROTOCOLS, true));
        assertEquals("true", getValue(context, ElytronDescriptionConstants.WANT_CLIENT_AUTH));
        assertEquals("true", getValue(context, ElytronDescriptionConstants.NEED_CLIENT_AUTH));
        assertEquals("true", getValue(context, ElytronDescriptionConstants.AUTHENTICATION_OPTIONAL));
        assertEquals("false", getValue(context, ElytronDescriptionConstants.USE_CIPHER_SUITES_ORDER));
        assertEquals("false", getValue(context, ElytronDescriptionConstants.WRAP));
        assertEquals("first", getValue(context, ElytronDescriptionConstants.PROVIDER_NAME));
        assertEquals("DEFAULT", getValue(context, ElytronDescriptionConstants.CIPHER_SUITE_FILTER));
        assertEquals("name", getValue(context, ElytronDescriptionConstants.CIPHER_SUITE_NAMES));
        assertEquals("10", getValue(context, ElytronDescriptionConstants.MAXIMUM_SESSION_CACHE_SIZE));
        assertEquals("120", getValue(context, ElytronDescriptionConstants.SESSION_TIMEOUT));

        // Trust Managers
        ModelNode tm = serverModel.get(ElytronDescriptionConstants.TRUST_MANAGER).get("trust-with-ocsp").get(ElytronDescriptionConstants.OCSP);
        assertEquals("http://localhost/ocsp", getValue(tm, ElytronDescriptionConstants.RESPONDER));
        assertEquals("jceks_store", getValue(tm, ElytronDescriptionConstants.RESPONDER_KEYSTORE));
        assertEquals("responder-alias", getValue(tm, ElytronDescriptionConstants.RESPONDER_CERTIFICATE));

        tm = serverModel.get(ElytronDescriptionConstants.TRUST_MANAGER).get("trust-with-crl").get(ElytronDescriptionConstants.CERTIFICATE_REVOCATION_LIST);
        assertEquals("crl.pem", getValue(tm, ElytronDescriptionConstants.PATH));
        assertEquals("2", getValue(tm, ElytronDescriptionConstants.MAXIMUM_CERT_PATH));

        // Key Managers
        ModelNode keyManager = serverModel.get(ElytronDescriptionConstants.KEY_MANAGER).get("serverKey2");
        assertEquals("SunX509", getValue(keyManager, ElytronDescriptionConstants.ALGORITHM));
        assertEquals("one,two,three", getValue(keyManager, ElytronDescriptionConstants.ALIAS_FILTER));
        assertEquals("localhost", getValue(keyManager, ElytronDescriptionConstants.GENERATE_SELF_SIGNED_CERTIFICATE_HOST));
    }

    private void testTokenRealm() {
        ModelNode realm = serverModel.get(ElytronDescriptionConstants.TOKEN_REALM).get("JwtRealmOne");
        assertEquals("sub", getValue(realm, ElytronDescriptionConstants.PRINCIPAL_CLAIM));

        ModelNode jwt = realm.get(ElytronDescriptionConstants.JWT);
        assertEquals(Arrays.asList("some-issuer-a"), getValue(jwt, ElytronDescriptionConstants.ISSUER, true));
        assertEquals(Arrays.asList("some-audience-a"), getValue(jwt, ElytronDescriptionConstants.AUDIENCE, true));
        assertEquals("MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCrVrCuTtArbgaZzL1hvh0xtL5mc7o0NqP", getValue(jwt, ElytronDescriptionConstants.PUBLIC_KEY));
        assertEquals("cert", getValue(jwt, ElytronDescriptionConstants.CERTIFICATE));
        assertEquals("public", getValue(jwt.get(ElytronDescriptionConstants.KEY_MAP), ElytronDescriptionConstants.KID));

        // OAuth
        ModelNode oauth = serverModel.get(ElytronDescriptionConstants.TOKEN_REALM).get("OAuth2Realm").get(ElytronDescriptionConstants.OAUTH2_INTROSPECTION);
        assertEquals("host", getValue(oauth, ElytronDescriptionConstants.HOST_NAME_VERIFICATION_POLICY));
        assertEquals("a", getValue(oauth, ElytronDescriptionConstants.CLIENT_ID));
        assertEquals("b", getValue(oauth, ElytronDescriptionConstants.CLIENT_SECRET));
        assertEquals("https://localhost/token/introspect", getValue(oauth, ElytronDescriptionConstants.INTROSPECTION_URL));
    }

    private Object getValue(ModelNode node, String attributeName) {
        return getValue(node, attributeName, false);
    }

    private Object getValue(ModelNode node, String attributeName, boolean isList) {
        ModelNode result = node.get(attributeName).resolve();
        if (! isList) {
            return result.asString();
        }
        List<String> results = new ArrayList<>();
        for (ModelNode n : result.asList()) {
            results.add(n.asString());
        }
        return results;
    }
}
