@Test
  public void testAtlasClientTimeouts() throws Exception {
    try (MockedStatic<UserGroupInformation> userGroupInformationMockedStatic = mockStatic(UserGroupInformation.class);
         MockedStatic<ConfigurationConverter> configurationConverterMockedStatic = mockStatic(ConfigurationConverter.class)) {
      userGroupInformationMockedStatic.when(UserGroupInformation::getLoginUser).
              thenReturn(mock(UserGroupInformation.class));
      configurationConverterMockedStatic.when(() -> ConfigurationConverter.getConfiguration(any(Properties.class))).
              thenCallRealMethod();

      when(conf.getTimeVar(HiveConf.ConfVars.REPL_EXTERNAL_CLIENT_CONNECT_TIMEOUT,
              TimeUnit.MILLISECONDS)).thenReturn(20L);
      when(conf.getTimeVar(HiveConf.ConfVars.REPL_ATLAS_CLIENT_READ_TIMEOUT, TimeUnit.MILLISECONDS)).thenReturn(500L);
      AtlasRestClientBuilder atlasRestCleintBuilder = new AtlasRestClientBuilder("http://localhost:31000");
      AtlasRestClient atlasClient = atlasRestCleintBuilder.getClient(conf);
      Assert.assertNotNull(atlasClient);
      ArgumentCaptor<Properties> propsCaptor = ArgumentCaptor.forClass(Properties.class);
      configurationConverterMockedStatic.verify(() -> ConfigurationConverter.getConfiguration(propsCaptor.capture()));
      Assert.assertEquals("20", propsCaptor.getValue().getProperty(
              AtlasRestClientBuilder.ATLAS_PROPERTY_CONNECT_TIMEOUT_IN_MS));
      Assert.assertEquals("500", propsCaptor.getValue().getProperty(
              AtlasRestClientBuilder.ATLAS_PROPERTY_READ_TIMEOUT_IN_MS));
    }
  }
Unit test class for testing Atlas metadata Dump.
ql/src/test/org/apache/hadoop/hive/ql/exec/repl/TestAtlasDumpTask.java