From ceeb89bc581815715756c7477ce419f147b00074 Mon Sep 17 00:00:00 2001
From: Peter Veentjer <alarmnummer@gmail.com>
Date: Mon, 24 Feb 2025 06:19:29 +0200
Subject: [PATCH] Added tests AtomicBuffer acq/rel/opaque methods

Also fixed a bug with the opaque int methods. It was checking
if there was 'long' space available instead of 'int' space.
---
 .../org/agrona/concurrent/UnsafeBuffer.java   |  6 +--
 .../agrona/concurrent/AtomicBufferTest.java   | 48 +++++++++++++++++++
 2 files changed, 51 insertions(+), 3 deletions(-)

diff --git a/agrona/src/main/java/org/agrona/concurrent/UnsafeBuffer.java b/agrona/src/main/java/org/agrona/concurrent/UnsafeBuffer.java
index f46e76c0a..1f4ec33a6 100644
--- a/agrona/src/main/java/org/agrona/concurrent/UnsafeBuffer.java
+++ b/agrona/src/main/java/org/agrona/concurrent/UnsafeBuffer.java
@@ -623,7 +623,7 @@ public void putIntOpaque(final int index, final int value)
     {
         if (SHOULD_BOUNDS_CHECK)
         {
-            boundsCheck0(index, SIZE_OF_LONG);
+            boundsCheck0(index, SIZE_OF_INT);
         }
 
         UnsafeApi.putIntOpaque(byteArray, addressOffset + index, value);
@@ -636,7 +636,7 @@ public int getIntOpaque(final int index)
     {
         if (SHOULD_BOUNDS_CHECK)
         {
-            boundsCheck0(index, SIZE_OF_LONG);
+            boundsCheck0(index, SIZE_OF_INT);
         }
 
         return UnsafeApi.getIntOpaque(byteArray, addressOffset + index);
@@ -649,7 +649,7 @@ public int addIntOpaque(final int index, final int increment)
     {
         if (SHOULD_BOUNDS_CHECK)
         {
-            boundsCheck0(index, SIZE_OF_LONG);
+            boundsCheck0(index, SIZE_OF_INT);
         }
 
         final int oldValue = UnsafeApi.getIntOpaque(byteArray, addressOffset + index);
diff --git a/agrona/src/test/java/org/agrona/concurrent/AtomicBufferTest.java b/agrona/src/test/java/org/agrona/concurrent/AtomicBufferTest.java
index b4876a50e..04469bce0 100644
--- a/agrona/src/test/java/org/agrona/concurrent/AtomicBufferTest.java
+++ b/agrona/src/test/java/org/agrona/concurrent/AtomicBufferTest.java
@@ -555,6 +555,54 @@ void shouldGetBytesFromBufferToSlice(final ByteBuffer buffer)
         assertThat(result, is(testBytes));
     }
 
+    @Test
+    void verifyIntOpaqueOperations()
+    {
+        final UnsafeBuffer unsafeBuffer = new UnsafeBuffer(new byte[4]);
+        final int index = 0;
+        assertEquals(0, unsafeBuffer.getIntOpaque(index));
+        unsafeBuffer.putIntOpaque(index, 10);
+        assertEquals(10, unsafeBuffer.getIntOpaque(index));
+        assertEquals(10, unsafeBuffer.addIntOpaque(0, 2));
+        assertEquals(12, unsafeBuffer.getIntOpaque(index));
+    }
+
+    @Test
+    void verifyIntReleaseAcquireOperations()
+    {
+        final UnsafeBuffer unsafeBuffer = new UnsafeBuffer(new byte[4]);
+        final int index = 0;
+        assertEquals(0, unsafeBuffer.getIntAcquire(index));
+        unsafeBuffer.putIntRelease(index, 10);
+        assertEquals(10, unsafeBuffer.getIntAcquire(index));
+        assertEquals(10, unsafeBuffer.addIntRelease(0, 2));
+        assertEquals(12, unsafeBuffer.getIntAcquire(index));
+    }
+
+    @Test
+    void verifyLongOpaqueOperations()
+    {
+        final UnsafeBuffer unsafeBuffer = new UnsafeBuffer(new byte[8]);
+        final int index = 0;
+        assertEquals(0, unsafeBuffer.getLongOpaque(index));
+        unsafeBuffer.putLongOpaque(index, 10);
+        assertEquals(10, unsafeBuffer.getLongOpaque(index));
+        assertEquals(10, unsafeBuffer.addLongOpaque(index, 2));
+        assertEquals(12, unsafeBuffer.getLongOpaque(index));
+    }
+
+    @Test
+    void verifyLongReleaseAcquireOperations()
+    {
+        final UnsafeBuffer unsafeBuffer = new UnsafeBuffer(new byte[8]);
+        final int index = 0;
+        assertEquals(0, unsafeBuffer.getLongAcquire(index));
+        unsafeBuffer.putLongRelease(index, 10);
+        assertEquals(10, unsafeBuffer.getLongAcquire(index));
+        assertEquals(10, unsafeBuffer.addLongRelease(index, 2));
+        assertEquals(12, unsafeBuffer.getLongAcquire(index));
+    }
+
     @Test
     void verifyAlignmentShouldThrowForAByteArrayIfStrictAlignmentChecksAreEnabled()
     {