@Bean public DataService dataService() throws Exception { return when(mock(DataService.class).listClusters(ListClustersRequest.builder().build())) .thenReturn( ListClustersResponse.builder() .clusters( Collections.singletonList( Cluster.builder() .accountId(ACCOUNT_ID) .clusterName(CLUSTER_NAME) .build())) .build()) .getMock(); }
@Override public ListClustersResponse listClusters(ListClustersRequest request) { return ListClustersResponse.builder() .clusters( Collections.singletonList( Cluster.builder().accountId(accountId).clusterName(clusterName).build())) .build(); }
@Bean public DataService dataService() throws Exception { return when(mock(DataService.class) .listEnvironments( ListEnvironmentsRequest.builder() .cluster( Cluster.builder().accountId(ACCOUNT_ID).clusterName(CLUSTER_NAME).build()) .build())) .thenReturn( ListEnvironmentsResponse.builder() .environmentIds( Collections.singletonList( EnvironmentId.builder() .accountId(ACCOUNT_ID) .cluster(CLUSTER_NAME) .environmentName(ENVIRONMENT_NAME) .build())) .build()) .getMock(); }
@Test public void testListClusterWithTwoEnvironments() throws Exception { dataService.createEnvironment( models.createEnvironmentRequest().environmentId(createdEnvironmentId1).build()); dataService.createEnvironment( models.createEnvironmentRequest().environmentId(createdEnvironmentId2).build()); final ListClustersResponse listClustersResponse = dataService.listClusters(models.listClustersRequest().clusterNamePrefix(null).build()); assertThat(listClustersResponse.getClusters()) .isEqualTo( Arrays.asList( models.cluster().clusterName(CLUSTER_ONE).build(), models.cluster().clusterName(CLUSTER_TWO).build())); } }
public ClusterBuilder cluster() { return Cluster.builder().accountId(accountId).clusterName(clusterName); } }