catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Run black on moto & test directories.

Browse source
This commit is contained in:
Asher Foa 2019-10-31 08:44:26 -07:00
commit 96e5b1993d
507 changed files with 52541 additions and 47814 deletions
Download patch file Download diff file Expand all files Collapse all files

2195
tests/test_iam/test_iam.py
View file

File diff suppressed because it is too large Load diff

10
tests/test_iam/test_iam_account_aliases.py
View file

@ -5,16 +5,16 @@ from moto import mock_iam
@mock_iam()
def test_account_aliases():
client = boto3.client('iam', region_name='us-east-1')
client = boto3.client("iam", region_name="us-east-1")
alias = 'my-account-name'
alias = "my-account-name"
aliases = client.list_account_aliases()
aliases.should.have.key('AccountAliases').which.should.equal([])
aliases.should.have.key("AccountAliases").which.should.equal([])
client.create_account_alias(AccountAlias=alias)
aliases = client.list_account_aliases()
aliases.should.have.key('AccountAliases').which.should.equal([alias])
aliases.should.have.key("AccountAliases").which.should.equal([alias])
client.delete_account_alias(AccountAlias=alias)
aliases = client.list_account_aliases()
aliases.should.have.key('AccountAliases').which.should.equal([])
aliases.should.have.key("AccountAliases").which.should.equal([])

159
tests/test_iam/test_iam_groups.py
View file

@ -26,46 +26,50 @@ MOCK_POLICY = """
@mock_iam_deprecated()
def test_create_group():
conn = boto.connect_iam()
conn.create_group('my-group')
conn.create_group("my-group")
with assert_raises(BotoServerError):
conn.create_group('my-group')
conn.create_group("my-group")
@mock_iam_deprecated()
def test_get_group():
conn = boto.connect_iam()
conn.create_group('my-group')
conn.get_group('my-group')
conn.create_group("my-group")
conn.get_group("my-group")
with assert_raises(BotoServerError):
conn.get_group('not-group')
conn.get_group("not-group")
@mock_iam()
def test_get_group_current():
conn = boto3.client('iam', region_name='us-east-1')
conn.create_group(GroupName='my-group')
result = conn.get_group(GroupName='my-group')
conn = boto3.client("iam", region_name="us-east-1")
conn.create_group(GroupName="my-group")
result = conn.get_group(GroupName="my-group")
assert result['Group']['Path'] == '/'
assert result['Group']['GroupName'] == 'my-group'
assert isinstance(result['Group']['CreateDate'], datetime)
assert result['Group']['GroupId']
assert result['Group']['Arn'] == 'arn:aws:iam::123456789012:group/my-group'
assert not result['Users']
assert result["Group"]["Path"] == "/"
assert result["Group"]["GroupName"] == "my-group"
assert isinstance(result["Group"]["CreateDate"], datetime)
assert result["Group"]["GroupId"]
assert result["Group"]["Arn"] == "arn:aws:iam::123456789012:group/my-group"
assert not result["Users"]
# Make a group with a different path:
other_group = conn.create_group(GroupName='my-other-group', Path='some/location')
assert other_group['Group']['Path'] == 'some/location'
assert other_group['Group']['Arn'] == 'arn:aws:iam::123456789012:group/some/location/my-other-group'
other_group = conn.create_group(GroupName="my-other-group", Path="some/location")
assert other_group["Group"]["Path"] == "some/location"
assert (
other_group["Group"]["Arn"]
== "arn:aws:iam::123456789012:group/some/location/my-other-group"
)
@mock_iam_deprecated()
def test_get_all_groups():
conn = boto.connect_iam()
conn.create_group('my-group1')
conn.create_group('my-group2')
groups = conn.get_all_groups()['list_groups_response'][
'list_groups_result']['groups']
conn.create_group("my-group1")
conn.create_group("my-group2")
groups = conn.get_all_groups()["list_groups_response"]["list_groups_result"][
"groups"
]
groups.should.have.length_of(2)
@ -73,95 +77,108 @@ def test_get_all_groups():
def test_add_user_to_group():
conn = boto.connect_iam()
with assert_raises(BotoServerError):
conn.add_user_to_group('my-group', 'my-user')
conn.create_group('my-group')
conn.add_user_to_group("my-group", "my-user")
conn.create_group("my-group")
with assert_raises(BotoServerError):
conn.add_user_to_group('my-group', 'my-user')
conn.create_user('my-user')
conn.add_user_to_group('my-group', 'my-user')
conn.add_user_to_group("my-group", "my-user")
conn.create_user("my-user")
conn.add_user_to_group("my-group", "my-user")
@mock_iam_deprecated()
def test_remove_user_from_group():
conn = boto.connect_iam()
with assert_raises(BotoServerError):
conn.remove_user_from_group('my-group', 'my-user')
conn.create_group('my-group')
conn.create_user('my-user')
conn.remove_user_from_group("my-group", "my-user")
conn.create_group("my-group")
conn.create_user("my-user")
with assert_raises(BotoServerError):
conn.remove_user_from_group('my-group', 'my-user')
conn.add_user_to_group('my-group', 'my-user')
conn.remove_user_from_group('my-group', 'my-user')
conn.remove_user_from_group("my-group", "my-user")
conn.add_user_to_group("my-group", "my-user")
conn.remove_user_from_group("my-group", "my-user")
@mock_iam_deprecated()
def test_get_groups_for_user():
conn = boto.connect_iam()
conn.create_group('my-group1')
conn.create_group('my-group2')
conn.create_group('other-group')
conn.create_user('my-user')
conn.add_user_to_group('my-group1', 'my-user')
conn.add_user_to_group('my-group2', 'my-user')
conn.create_group("my-group1")
conn.create_group("my-group2")
conn.create_group("other-group")
conn.create_user("my-user")
conn.add_user_to_group("my-group1", "my-user")
conn.add_user_to_group("my-group2", "my-user")
groups = conn.get_groups_for_user(
'my-user')['list_groups_for_user_response']['list_groups_for_user_result']['groups']
groups = conn.get_groups_for_user("my-user")["list_groups_for_user_response"][
"list_groups_for_user_result"
]["groups"]
groups.should.have.length_of(2)
@mock_iam_deprecated()
def test_put_group_policy():
conn = boto.connect_iam()
conn.create_group('my-group')
conn.put_group_policy('my-group', 'my-policy', MOCK_POLICY)
conn.create_group("my-group")
conn.put_group_policy("my-group", "my-policy", MOCK_POLICY)
@mock_iam
def test_attach_group_policies():
conn = boto3.client('iam', region_name='us-east-1')
conn.create_group(GroupName='my-group')
conn.list_attached_group_policies(GroupName='my-group')['AttachedPolicies'].should.be.empty
policy_arn = 'arn:aws:iam::aws:policy/service-role/AmazonElasticMapReduceforEC2Role'
conn.list_attached_group_policies(GroupName='my-group')['AttachedPolicies'].should.be.empty
conn.attach_group_policy(GroupName='my-group', PolicyArn=policy_arn)
conn.list_attached_group_policies(GroupName='my-group')['AttachedPolicies'].should.equal(
[
{
'PolicyName': 'AmazonElasticMapReduceforEC2Role',
'PolicyArn': policy_arn,
}
])
conn = boto3.client("iam", region_name="us-east-1")
conn.create_group(GroupName="my-group")
conn.list_attached_group_policies(GroupName="my-group")[
"AttachedPolicies"
].should.be.empty
policy_arn = "arn:aws:iam::aws:policy/service-role/AmazonElasticMapReduceforEC2Role"
conn.list_attached_group_policies(GroupName="my-group")[
"AttachedPolicies"
].should.be.empty
conn.attach_group_policy(GroupName="my-group", PolicyArn=policy_arn)
conn.list_attached_group_policies(GroupName="my-group")[
"AttachedPolicies"
].should.equal(
[{"PolicyName": "AmazonElasticMapReduceforEC2Role", "PolicyArn": policy_arn}]
)
conn.detach_group_policy(GroupName='my-group', PolicyArn=policy_arn)
conn.list_attached_group_policies(GroupName='my-group')['AttachedPolicies'].should.be.empty
conn.detach_group_policy(GroupName="my-group", PolicyArn=policy_arn)
conn.list_attached_group_policies(GroupName="my-group")[
"AttachedPolicies"
].should.be.empty
@mock_iam_deprecated()
def test_get_group_policy():
conn = boto.connect_iam()
conn.create_group('my-group')
conn.create_group("my-group")
with assert_raises(BotoServerError):
conn.get_group_policy('my-group', 'my-policy')
conn.get_group_policy("my-group", "my-policy")
conn.put_group_policy('my-group', 'my-policy', MOCK_POLICY)
conn.get_group_policy('my-group', 'my-policy')
conn.put_group_policy("my-group", "my-policy", MOCK_POLICY)
conn.get_group_policy("my-group", "my-policy")
@mock_iam_deprecated()
def test_get_all_group_policies():
conn = boto.connect_iam()
conn.create_group('my-group')
policies = conn.get_all_group_policies('my-group')['list_group_policies_response']['list_group_policies_result']['policy_names']
conn.create_group("my-group")
policies = conn.get_all_group_policies("my-group")["list_group_policies_response"][
"list_group_policies_result"
]["policy_names"]
assert policies == []
conn.put_group_policy('my-group', 'my-policy', MOCK_POLICY)
policies = conn.get_all_group_policies('my-group')['list_group_policies_response']['list_group_policies_result']['policy_names']
assert policies == ['my-policy']
conn.put_group_policy("my-group", "my-policy", MOCK_POLICY)
policies = conn.get_all_group_policies("my-group")["list_group_policies_response"][
"list_group_policies_result"
]["policy_names"]
assert policies == ["my-policy"]
@mock_iam()
def test_list_group_policies():
conn = boto3.client('iam', region_name='us-east-1')
conn.create_group(GroupName='my-group')
conn.list_group_policies(GroupName='my-group')['PolicyNames'].should.be.empty
conn.put_group_policy(GroupName='my-group', PolicyName='my-policy', PolicyDocument=MOCK_POLICY)
conn.list_group_policies(GroupName='my-group')['PolicyNames'].should.equal(['my-policy'])
conn = boto3.client("iam", region_name="us-east-1")
conn.create_group(GroupName="my-group")
conn.list_group_policies(GroupName="my-group")["PolicyNames"].should.be.empty
conn.put_group_policy(
GroupName="my-group", PolicyName="my-policy", PolicyDocument=MOCK_POLICY
)
conn.list_group_policies(GroupName="my-group")["PolicyNames"].should.equal(
["my-policy"]
)

1674
tests/test_iam/test_iam_policies.py
View file

File diff suppressed because it is too large Load diff

7
tests/test_iam/test_server.py
View file

@ -7,9 +7,9 @@ import sure # noqa
import moto.server as server
'''
"""
Test the different server responses
'''
"""
def test_iam_server_get():
@ -17,7 +17,8 @@ def test_iam_server_get():
test_client = backend.test_client()
group_data = test_client.action_data(
"CreateGroup", GroupName="test group", Path="/")
"CreateGroup", GroupName="test group", Path="/"
)
group_id = re.search("<GroupId>(.*)</GroupId>", group_data).groups()[0]
groups_data = test_client.action_data("ListGroups")