public void cleanup(QueryId queryId)
    {
        queryPartitionFileCounterMap.remove(queryId);
        queryHiveMetadataResultMap.remove(queryId);
    }
@Test
    public void testCleanup()
    {
        HiveFileRenamer hiveFileRenamer = new HiveFileRenamer();
        List<ConnectorMetadataUpdateHandle> requests = ImmutableList.of(TEST_HIVE_METADATA_UPDATE_REQUEST);
        List<ConnectorMetadataUpdateHandle> results = hiveFileRenamer.getMetadataUpdateResults(requests, TEST_QUERY_ID);
        assertEquals(results.size(), 1);
        HiveMetadataUpdateHandle result = (HiveMetadataUpdateHandle) results.get(0);
        assertEquals(result.getMetadataUpdate(), Optional.of("0"));

        hiveFileRenamer.cleanup(TEST_QUERY_ID);

        requests = ImmutableList.of(TEST_HIVE_METADATA_UPDATE_REQUEST);
        results = hiveFileRenamer.getMetadataUpdateResults(requests, TEST_QUERY_ID);
        assertEquals(results.size(), 1);
        result = (HiveMetadataUpdateHandle) results.get(0);
        assertEquals(result.getMetadataUpdate(), Optional.of("0"));
    }