luciano/amazon-kinesis-client
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Set the region for test runs (#488)

Browse source 
The AWS SDK depends on being able to find which region it should
operate in.  In many cases this is either explicitly set or retrieved
from a variety of sources.  For these test cases we don't actually
care what the region. To ensure that the tests operate as expected the
region is set before the test runs, and reset upon completion.
This commit is contained in:
Justin Pfifer 2019-01-29 18:53:54 -08:00 committed by Sahil Palvia
parent d79691cb3f
commit a15713911c
1 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
amazon-kinesis-client-multilang/src/test/java/software/amazon/kinesis/multilang/config/MultiLangDaemonConfigurationTest.java
View file

@ -1,5 +1,5 @@
/*
* Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
* Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Amazon Software License (the "License").
* You may not use this file except in compliance with the License.
@ -21,6 +21,7 @@ import static org.junit.Assert.assertThat;
import org.apache.commons.beanutils.BeanUtilsBean;
import org.apache.commons.beanutils.ConvertUtilsBean;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
@ -37,8 +38,11 @@ import software.amazon.kinesis.retrieval.polling.PollingConfig;
@RunWith(MockitoJUnitRunner.class)
public class MultiLangDaemonConfigurationTest {
private static final String AWS_REGION_PROPERTY_NAME = "aws.region";
private BeanUtilsBean utilsBean;
private ConvertUtilsBean convertUtilsBean;
private String originalRegionValue;
@Mock
private ShardRecordProcessorFactory shardRecordProcessorFactory;
@ -48,10 +52,22 @@ public class MultiLangDaemonConfigurationTest {
@Before
public void setup() {
originalRegionValue = System.getProperty(AWS_REGION_PROPERTY_NAME);
System.setProperty(AWS_REGION_PROPERTY_NAME, "us-east-1");
convertUtilsBean = new ConvertUtilsBean();
utilsBean = new BeanUtilsBean(convertUtilsBean);
}
@After
public void after() {
if (originalRegionValue != null) {
System.setProperty(AWS_REGION_PROPERTY_NAME, originalRegionValue);
} else {
System.clearProperty(AWS_REGION_PROPERTY_NAME);
}
}
public MultiLangDaemonConfiguration baseConfiguration() {
MultiLangDaemonConfiguration configuration = new MultiLangDaemonConfiguration(utilsBean, convertUtilsBean);
configuration.setApplicationName("Test");