def test_set_regional_configs(patched_config, default_bootstrapper: Bootstrapper) -> None: default_bootstrapper.get_regional_configs = MagicMock( return_value=[patched_config]) default_bootstrapper.set_regional_configs() default_bootstrapper.get_regional_configs.assert_called_once() patched_config.write_config.assert_called_once()
def test_get_regional_configs_gov(patched_getenv, default_bootstrapper: Bootstrapper) -> None: patched_getenv.return_value = "us-gov-west-1" regional_configs_list = default_bootstrapper.get_regional_configs() assert len(regional_configs_list) == 1 assert regional_configs_list[0] == Configuration( Classification="core-site", Properties={"fs.s3a.endpoint": "s3.us-gov-west-1.amazonaws.com"} ) patched_getenv.assert_called_once_with("AWS_REGION")
def test_get_regional_configs_missing_region( patched_getenv, default_bootstrapper: Bootstrapper) -> None: patched_getenv.return_value = None regional_configs_list = default_bootstrapper.get_regional_configs() assert len(regional_configs_list) == 0 patched_getenv.assert_called_once_with("AWS_REGION")