@Test
  public void testAtlasServerEntity() throws AtlasServiceException, SemanticException {
    AtlasClientV2 atlasClientV2 = mock(AtlasClientV2.class);
    AtlasServer atlasServer = mock(AtlasServer.class);
    when(atlasClientV2.getServer(Mockito.anyString())).thenReturn(atlasServer);
    AtlasRestClient atlasClient = new AtlasRestClientImpl(atlasClientV2, conf);
    AtlasServer atlasServerRet = atlasClient.getServer("src", conf);
    Assert.assertTrue(atlasServer == atlasServerRet);
  }
Unit test class for testing Atlas metadata Dump.
ql/src/test/org/apache/hadoop/hive/ql/exec/repl/TestAtlasDumpTask.java