Address review comments

apache · Jan 30, 2024 · e7720d5 · e7720d5
1 parent cadcdd3
commit e7720d5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ql/src/test/org/apache/hadoop/hive/ql/txn/compactor/handler/TestAbortedTxnCleaner.java b/ql/src/test/org/apache/hadoop/hive/ql/txn/compactor/handler/TestAbortedTxnCleaner.java
@@ -286,6 +286,9 @@ public void testCleaningOfAbortedDirectoriesBelowBase() throws Exception {
 
     Mockito.verifyNoInteractions(mockedFSRemover);
     Mockito.verify(mockedTaskHandler, Mockito.times(1)).getTasks();
+    String compactionQueuePresence = "SELECT COUNT(*) FROM \"COMPACTION_QUEUE\" " +
+            " WHERE \"CQ_DATABASE\" = '" + dbName+ "' AND \"CQ_TABLE\" = '" + tableName + "' AND \"CQ_PARTITION\" IS NULL";
+    Assert.assertEquals(1, TestTxnDbUtil.countQueryAgent(conf, compactionQueuePresence));
 
     directories = getDirectories(conf, t, null);
     // Both base and delta files are present since the cleaner skips them as there is a newer write.
@@ -327,6 +330,7 @@ public void testAbortedCleaningWithThreeTxnsWithDiffWriteIds() throws Exception
     addDeltaFile(t, null, writeId1, writeId1, 2);
     addDeltaFile(t, null, writeId2, writeId2, 2);
 
+    ms.abortTxns(Collections.singletonList(openTxnId2));
     ms.commitTxn(openTxnId3);
 
     HiveConf.setIntVar(conf, HiveConf.ConfVars.HIVE_COMPACTOR_ABORTEDTXN_THRESHOLD, 0);