Java Code Examples for org.springframework.boot.test.util.EnvironmentTestUtils#addEnvironment()

@Before
public void setUp() {
	context = new AnnotationConfigApplicationContext();
	EnvironmentTestUtils.addEnvironment(context, 
			"mail.smtp.host="+ServerSetupTest.SMTP.getBindAddress(), 
			"mail.smtp.port="+ServerSetupTest.SMTP.getPort(),
			"ogham.sms.smpp.host=127.0.0.1",
			"ogham.sms.smpp.port="+smppServer.getPort(),
			"spring.freemarker.suffix=");
	context.register(TestConfig.class, ThymeleafAutoConfiguration.class, FreeMarkerAutoConfiguration.class, OghamSpringBoot1AutoConfiguration.class);
	context.refresh();
	messagingService = context.getBean(MessagingService.class);
}
 

private void load(boolean refresh, String... environment) {
    AnnotationConfigApplicationContext ctx = new AnnotationConfigApplicationContext();
    ctx.register(RocketMqAutoConfiguration.class);
    EnvironmentTestUtils.addEnvironment(ctx, environment);
    if (refresh) {
        ctx.refresh();
    }
    this.context = ctx;
}
 

@Test
public void testFooCreatePropertyTrue() {
  EnvironmentTestUtils.addEnvironment(context, "foo.create=true");
  context.register(FooConfiguration.class);
  context.refresh();
  assertNotNull(context.getBean(Foo.class));
}
 

private void prepareApplicationContextMissingParent() {
    this.context = new AnnotationConfigApplicationContext();
    EnvironmentTestUtils.addEnvironment(this.context, "spring.rocketmq.name-server-address:127.0.0.1:9876");
    this.context.register(TestConsumerMissingParent.class);
    this.context.register(MQConsumerAutoConfiguration.class);
    this.context.refresh();
}
 

private void prepareApplicationContextCMError() {
    this.context = new AnnotationConfigApplicationContext();
    EnvironmentTestUtils.addEnvironment(this.context, "spring.rocketmq.name-server-address:127.0.0.1:9876");
    this.context.register(TestConsumerErrorCM.class);
    this.context.register(MQConsumerAutoConfiguration.class);
    this.context.refresh();
}
 

private void prepareApplicationContextCMOrderly() {
    this.context = new AnnotationConfigApplicationContext();
    EnvironmentTestUtils.addEnvironment(this.context, "spring.rocketmq.name-server-address:127.0.0.1:9876");
    this.context.register(TestConsumerOrderly.class);
    this.context.register(MQConsumerAutoConfiguration.class);
    this.context.refresh();
}
 

private void prepareApplicationContext() {
    this.context = new AnnotationConfigApplicationContext();
    EnvironmentTestUtils.addEnvironment(this.context, "spring.rocketmq.name-server-address:127.0.0.1:9876");
    this.context.register(TestConsumer.class);
    this.context.register(MQConsumerAutoConfiguration.class);
    this.context.refresh();
}
 

@Before
public void setUp() {
	context = new AnnotationConfigApplicationContext();
	EnvironmentTestUtils.addEnvironment(context, 
			"mail.smtp.host="+ServerSetupTest.SMTP.getBindAddress(), 
			"mail.smtp.port="+ServerSetupTest.SMTP.getPort(),
			"ogham.sms.smpp.host=127.0.0.1",
			"ogham.sms.smpp.port="+smppServer.getPort(),
			"spring.sendgrid.api-key=spring",
			"ogham.freemarker.default-encoding="+StandardCharsets.US_ASCII.name(),
			"spring.freemarker.charset="+StandardCharsets.UTF_16BE.name());
}
 

private void load(Class<?> config, String... environment) {
    AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext();
    EnvironmentTestUtils.addEnvironment(applicationContext, environment);
    applicationContext.register(config);
    applicationContext.register(AmazonAutoConfiguration.class);
    applicationContext.refresh();
    this.context = applicationContext;
}
 

@Test
public void saveOutputInHeaderCanBeCustomized() {
	EnvironmentTestUtils.addEnvironment(context, "tensorflow.saveOutputInHeader:false");
	context.register(Conf.class);
	context.refresh();
	TensorflowProcessorProperties properties = context.getBean(TensorflowProcessorProperties.class);
	assertFalse(properties.isSaveOutputInHeader());
}
 

@Test
public void outputIndexCanBeCustomized() {
	EnvironmentTestUtils.addEnvironment(context, "tensorflow.outputIndex:666");
	context.register(Conf.class);
	context.refresh();
	TensorflowProcessorProperties properties = context.getBean(TensorflowProcessorProperties.class);
	assertThat(properties.getOutputIndex(), equalTo(666));
}
 

private void load(Class<?> config, String... environment) {
    AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext();
    EnvironmentTestUtils.addEnvironment(applicationContext, environment);
    applicationContext.register(config);
    applicationContext.register(GitHubAutoConfiguration.class);
    applicationContext.refresh();
    this.context = applicationContext;
}
 

@Test
public void oghamPropertiesWithSpringPropsShouldUseOghamPropertiesPrecedence() throws Exception {
	EnvironmentTestUtils.addEnvironment(context, "ogham.email.javamail.host=ogham", "spring.mail.host=spring");
	context.register(ManuallyEnableSpringPropertiesConfig.class, OghamSpringBoot1AutoConfiguration.class);
	context.refresh();
	MessagingService messagingService = context.getBean(MessagingService.class);
	OghamInternalAssertions.assertThat(messagingService)
		.javaMail()
			.host(equalTo("ogham"));
}
 

@Test
public void alternativesLengthCanBeCustomized() {
	AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext();
	EnvironmentTestUtils.addEnvironment(context, "inception.labelsLocation:/remote");
	EnvironmentTestUtils.addEnvironment(context, "inception.alternativesLength:5");
	context.register(Conf.class);
	context.refresh();
	LabelImageProcessorProperties properties = context.getBean(LabelImageProcessorProperties.class);
	assertThat(properties.getAlternativesLength(), equalTo(5));
}
 

@Test
public void useCustomSendGridBean() throws Exception {
	EnvironmentTestUtils.addEnvironment(context, "ogham.email.sendgrid.api-key=ogham");
	context.register(CustomSendGridConfig.class, SendGridAutoConfiguration.class, OghamSpringBoot1AutoConfiguration.class);
	context.refresh();
	
	MessagingService messagingService = context.getBean(MessagingService.class);
	
	checkEmail(messagingService);
	checkSms(messagingService);
	OghamInternalAssertions.assertThat(messagingService)
		.sendGrid()
			.apiKey(nullValue(String.class))
			.client(allOf(isA(CustomSendGrid.class), isSpringBeanInstance(context, SendGrid.class)));
}
 

@Test
public void oghamWithJavaMailAutoConfigShouldUseSpringProperties() throws Exception {
	EnvironmentTestUtils.addEnvironment(context, "spring.mail.host=spring");
	context.register(MailSenderAutoConfiguration.class, OghamSpringBoot1AutoConfiguration.class);
	context.refresh();
	MessagingService messagingService = context.getBean(MessagingService.class);
	OghamInternalAssertions.assertThat(messagingService)
		.javaMail()
			.host(equalTo("spring"));
}
 

/**
 * Process the {@link SpringletsWebJacksonAutoConfiguration} class by the
 * {@link ApplicationContext}. After the processing the {@link BindingResultModule}
 * must be in the ApplicationContext.
 * 
 * @see EnvironmentTestUtils#addEnvironment(String, org.springframework.core.env.ConfigurableEnvironment, String...)
 */
private void registerAndRefreshContext(String... env) {
  EnvironmentTestUtils.addEnvironment(this.context, env);
  this.context.register(SpringletsWebJacksonAutoConfiguration.class);
  this.context.refresh();
}
 

/**
 * Process the {@link SpringletsWebMvcAutoConfiguration} class by the
 * {@link ApplicationContext}. After the processing the {@link SpringletsImageFileConverter}
 * must be in the ApplicationContext.
 *
 * Additionally add (high priority) values to the {@link Environment} owned by the
 * {@link ApplicationContext}.
 *
 * @param env Strings following the pattern "property-name:property-value"
 * @see EnvironmentTestUtils#addEnvironment(String, org.springframework.core.env.ConfigurableEnvironment, String...)
 */
private void registerAndRefreshContext(String... env) {
  EnvironmentTestUtils.addEnvironment(this.context, env);
  this.context.register(SpringletsWebMvcAutoConfiguration.class);
  this.context.refresh();
}
 

/**
 * Process the {@link SpringletsWebMvcAutoConfiguration} class by the
 * {@link ApplicationContext}. After the processing the {@link StringTrimmerAdvice}
 * must be in the ApplicationContext.
 *
 * Additionally add (high priority) values to the {@link Environment} owned by the
 * {@link ApplicationContext}.
 *
 * @param env Strings following the pattern "property-name:property-value"
 * @see EnvironmentTestUtils#addEnvironment(ConfigurableEnvironment, String...)
 */
private void registerAndRefreshContext(String... env) {
  EnvironmentTestUtils.addEnvironment(this.context, env);
  this.context.register(SpringletsWebMvcAutoConfiguration.class);
  this.context.refresh();
}
 

/**
 * Process the {@link SpringletsWebMvcAutoConfiguration} class by the
 * {@link ApplicationContext}. After the processing the {@link JsonpAdvice}
 * must be in the ApplicationContext.
 *
 * Additionally add (high priority) values to the {@link Environment} owned by the
 * {@link ApplicationContext}.
 *
 * @param env Strings following the pattern "property-name:property-value"
 * @see EnvironmentTestUtils#addEnvironment(String, org.springframework.core.env.ConfigurableEnvironment, String...)
 */
private void registerAndRefreshContext(String... env) {
  EnvironmentTestUtils.addEnvironment(this.context, env);
  this.context.register(SpringletsWebMvcAutoConfiguration.class);
  this.context.refresh();
}