ninja: fix tests after CASSANDRA-21074 landed

Author: smiklosovic

test/unit/org/apache/cassandra/io/compress/CompressionMetadataTest.java

@@ -101,9 +101,10 @@ public void testDictionaryRefCountingDuringSchemaChange()
         byte[] dictBytes = "sample dictionary data for compression".getBytes();
         ZstdCompressionDictionary dictionary = new ZstdCompressionDictionary(dictId, dictBytes);
 
-        assertThat(dictionary.selfRef().globalCount()).isOne();
+        assertThat(dictionary.selfRef()).isNull();
         CompressionDictionaryCache cache = new CompressionDictionaryCache();
         cache.add(dictionary);
+        assertThat(dictionary.selfRef().globalCount()).isOne();
 
         // Verify dictionary is in cache
         CompressionDictionary cachedDict = cache.get(dictId);
@@ -187,10 +188,12 @@ public void testMultipleMetadataInstancesSharingDictionary()
         byte[] dictBytes = "shared dictionary data".getBytes();
         ZstdCompressionDictionary dictionary = new ZstdCompressionDictionary(dictId, dictBytes);
 
-        assertThat(dictionary.selfRef().globalCount()).isOne();
+        assertThat(dictionary.selfRef()).isNull();
         CompressionDictionaryCache cache = new CompressionDictionaryCache();
         cache.add(dictionary);
 
+        assertThat(dictionary.selfRef().globalCount()).isOne();
+
         // Create multiple CompressionMetadata instances (simulating multiple SSTables)
         Memory memory1 = Memory.allocate(100);
         CompressionMetadata metadata1 = new CompressionMetadata(

test/unit/org/apache/cassandra/io/compress/ZstdDictionaryCompressorTest.java

@@ -281,6 +281,7 @@ public void testClosedDictionaryHandling()
     {
         ZstdDictionaryCompressor.invalidateCache();
         ZstdCompressionDictionary closedDict = createTestDictionary();
+        closedDict.initRefLazily();
         closedDict.close();
 
         // This should throw IllegalStateException