Avoid retaining generated classes in TestFlatHashStrategy

trinodb · Nov 13, 2023 · 9ac52d5 · 9ac52d5
1 parent 5617bce
commit 9ac52d5
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/core/trino-main/src/test/java/io/trino/operator/TestFlatHashStrategy.java b/core/trino-main/src/test/java/io/trino/operator/TestFlatHashStrategy.java
@@ -51,14 +51,14 @@
 
 public class TestFlatHashStrategy
 {
-    private static final TypeOperators TYPE_OPERATORS = new TypeOperators();
-    private static final JoinCompiler JOIN_COMPILER = new JoinCompiler(TYPE_OPERATORS);
+    private final TypeOperators typeOperators = new TypeOperators();
+    private final JoinCompiler joinCompiler = new JoinCompiler(typeOperators);
 
     @Test
     public void testBatchedRawHashesZeroLength()
     {
-        List<Type> types = createTestingTypes();
-        FlatHashStrategy flatHashStrategy = JOIN_COMPILER.getFlatHashStrategy(types);
+        List<Type> types = createTestingTypes(typeOperators);
+        FlatHashStrategy flatHashStrategy = joinCompiler.getFlatHashStrategy(types);
 
         int positionCount = 10;
         // Attempting to touch any of the blocks would result in a NullPointerException
@@ -68,8 +68,8 @@ public void testBatchedRawHashesZeroLength()
     @Test
     public void testBatchedRawHashesMatchSinglePositionHashes()
     {
-        List<Type> types = createTestingTypes();
-        FlatHashStrategy flatHashStrategy = JOIN_COMPILER.getFlatHashStrategy(types);
+        List<Type> types = createTestingTypes(typeOperators);
+        FlatHashStrategy flatHashStrategy = joinCompiler.getFlatHashStrategy(types);
 
         int positionCount = 1024;
         Block[] blocks = new Block[types.size()];
@@ -102,7 +102,7 @@ private static void assertHashesEqual(List<Type> types, Block[] blocks, long[] b
         }
     }
 
-    private static List<Type> createTestingTypes()
+    private static List<Type> createTestingTypes(TypeOperators typeOperators)
     {
         List<Type> baseTypes = List.of(
                 BIGINT,
@@ -128,7 +128,7 @@ private static List<Type> createTestingTypes()
         builder.add(RowType.anonymous(baseTypes));
         for (Type baseType : baseTypes) {
             builder.add(new ArrayType(baseType));
-            builder.add(new MapType(baseType, baseType, TYPE_OPERATORS));
+            builder.add(new MapType(baseType, baseType, typeOperators));
         }
         return builder.build();
     }