org.springframework.cloud.service.common.PostgresqlServiceInfo Java Examples

@Test
public void postgresqlServiceCreationWithJdbcUrl() {
	String name1 = "database-1";
	String name2 = "database-2";
	when(mockEnvironment.getEnvValue("VCAP_SERVICES"))
			.thenReturn(getServicesPayload(
					getPostgresqlServicePayloadWithJdbcUrl("postgresql-1", hostname, port, username, password, name1),
					getPostgresqlServicePayloadWithJdbcUrl("postgresql-2", hostname, port, username, password, name2)));

	List<ServiceInfo> serviceInfos = testCloudConnector.getServiceInfos();
	ServiceInfo info1 = getServiceInfo(serviceInfos, "postgresql-1");
	ServiceInfo info2 = getServiceInfo(serviceInfos, "postgresql-2");

	assertServiceFoundOfType(info1, PostgresqlServiceInfo.class);
	assertServiceFoundOfType(info2, PostgresqlServiceInfo.class);

	assertJdbcUrlEqual(info1, POSTGRES_SCHEME, name1);
	assertJdbcUrlEqual(info2, POSTGRES_SCHEME, name2);

	assertUriBasedServiceInfoFields(info1, POSTGRES_SCHEME, hostname, port, username, password, name1);
	assertUriBasedServiceInfoFields(info2, POSTGRES_SCHEME, hostname, port, username, password, name2);
}
 

@Test
public void postgresqlServiceCreationNoLabelNoTags() {
	String name1 = "database-1";
	String name2 = "database-2";
	when(mockEnvironment.getEnvValue("VCAP_SERVICES"))
			.thenReturn(getServicesPayload(
					getPostgresqlServicePayloadNoLabelNoTags("postgresql-1", hostname, port, username, password, name1),
					getPostgresqlServicePayloadNoLabelNoTags("postgresql-2", hostname, port, username, password, name2)));

	List<ServiceInfo> serviceInfos = testCloudConnector.getServiceInfos();
	ServiceInfo info1 = getServiceInfo(serviceInfos, "postgresql-1");
	ServiceInfo info2 = getServiceInfo(serviceInfos, "postgresql-2");

	assertServiceFoundOfType(info1, PostgresqlServiceInfo.class);
	assertServiceFoundOfType(info2, PostgresqlServiceInfo.class);

	assertJdbcUrlEqual(info1, POSTGRES_SCHEME, name1);
	assertJdbcUrlEqual(info2, POSTGRES_SCHEME, name2);

	assertUriBasedServiceInfoFields(info1, POSTGRES_SCHEME, hostname, port, username, password, name1);
	assertUriBasedServiceInfoFields(info2, POSTGRES_SCHEME, hostname, port, username, password, name2);
}
 

@Test
public void postgresqlWithSpecialCharsServiceCreation() {
	String userWithSpecialChars = "u%u:u+";
	String passwordWithSpecialChars = "p%p:p+";

	String name1 = "database-1";
	when(mockEnvironment.getEnvValue("VCAP_SERVICES")).thenReturn(
			getServicesPayload(getPostgresqlServicePayload("postgresql-1", hostname,
					port, userWithSpecialChars, passwordWithSpecialChars, name1)));

	List<ServiceInfo> serviceInfos = testCloudConnector.getServiceInfos();
	ServiceInfo info1 = getServiceInfo(serviceInfos, "postgresql-1");

	assertServiceFoundOfType(info1, PostgresqlServiceInfo.class);

	assertEquals(getJdbcUrl(hostname, port, userWithSpecialChars,
			passwordWithSpecialChars, name1),
			((RelationalServiceInfo) info1).getJdbcUrl());

	assertUriBasedServiceInfoFields(info1, POSTGRES_SCHEME, hostname, port,
			userWithSpecialChars, passwordWithSpecialChars, name1);
}
 

@Test
public void postgresqlServiceCreation() {
	String name1 = "database-1";
	String name2 = "database-2";
	when(mockEnvironment.getEnvValue("VCAP_SERVICES"))
			.thenReturn(getServicesPayload(
					getPostgresqlServicePayload("postgresql-1", hostname, port, username, password, name1),
					getPostgresqlServicePayload("postgresql-2", hostname, port, username, password, name2)));

	List<ServiceInfo> serviceInfos = testCloudConnector.getServiceInfos();
	ServiceInfo info1 = getServiceInfo(serviceInfos, "postgresql-1");
	ServiceInfo info2 = getServiceInfo(serviceInfos, "postgresql-2");

	assertServiceFoundOfType(info1, PostgresqlServiceInfo.class);
	assertServiceFoundOfType(info2, PostgresqlServiceInfo.class);

	assertJdbcUrlEqual(info1, POSTGRES_SCHEME, name1);
	assertJdbcUrlEqual(info2, POSTGRES_SCHEME, name2);

	assertUriBasedServiceInfoFields(info1, POSTGRES_SCHEME, hostname, port, username, password, name1);
	assertUriBasedServiceInfoFields(info2, POSTGRES_SCHEME, hostname, port, username, password, name2);
}
 

private void assertPostgresServiceCreated(String envVarName, String serviceInstanceName) {
	Map<String, String> env = new HashMap<String, String>();
	String postgresUrl = getRelationalServiceUrl("db");
	env.put(envVarName, postgresUrl);
	when(mockEnvironment.getEnv()).thenReturn(env);

	List<ServiceInfo> serviceInfos = testCloudConnector.getServiceInfos();
	ServiceInfo serviceInfo = getServiceInfo(serviceInfos, serviceInstanceName);
	assertNotNull(serviceInfo);
	assertTrue(serviceInfo instanceof PostgresqlServiceInfo);
	assertReleationServiceInfo((PostgresqlServiceInfo) serviceInfo, "db");
}
 

@Override
public PostgresqlServiceInfo createServiceInfo(Map<String, Object> serviceData) {
    String id = getId(serviceData);

    Map<String, Object> credentials = getCredentials(serviceData);
    String uri = getUriFromCredentials(credentials);

    String cert64 = getRootCaFromCredentials(credentials);
    if (cert64 != null && uri.contains("sslmode")) {
        if (!uri.contains("sslfactory")) {
            try {
                StringBasedTrustManager.getTrustManager().addCert(Base64.getDecoder().decode(cert64));
                if (!uri.contains("?")) {
                    uri += "?sslfactory=" + StringBasedSSLSocketFactory.class.getCanonicalName();
                } else {
                    uri += "&sslfactory=" + StringBasedSSLSocketFactory.class.getCanonicalName();
                }
            } catch (Exception e) {
                LOG.log(Level.SEVERE, "Unable to add certificate to trust manager", e);
            }
        } else {
            LOG.log(Level.WARNING,
                    "sslfactory already present in postgresql, not adding custom handler. Expect truststore issues.");
        }
    }

    return new PostgresqlServiceInfo(id, uri);
}
 

@Test
public void serviceCreation() {
	List<ServiceInfo> services = connector.getServiceInfos();
	ServiceInfo service = getServiceInfo(services, "ingres");
	assertNotNull(service);
	assertTrue(service instanceof PostgresqlServiceInfo);
	assertUriParameters((PostgresqlServiceInfo) service);
}
 

@Override
public PostgresqlServiceInfo createServiceInfo(String id, String url, String jdbcUrl) {
	return new PostgresqlServiceInfo(id, url, jdbcUrl);
}
 

@Override
public PostgresqlServiceInfo createService(String id) {
	return createPostgresqlService(id);
}
 

@Override
public PostgresqlServiceInfo createServiceInfo() {
	when(mockPostgresqlServiceInfo.getJdbcUrl()).thenReturn("jdbc:postgresql://myuser:[email protected]:5432/database-123");
	
	return mockPostgresqlServiceInfo;
}
 

public PostgresqlServiceInfoCreator() {
	super(PostgresqlServiceInfo.POSTGRES_SCHEME);
}
 

@Override
public PostgresqlServiceInfo createServiceInfo(String id, String uri) {
	return new PostgresqlServiceInfo(id, uri);
}
 

public PostgresqlServiceInfo getTestServiceInfo(String id) {
	return new PostgresqlServiceInfo(id, "jdbc:postgres://username:pass@host:port/db");
}
 

public PostgresqlServiceInfoCreator() {
	super(PostgresqlServiceInfo.POSTGRES_SCHEME);
}
 

@Override
public PostgresqlServiceInfo createServiceInfo(String id, String uri) {
	return new PostgresqlServiceInfo(HerokuUtil.computeServiceName(id), uri);
}
 

public HerokuConnectorPostgresqlServiceTest() {
	super(PostgresqlServiceInfo.POSTGRES_SCHEME);
}
 

protected PostgresqlServiceInfo createPostgresqlService(String id) {
	return new PostgresqlServiceInfo(id, "postgres://username:pass@host:port/db");
}