@Test
  public void testQueryMonth() throws Exception {
    long readSuccessCountBefore = metrics.readSuccessCount.getCount();
    String monthValue = aggregatedAccountReportsDao.queryMonthForCluster(clusterName);
    assertTrue(queryMonthValue.equals(monthValue));
    verify(mockConnection).prepareStatement(anyString());
    assertEquals("Read success count should be " + (readSuccessCountBefore + 1), readSuccessCountBefore + 1,
        metrics.readSuccessCount.getCount());
  }
Utility to get a {@link DataSource}.
@param mockConnection the connection to use.
@return the {@link DataSource}.
@throws SQLException
ambry-mysql/src/test/java/com/github/ambry/accountstats/AggregatedAccountReportsDaoTest.java