62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
import boto3
|
|
import mock
|
|
import os
|
|
import pytest
|
|
from moto import mock_dynamodb2, mock_sns, settings
|
|
from unittest import SkipTest
|
|
|
|
|
|
@mock_sns
|
|
def test_use_invalid_region():
|
|
if settings.TEST_SERVER_MODE:
|
|
raise SkipTest("ServerMode will throw different errors")
|
|
client = boto3.client("sns", region_name="any-region")
|
|
with pytest.raises(KeyError) as exc:
|
|
client.list_platform_applications()
|
|
str(exc.value).should.contain("any-region")
|
|
|
|
|
|
@mock_sns
|
|
@mock.patch.dict(os.environ, {"AWS_DEFAULT_REGION": "us-east-2"})
|
|
def test_use_region_from_env():
|
|
client = boto3.client("sns")
|
|
client.list_platform_applications()["PlatformApplications"].should.equal([])
|
|
|
|
|
|
@mock_sns
|
|
@mock.patch.dict(os.environ, {"AWS_DEFAULT_REGION": "any-region"})
|
|
def test_use_unknown_region_from_env():
|
|
if settings.TEST_SERVER_MODE:
|
|
raise SkipTest("Cannot set environemnt variables in ServerMode")
|
|
client = boto3.client("sns")
|
|
with pytest.raises(KeyError) as exc:
|
|
client.list_platform_applications()
|
|
str(exc.value).should.contain("any-region")
|
|
|
|
|
|
@mock_sns
|
|
@mock.patch.dict(os.environ, {"AWS_DEFAULT_REGION": "any-region"})
|
|
@mock.patch.dict(os.environ, {"MOTO_ALLOW_NONEXISTENT_REGION": "trUe"})
|
|
def test_use_unknown_region_from_env_but_allow_it():
|
|
if settings.TEST_SERVER_MODE:
|
|
raise SkipTest("Cannot set environemnt variables in ServerMode")
|
|
client = boto3.client("sns")
|
|
client.list_platform_applications()["PlatformApplications"].should.equal([])
|
|
|
|
|
|
@mock_dynamodb2
|
|
@mock.patch.dict(os.environ, {"MOTO_ALLOW_NONEXISTENT_REGION": "trUe"})
|
|
def test_use_unknown_region_from_env_but_allow_it__dynamo():
|
|
if settings.TEST_SERVER_MODE:
|
|
raise SkipTest("Cannot set environemnt variables in ServerMode")
|
|
dynamo_db = boto3.resource("dynamodb", region_name="test")
|
|
dynamo_db.create_table(
|
|
TableName="test_table",
|
|
KeySchema=[{"AttributeName": "key", "KeyType": "HASH"}],
|
|
AttributeDefinitions=[{"AttributeName": "key", "AttributeType": "S"}],
|
|
BillingMode="PAY_PER_REQUEST",
|
|
)
|
|
tables = list(dynamo_db.tables.all())
|
|
tables.should.have.length_of(1)
|
|
[table.name for table in tables].should.equal(["test_table"])
|