@Override public Cluster getEmrClusterById(String clusterId, AwsParamsDto awsParams) { Cluster cluster = null; if (StringUtils.isNotBlank(clusterId)) { DescribeClusterResult describeClusterResult = emrOperations.describeClusterRequest(getEmrClient(awsParams), new DescribeClusterRequest().withClusterId(clusterId)); if (describeClusterResult != null && describeClusterResult.getCluster() != null) { cluster = describeClusterResult.getCluster(); } } return cluster; }
@Override public Cluster getEmrClusterById(String clusterId, AwsParamsDto awsParams) { Cluster cluster = null; if (StringUtils.isNotBlank(clusterId)) { DescribeClusterResult describeClusterResult = emrOperations.describeClusterRequest(getEmrClient(awsParams), new DescribeClusterRequest().withClusterId(clusterId)); if (describeClusterResult != null && describeClusterResult.getCluster() != null) { cluster = describeClusterResult.getCluster(); } } return cluster; }
@Test public void getEmrClusterByIdAssertReturnNullWhenDescribeClusterResponseIsNull() throws Exception { String clusterId = "clusterId"; when(mockEmrOperations.describeClusterRequest(any(), any())).thenReturn(null); assertNull(emrDao.getEmrClusterById(clusterId, getAwsParamsDto())); }
@Test public void getEmrClusterByIdAssertReturnNullWhenDescribeClusterResponseClusterIsNull() throws Exception { String clusterId = "clusterId"; when(mockEmrOperations.describeClusterRequest(any(), any())).thenReturn(new DescribeClusterResult()); assertNull(emrDao.getEmrClusterById(clusterId, getAwsParamsDto())); }
@Test public void getEmrClusterStatusByIdAssertReturnNullWhenClusterIsNull() throws Exception { String clusterId = "clusterId"; when(mockEmrOperations.describeClusterRequest(any(), any())).then(new Answer<DescribeClusterResult>() { @Override public DescribeClusterResult answer(InvocationOnMock invocation) throws Throwable { DescribeClusterRequest describeClusterRequest = invocation.getArgument(1); assertEquals(clusterId, describeClusterRequest.getClusterId()); return new DescribeClusterResult(); } }); assertNull(emrDao.getEmrClusterStatusById(clusterId, getAwsParamsDto())); }
@Test public void getEmrClusterByIdAssertCallDescribeCluster() throws Exception { String clusterId = "clusterId"; Cluster expectedCluster = new Cluster(); when(mockEmrOperations.describeClusterRequest(any(), any())).thenAnswer(new Answer<DescribeClusterResult>() { @Override public DescribeClusterResult answer(InvocationOnMock invocation) throws Throwable { DescribeClusterRequest describeClusterRequest = invocation.getArgument(1); assertEquals(clusterId, describeClusterRequest.getClusterId()); DescribeClusterResult describeClusterResult = new DescribeClusterResult(); describeClusterResult.setCluster(expectedCluster); return describeClusterResult; } }); assertEquals(expectedCluster, emrDao.getEmrClusterById(clusterId, getAwsParamsDto())); }
@Test public void getEmrClusterStatusByIdAssertReturnClusterState() throws Exception { String clusterId = "clusterId"; ClusterState expectedState = ClusterState.BOOTSTRAPPING; when(mockEmrOperations.describeClusterRequest(any(), any())).then(new Answer<DescribeClusterResult>() { @Override public DescribeClusterResult answer(InvocationOnMock invocation) throws Throwable { DescribeClusterRequest describeClusterRequest = invocation.getArgument(1); assertEquals(clusterId, describeClusterRequest.getClusterId()); DescribeClusterResult describeClusterResult = new DescribeClusterResult(); Cluster cluster = new Cluster(); ClusterStatus status = new ClusterStatus(); status.setState(expectedState); cluster.setStatus(status); describeClusterResult.setCluster(cluster); return describeClusterResult; } }); assertEquals(expectedState.toString(), emrDao.getEmrClusterStatusById(clusterId, getAwsParamsDto())); }