java类org.testng.annotations.Optional的实例源码

@Test
@Parameters({"credentialName", "region", "availabilityZone", "poolName"})
public void testIpPoolSelection(@Optional("") String credentialName, @Optional("") String region, @Optional("") String availabilityZone,
        String poolName) {
    // GIVEN
    IntegrationTestContext itContext = getItContext();
    credentialName = StringUtils.hasText(credentialName) ? credentialName : itContext.getContextParam(CloudbreakV2Constants.CREDENTIAL_NAME);
    region = StringUtils.hasText(region) ? region : itContext.getContextParam(CloudbreakV2Constants.REGION);
    availabilityZone = StringUtils.hasText(availabilityZone) ? availabilityZone : itContext.getContextParam(CloudbreakV2Constants.AVAILABILTYZONE);
    PlatformResourceRequestJson resourceRequestJson = new PlatformResourceRequestJson();
    resourceRequestJson.setCredentialName(credentialName);
    resourceRequestJson.setRegion(region);
    resourceRequestJson.setAvailabilityZone(availabilityZone);
    // WHEN
    PlatformIpPoolsResponse response = getCloudbreakClient().connectorV1Endpoint().getIpPoolsCredentialId(resourceRequestJson);
    // THEN
    Set<IpPoolJson> ipPools = response.getIppools().get(availabilityZone);
    Assert.assertNotNull(ipPools, "ippools cannot be null for " + region);
    java.util.Optional<IpPoolJson> selected = ipPools.stream().filter(rk -> rk.getName().equals(poolName)).findFirst();
    Assert.assertTrue(selected.isPresent(), "the ippool list doesn't contain [" + poolName + "]");
    getItContext().putContextParam(CloudbreakV2Constants.OPENSTACK_FLOATING_POOL, selected.get().getId());
}

@Test
@Parameters({ "name", "ports", "provider" })
public void testSecurityGroupCreation(@Optional("it-restricted-ambari") String name, @Optional("22,443,9443,8080") String ports,
        @Optional("MOCK") String provider) throws Exception {
    // GIVEN
    // WHEN
    SecurityGroupRequest securityGroupRequest = new SecurityGroupRequest();
    securityGroupRequest.setDescription("Security group created by IT");
    securityGroupRequest.setName(name);
    SecurityRuleRequest securityRuleRequest = new SecurityRuleRequest("");
    securityRuleRequest.setProtocol("tcp");
    securityRuleRequest.setSubnet("0.0.0.0/0");
    securityRuleRequest.setPorts(ports);
    securityGroupRequest.setSecurityRules(Collections.singletonList(securityRuleRequest));
    securityGroupRequest.setCloudPlatform(provider);

    String id = getCloudbreakClient().securityGroupEndpoint().postPrivate(securityGroupRequest).getId().toString();
    // THEN
    Assert.assertNotNull(id);
    getItContext().putContextParam(CloudbreakITContextConstants.SECURITY_GROUP_ID, id, true);
}

@Test
@Parameters({ "credentialName", "projectId", "serviceAccountId", "serviceAccountPrivateKeyP12File" })
public void testGCPCredentialCreation(@Optional("")String credentialName, @Optional("")String projectId, @Optional("")String serviceAccountId,
        @Optional("")String serviceAccountPrivateKeyP12File) throws Exception {
    // GIVEN
    credentialName = StringUtils.hasLength(credentialName) ? credentialName : defaultName;
    projectId = StringUtils.hasLength(projectId) ? projectId : defaultProjectId;
    serviceAccountId = StringUtils.hasLength(serviceAccountId) ? serviceAccountId : defaultServiceAccountId;
    serviceAccountPrivateKeyP12File = StringUtils.hasLength(serviceAccountPrivateKeyP12File) ? serviceAccountPrivateKeyP12File : defaultP12File;
    String serviceAccountPrivateKey = ResourceUtil.readBase64EncodedContentFromResource(applicationContext, serviceAccountPrivateKeyP12File);
    CredentialRequest credentialRequest = new CredentialRequest();
    credentialRequest.setCloudPlatform("GCP");
    credentialRequest.setDescription("GCP credential for integartiontest");
    credentialRequest.setName(credentialName);
    Map<String, Object> map = new HashMap<>();
    map.put("projectId", projectId);
    map.put("serviceAccountId", serviceAccountId);
    map.put("serviceAccountPrivateKey", serviceAccountPrivateKey);
    credentialRequest.setParameters(map);
    // WHEN
    String id = getCloudbreakClient().credentialEndpoint().postPrivate(credentialRequest).getId().toString();
    // THEN
    Assert.assertNotNull(id);
    getItContext().putContextParam(CloudbreakITContextConstants.CREDENTIAL_ID, id, true);
}

@Test
@Parameters({ "azureTemplateName", "azureVmType", "azureVolumeCount", "azureVolumeSize" })
public void testAzureTemplateCreation(@Optional("it-azure-template") String azureTemplateName, @Optional("MEDIUM") String azureVmType,
        @Optional("1") String azureVolumeCount, @Optional("10") String azureVolumeSize) throws Exception {
    // GIVEN
    // WHEN
    TemplateRequest templateRequest = new TemplateRequest();
    templateRequest.setName(azureTemplateName);
    templateRequest.setDescription("AZURE_RM template for integration testing");
    templateRequest.setCloudPlatform("AZURE_RM");
    templateRequest.setInstanceType(azureVmType);
    templateRequest.setVolumeType("Standard_LRS");
    templateRequest.setVolumeCount(Integer.valueOf(azureVolumeCount));
    templateRequest.setVolumeSize(Integer.valueOf(azureVolumeSize));
    String id = getCloudbreakClient().templateEndpoint().postPrivate(templateRequest).getId().toString();
    // THEN
    Assert.assertNotNull(id);
    templateAdditionHelper.handleTemplateAdditions(getItContext(), id, additions);
}

@Test
@Parameters({ "blueprintName", "blueprintFile" })
public void testBlueprintCreation(@Optional("it-hdp-multi-blueprint") String blueprintName,
        @Optional("classpath:/blueprint/hdp-multinode-default.bp") String blueprintFile) throws Exception {
    // GIVEN
    String blueprintContent = ResourceUtil.readStringFromResource(applicationContext, blueprintFile);
    // WHEN
    BlueprintRequest blueprintRequest = new BlueprintRequest();
    blueprintRequest.setName(blueprintName);
    blueprintRequest.setDescription("Blueprint for integration testing");
    blueprintRequest.setAmbariBlueprint(blueprintContent);
    String id = getCloudbreakClient().blueprintEndpoint().postPrivate(blueprintRequest).getId().toString();
    // THEN
    Assert.assertNotNull(id);
    getItContext().putContextParam(CloudbreakITContextConstants.BLUEPRINT_ID, id, true);
}

@Test
@Parameters({ "azureTemplateName", "azureVmType", "azureVolumeCount", "azureVolumeSize" })
public void testAzureTemplateCreation(@Optional("it-azure-template") String azureTemplateName, @Optional("MEDIUM") String azureVmType,
        @Optional("1") String azureVolumeCount, @Optional("10") String azureVolumeSize) throws Exception {
    // GIVEN
    // WHEN
    // TODO publicInAccount
    TemplateRequest templateRequest = new TemplateRequest();
    templateRequest.setName(azureTemplateName);
    templateRequest.setDescription("AZURE template for integration testing");
    templateRequest.setCloudPlatform("AZURE");
    templateRequest.setInstanceType(azureVmType);
    templateRequest.setVolumeType("Standard_LRS");
    templateRequest.setVolumeCount(Integer.valueOf(azureVolumeCount));
    templateRequest.setVolumeSize(Integer.valueOf(azureVolumeSize));
    String id = getCloudbreakClient().templateEndpoint().postPrivate(templateRequest).getId().toString();
    // THEN
    Assert.assertNotNull(id);
    templateAdditionHelper.handleTemplateAdditions(getItContext(), id, additions);
}