private HostPartitionClassStorageStatsWrapper convertHostAccountStorageStatsToHostPartitionClassStorageStats(
      HostAccountStorageStatsWrapper accountStatsWrapper, Map<Long, String> partitionIdToClassName) {
    HostPartitionClassStorageStats hostPartitionClassStorageStats = new HostPartitionClassStorageStats();
    Map<Long, Map<Short, Map<Short, ContainerStorageStats>>> storageStats =
        accountStatsWrapper.getStats().getStorageStats();
    for (long partitionId : storageStats.keySet()) {
      Map<Short, Map<Short, ContainerStorageStats>> accountStorageStatsMap = storageStats.get(partitionId);
      String partitionClassName = partitionIdToClassName.get(partitionId);
      for (short accountId : accountStorageStatsMap.keySet()) {
        accountStorageStatsMap.get(accountId)
            .values()
            .forEach(containerStats -> hostPartitionClassStorageStats.addContainerStorageStats(partitionClassName,
                partitionId, accountId, containerStats));
      }
    }
    return new HostPartitionClassStorageStatsWrapper(new StatsHeader(accountStatsWrapper.getHeader()),
        hostPartitionClassStorageStats);
  }
Test methods to store and fetch partition class, partition name partition id and partition class storage stats.
@throws Exception
ambry-mysql/src/integration-test/java/com/github/ambry/accountstats/AccountStatsMySqlStoreIntegrationTest.java