From 27bc5a4aaf56acf312a766947c0d37cda7a33d53 Mon Sep 17 00:00:00 2001
From: Pedro Hos <pedro-hos@outlook.com>
Date: Fri, 6 Sep 2024 13:39:08 -0300
Subject: [PATCH] WFCORE-6937 - Testing of write-attribute calls is low for the
 Elytron subsystem

https://issues.redhat.com/browse/WFCORE-6937
---
 .../elytron/CredentialStoreTestCase.java      | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/elytron/src/test/java/org/wildfly/extension/elytron/CredentialStoreTestCase.java b/elytron/src/test/java/org/wildfly/extension/elytron/CredentialStoreTestCase.java
index e863b09bd0b..9be78e15799 100644
--- a/elytron/src/test/java/org/wildfly/extension/elytron/CredentialStoreTestCase.java
+++ b/elytron/src/test/java/org/wildfly/extension/elytron/CredentialStoreTestCase.java
@@ -9,6 +9,7 @@
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.FAILED;
 import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.OUTCOME;
+import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.SUBSYSTEM;
 import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.SUCCESS;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
@@ -28,7 +29,9 @@
 
 import javax.crypto.SecretKey;
 
+import org.jboss.as.controller.PathAddress;
 import org.jboss.as.controller.client.helpers.ClientConstants;
+import org.jboss.as.controller.operations.common.Util;
 import org.jboss.as.subsystem.test.AbstractSubsystemTest;
 import org.jboss.as.subsystem.test.KernelServices;
 import org.jboss.dmr.ModelNode;
@@ -53,6 +56,10 @@ public class CredentialStoreTestCase extends AbstractSubsystemTest {
     private static final String CLEAR_TEXT = "Lorem ipsum dolor sit amet";
     private static final String CONFIGURATION = "credential-store.xml";
 
+    private static final PathAddress ROOT_ADDRESS = PathAddress.pathAddress(SUBSYSTEM, ElytronExtension.SUBSYSTEM_NAME);
+    private static final PathAddress CRED_STORE_ADDRESS = ROOT_ADDRESS.append(ElytronDescriptionConstants.CREDENTIAL_STORE, "test");
+    private static final PathAddress SECRET_KEY_CRED_STORE_128 = ROOT_ADDRESS.append(ElytronDescriptionConstants.SECRET_KEY_CREDENTIAL_STORE, "test128");
+
     private KernelServices services = null;
 
     public CredentialStoreTestCase() {
@@ -91,6 +98,24 @@ public Void run() {
         });
     }
 
+    @Test
+    public void testWriteAttributeCreateCredentialStoreToFalse() {
+        ModelNode operation = Util.getWriteAttributeOperation(CRED_STORE_ADDRESS, ElytronDescriptionConstants.CREATE, false);
+        assertSuccess(services.executeOperation(operation));
+        assertEquals(false, readResource(CRED_STORE_ADDRESS).get(ElytronDescriptionConstants.CREATE).asBoolean());
+    }
+
+    @Test
+    public void testWriteAttributeKeySizeSecretKeyCredStoreTo192() {
+        ModelNode operation = Util.getWriteAttributeOperation(SECRET_KEY_CRED_STORE_128, ElytronDescriptionConstants.KEY_SIZE, 192);
+        assertSuccess(services.executeOperation(operation));
+        assertEquals(192, readResource(SECRET_KEY_CRED_STORE_128).get(ElytronDescriptionConstants.KEY_SIZE).asInt());
+    }
+
+    private ModelNode readResource(PathAddress address) {
+        return assertSuccess(services.executeOperation(Util.getReadResourceOperation(address))).get(ClientConstants.RESULT);
+    }
+
     // Test Contents of dynamically initialised credential stores
 
     @Test